better typing

Author: CTY-git

patchwork/steps/GitHubAgent/typed.py

@@ -3,10 +3,12 @@
 from patchwork.common.utils.step_typing import StepTypeConfig
 
 
-class GitHubAgentInputs(TypedDict, total=False):
+class __GitHubAgentRequiredInputs(TypedDict):
+    task: str
+
+class GitHubAgentInputs(__GitHubAgentRequiredInputs, total=False):
     base_path: str
     prompt_value: Dict[str, Any]
-    task: str
     max_llm_calls: Annotated[int, StepTypeConfig(is_config=True)]
     openai_api_key: Annotated[
         str, StepTypeConfig(is_config=True, or_op=["patched_api_key", "google_api_key", "anthropic_api_key"])