Add sensible defaults (#140)

* add sensible defaults

* update lockfile

* update tests

* fix bound value in lambda

Author: CTY-git

patchwork/common/utils.py

@@ -8,7 +8,7 @@
 from chromadb.api.types import Documents, EmbeddingFunction
 from chromadb.utils import embedding_functions
 from git import Head, Repo
-from typing_extensions import Callable
+from typing_extensions import Callable, TypedDict
 
 from patchwork.managed_files import HOME_FOLDER
 
@@ -162,3 +162,7 @@ def get_current_branch(repo: Repo) -> Head:
         )
 
     return from_branch
+
+
+def get_required_keys(cls: TypedDict) -> set:
+    return getattr(cls, "__required_keys__", set())

patchwork/steps/CommitChanges/CommitChanges.py

@@ -122,7 +122,7 @@ def __init__(self, inputs: dict):
             self.enabled = False
 
         self.force = inputs.get("force_branch_creation", True)
-        self.branch_prefix = inputs.get("branch_prefix", "")
+        self.branch_prefix = inputs.get("branch_prefix", "patchwork-")
         self.branch_suffix = inputs.get("branch_suffix", "")
         if self.enabled and self.branch_prefix == "" and self.branch_suffix == "":
             raise ValueError("Both branch_prefix and branch_suffix cannot be empty")

patchwork/steps/ExtractModelResponse/ExtractModelResponse.py

@@ -18,8 +18,12 @@ def __init__(self, inputs: dict):
 
     def run(self) -> dict:
         if len(self.partitions) <= 0:
-            logger.error("No partitions specified for model response. Exiting.")
-            return dict(extracted_responses=[])
+            outputs = []
+            for openai_response in self.openai_responses:
+                output = defaultdict(lambda bound_value=openai_response: bound_value)
+                outputs.append(output)
+            logger.error("No partitions specified for model response. defaulting.")
+            return dict(extracted_responses=outputs)
 
         outputs = []
         for openai_response in self.openai_responses:

patchwork/steps/ExtractModelResponse/typed.py

@@ -1,10 +1,10 @@
 from typing_extensions import Iterable, NotRequired, TypedDict
 
 
-class ExtractModelResponsesInputs(TypedDict):
+class ExtractModelResponseInputs(TypedDict):
     openai_responses: Iterable[str]
     response_partitions: NotRequired[dict[str, list[str]]]
 
 
-class ExtractModelResponsesOutputs(TypedDict):
+class ExtractModelResponseOutputs(TypedDict):
     extracted_responses: list[dict[str, str]]

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "patchwork-cli"
-version = "0.0.13"
+version = "0.0.14"
 description = ""
 authors = ["patched.codes"]
 license = "AGPL"

pyproject.toml

@@ -27,7 +27,9 @@ def test_init_missing_required_keys():
 def test_run_no_partitions(sample_inputs):
     step = ExtractModelResponse({**sample_inputs, "response_partitions": {}})
     output = step.run()
-    assert output == {"extracted_responses": []}
+    assert len(output["extracted_responses"]) == 2
+    assert output["extracted_responses"][0]["anyKeyHere"] == "partition1response1partition2"
+    assert output["extracted_responses"][1]["kEy"] == "response2partition3"
 
 
 def test_run_with_partitions(sample_inputs):