Fix types and keys on GithubAgent

Author: whoisarpit

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,4 +1,4 @@
-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
 
@@ -11,6 +11,8 @@ class GitHubAgentInputs(__GitHubAgentRequiredInputs, total=False):
     base_path: str
     prompt_value: Dict[str, Any]
     max_llm_calls: Annotated[int, StepTypeConfig(is_config=True)]
+    github_api_key: str
+    example_json: Optional[Dict]
     openai_api_key: Annotated[
         str,
         StepTypeConfig(