Skip to content

Commit c5554ff

Browse files
committed
use singular prompt value
1 parent 615a54b commit c5554ff

File tree

2 files changed

+3
-8
lines changed

2 files changed

+3
-8
lines changed

patchwork/steps/AgenticLLM/AgenticLLM.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def __init__(self, inputs):
1717
self.agentic_strategy = AgenticStrategy(
1818
llm_client=AioLlmClient.create_aio_client(inputs),
1919
tool_set=Tool.get_tools(path=base_path),
20-
template_data=inputs.get("prompt_values"),
20+
template_data=inputs.get("prompt_value"),
2121
system_prompt_template=inputs.get("system_prompt"),
2222
user_prompt_template=inputs.get("user_prompt"),
2323
)

patchwork/steps/AgenticLLM/typed.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,9 @@
44
from patchwork.steps.CallLLM.CallLLM import TOKEN_URL
55

66

7-
class __AgenticLLMInputsRequired(TypedDict):
8-
# PreparePromptInputs
9-
prompt_template_file: Annotated[str, StepTypeConfig(is_config=True)]
10-
prompt_id: Annotated[str, StepTypeConfig(is_config=True)]
117

12-
13-
class AgenticLLMInputs(__AgenticLLMInputsRequired, total=False):
14-
prompt_values: Annotated[List[Dict[str, Any]], StepTypeConfig(or_op=["prompt_value_file"])]
8+
class AgenticLLMInputs(TypedDict, total=False):
9+
prompt_value: Dict[str, Any]
1510
system_prompt: str
1611
user_prompt: str
1712
max_llm_calls: Annotated[int, StepTypeConfig(is_config=True)]

0 commit comments

Comments
 (0)