SimplifidedLLMPB change variable

patched-codes · Aug 21, 2024 · e668585 · e668585
1 parent 0c3abd3
commit e668585
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/patchwork/steps/SimplifiedLLMOncePB/SimplifiedLLMOncePB.py b/patchwork/steps/SimplifiedLLMOncePB/SimplifiedLLMOncePB.py
@@ -10,8 +10,8 @@ class SimplifiedLLMOncePB(Step, input_class=SimplifiedLLMOncePBInputs):
     def __init__(self, inputs):
         super().__init__(inputs)
 
-        self.user = inputs["prompt_user"]
-        self.system = inputs.get("prompt_system")
+        self.user = inputs["user_prompt"]
+        self.system = inputs.get("system_prompt")
         self.prompt_value = inputs["prompt_value"]
         self.json_schema = inputs["json_schema"]
         self.inputs = inputs

diff --git a/patchwork/steps/SimplifiedLLMOncePB/typed.py b/patchwork/steps/SimplifiedLLMOncePB/typed.py
@@ -7,12 +7,12 @@
 class __SimplifiedLLMOncePBInputsRequired(TypedDict):
     json_schema: Annotated[Dict[str, Any], StepTypeConfig(is_config=True)]
     # PreparePromptInputs
-    prompt_user: Annotated[str, StepTypeConfig(is_config=True)]
+    user_prompt: Annotated[str, StepTypeConfig(is_config=True)]
     prompt_value: Dict[str, Any]
 
 
 class SimplifiedLLMOncePBInputs(__SimplifiedLLMOncePBInputsRequired, total=False):
-    prompt_system: Annotated[str, StepTypeConfig(is_config=True)]
+    system_prompt: Annotated[str, StepTypeConfig(is_config=True)]
     # CallLLMInputs
     model: Annotated[str, StepTypeConfig(is_config=True)]
     openai_api_key: Annotated[

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "patchwork-cli"
-version = "0.0.49.dev7"
+version = "0.0.49.dev8"
 description = ""
 authors = ["patched.codes"]
 license = "AGPL"