use singular prompt value

patched-codes · Jan 6, 2025 · c5554ff · c5554ff
1 parent 615a54b
commit c5554ff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/patchwork/steps/AgenticLLM/AgenticLLM.py b/patchwork/steps/AgenticLLM/AgenticLLM.py
@@ -17,7 +17,7 @@ def __init__(self, inputs):
         self.agentic_strategy = AgenticStrategy(
             llm_client=AioLlmClient.create_aio_client(inputs),
             tool_set=Tool.get_tools(path=base_path),
-            template_data=inputs.get("prompt_values"),
+            template_data=inputs.get("prompt_value"),
             system_prompt_template=inputs.get("system_prompt"),
             user_prompt_template=inputs.get("user_prompt"),
         )

diff --git a/patchwork/steps/AgenticLLM/typed.py b/patchwork/steps/AgenticLLM/typed.py
@@ -4,14 +4,9 @@
 from patchwork.steps.CallLLM.CallLLM import TOKEN_URL
 
 
-class __AgenticLLMInputsRequired(TypedDict):
-    # PreparePromptInputs
-    prompt_template_file: Annotated[str, StepTypeConfig(is_config=True)]
-    prompt_id: Annotated[str, StepTypeConfig(is_config=True)]
 
-
-class AgenticLLMInputs(__AgenticLLMInputsRequired, total=False):
-    prompt_values: Annotated[List[Dict[str, Any]], StepTypeConfig(or_op=["prompt_value_file"])]
+class AgenticLLMInputs(TypedDict, total=False):
+    prompt_value: Dict[str, Any]
     system_prompt: str
     user_prompt: str
     max_llm_calls: Annotated[int, StepTypeConfig(is_config=True)]