add agent_model and strategy_model

Jatin-Roopchandani · whoisarpit · commit 21948cbec44d · 2025-04-18T16:13:52.000+08:00
diff --git a/patchwork/steps/SlackAgent/SlackAgent.py b/patchwork/steps/SlackAgent/SlackAgent.py
@@ -34,23 +34,23 @@ def __init__(self, inputs: dict):
 
         llm_client = AioLlmClient.create_aio_client(inputs)
 
-        model = inputs.get("model")
-        
+        agent_model = inputs.get("agent_model", "gemini-2.0-flash")
+        strategy_model = inputs.get("strategy_model", "gemini-2.0-flash")  
         api_tool = APIRequestTool(
             headers=self.headers,
         )
 
 
         self.agentic_strategy = AgenticStrategyV2(
-            model=model,
+            model=agent_model,
             llm_client=llm_client,
             system_prompt_template=system_prompt,
             template_data={},
             user_prompt_template=mustache_render(inputs.get("user_prompt"), inputs.get("prompt_value")),
             agent_configs=[
                 AgentConfig(
                     name="Slack Assistant",
-                    model=model,
+                    model=strategy_model,
                     tool_set=dict(
                         make_api_request=api_tool
                     ),
diff --git a/patchwork/steps/SlackAgent/typed.py b/patchwork/steps/SlackAgent/typed.py
@@ -12,7 +12,8 @@ class __SlackAgentInputsRequired(TypedDict):
 
 class SlackAgentInputs(__SlackAgentInputsRequired, total=False):
     max_agent_calls: int
-    model: str
+    strategy_model: str
+    agent_model: str
     openai_api_key: Annotated[str, StepTypeConfig(or_op=["google_api_key", "anthropic_api_key"])]
     anthropic_api_key: Annotated[str, StepTypeConfig(or_op=["google_api_key", "openai_api_key"])]
     google_api_key: Annotated[str, StepTypeConfig(or_op=["openai_api_key", "anthropic_api_key"])]
