Merge pull request #1398 from patched-codes/agentic-llm-retry

whoisarpit · web-flow · commit 3df47c6c00f8 · 2025-03-06T14:42:06.000+08:00
Add AgenticLLM retry
diff --git a/patchwork/common/multiturn_strategy/agentic_strategy_v2.py b/patchwork/common/multiturn_strategy/agentic_strategy_v2.py
@@ -57,6 +57,7 @@ def __init__(
         self.__user_prompt_template = user_prompt_template
         self.__summariser = Agent(
             llm_client,
+            result_retries=5,
             system_prompt=mustache_render(system_prompt_template, self.__template_data),
             result_type=example_json_to_base_model(example_json),
             model_settings=dict(
diff --git a/patchwork/steps/ManageEngineAgent/ManageEngineAgent.py b/patchwork/steps/ManageEngineAgent/ManageEngineAgent.py
@@ -46,6 +46,7 @@ def __init__(self, inputs: dict):
             agent_configs=[
                 AgentConfig(
                     name="ManageEngine Assistant",
+                    model="claude-3-7-sonnet-latest",
                     tool_set=dict(
                         make_api_request=APIRequestTool(
                             headers=self.headers,
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "patchwork-cli"
-version = "0.0.107"
+version = "0.0.108"
 description = ""
 authors = ["patched.codes"]
 license = "AGPL"
