Add fixes to GithubAgent and fix types for some other steps

patchwork/steps/AgenticLLMV2/AgenticLLMV2.py

@@ -18,15 +18,15 @@ def __init__(self, inputs):
             base_path = str(Path.cwd())
         self.conversation_limit = int(inputs.get("max_agent_calls", 1))
         self.agentic_strategy = AgenticStrategyV2(
-            model="claude-3-5-sonnet-latest",
+            model=inputs.get("strategy_model", "claude-3-5-sonnet-latest"),
             llm_client=AioLlmClient.create_aio_client(inputs),
             template_data=inputs.get("prompt_value", {}),
             system_prompt_template=inputs.get("system_prompt", "Summarise from our previous conversation"),
             user_prompt_template=inputs.get("user_prompt"),
             agent_configs=[
                 AgentConfig(
                     name="Assistant",
-                    model="claude-3-7-sonnet-latest",
+                    model=inputs.get("agent_model", "claude-3-7-sonnet-latest"),
                     tool_set=Tool.get_tools(path=base_path),
                     system_prompt=inputs.get("agent_system_prompt"),
                 )

patchwork/steps/AgenticLLMV2/typed.py

@@ -9,9 +9,32 @@ class AgenticLLMV2Inputs(TypedDict, total=False):
     system_prompt: str
     user_prompt: str
     max_agent_calls: Annotated[int, StepTypeConfig(is_config=True)]
-    anthropic_api_key: str
+    strategy_model: str
+    agent_model: str
     agent_system_prompt: str
     example_json: str
+    openai_api_key: Annotated[
+        str,
+        StepTypeConfig(
+            is_config=True, or_op=["patched_api_key", "google_api_key", "client_is_gcp", "anthropic_api_key"]
+        ),
+    ]
+    anthropic_api_key: Annotated[
+        str,
+        StepTypeConfig(is_config=True, or_op=["patched_api_key", "google_api_key", "client_is_gcp", "openai_api_key"]),
+    ]
+    google_api_key: Annotated[
+        str,
+        StepTypeConfig(
+            is_config=True, or_op=["patched_api_key", "openai_api_key", "client_is_gcp", "anthropic_api_key"]
+        ),
+    ]
+    client_is_gcp: Annotated[
+        str,
+        StepTypeConfig(
+            is_config=True, or_op=["patched_api_key", "openai_api_key", "anthropic_api_key", "google_api_key"]
+        ),
+    ]
 
 
 class AgenticLLMV2Outputs(TypedDict):

patchwork/steps/GitHubAgent/GitHubAgent.py

@@ -21,7 +21,7 @@ def __init__(self, inputs):
             model="claude-3-5-sonnet-latest",
             llm_client=AioLlmClient.create_aio_client(inputs),
             template_data=dict(),
-            system_prompt_template=f"""\
+            system_prompt_template="""\
 Please summarise the conversation given and provide the result in the structure that is asked of you.
 """,
             user_prompt_template=f"""\
@@ -34,15 +34,17 @@ def __init__(self, inputs):
                 AgentConfig(
                     name="Assistant",
                     model="gemini-2.0-flash",
-                    tool_set=dict(github_tool=GitHubTool(base_path, inputs["github_api_token"])),
+                    tool_set=dict(github_tool=GitHubTool(base_path, inputs["github_api_key"])),
                     system_prompt="""\
-You are a senior software developer helping the program manager to obtain some data from GitHub. 
-You can access github through the `gh` CLI app. 
+You are a senior software developer helping the program manager to obtain some data from GitHub.
+You can access github through the `gh` CLI app.
 Your `gh` app has already been authenticated.
 """,
                 )
             ],
-            example_json=inputs.get("example_json"),
+            example_json=inputs.get(
+                "example_json", '{"summary_of_actions": "1. Retrieved the list of repositories. 2. ..."}'
+            ),
         )
 
     def run(self) -> dict:

patchwork/steps/GitHubAgent/typed.py

@@ -1,16 +1,18 @@
-from typing_extensions import Annotated, Any, Dict, TypedDict
+from typing_extensions import Annotated, Any, Dict, Optional, TypedDict
 
 from patchwork.common.utils.step_typing import StepTypeConfig
 
 
 class __GitHubAgentRequiredInputs(TypedDict):
+    github_api_key: str
     task: str
 
 
 class GitHubAgentInputs(__GitHubAgentRequiredInputs, total=False):
     base_path: str
     prompt_value: Dict[str, Any]
     max_llm_calls: Annotated[int, StepTypeConfig(is_config=True)]
+    example_json: str
     openai_api_key: Annotated[
         str,
         StepTypeConfig(

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "patchwork-cli"
-version = "0.0.121"
+version = "0.0.122"
 description = ""
 authors = ["patched.codes"]
 license = "AGPL"