simplify and reduce context usagr

patched-codes · Dec 2, 2024 · 144bee7 · 144bee7
1 parent 0331578
commit 144bee7
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -207,6 +207,7 @@ jobs:
           poetry run patchwork GenerateDiagram --log debug \
           --patched_api_key=${{ secrets.PATCHED_API_KEY }} \
           --github_api_key=${{ secrets.SCM_GITHUB_KEY }} \
+          --folder_path=patchwork \
           --disable_telemetry
 
       - name: Generate UnitTests

diff --git a/patchwork/patchflows/GenerateDiagram/GenerateDiagram.py b/patchwork/patchflows/GenerateDiagram/GenerateDiagram.py
@@ -25,10 +25,6 @@ def __init__(self, inputs):
         final_inputs.update(inputs)
 
         final_inputs["prompt_id"] = "GenerateDiagram"
-        if "folder_path" not in final_inputs.keys():
-            final_inputs["folder_path"] = Path.cwd()
-        else:
-            final_inputs["folder_path"] = Path(final_inputs["folder_path"])
 
         if "prompt_template_file" not in final_inputs:
             final_inputs["prompt_template_file"] = _DEFAULT_PROMPT_JSON

diff --git a/patchwork/patchflows/GenerateDiagram/defaults.yml b/patchwork/patchflows/GenerateDiagram/defaults.yml
@@ -10,8 +10,8 @@
 # model_top_p: 0.95
 # model_max_tokens: 2000
 
-base_path : diagram.md
-
+base_path: diagram.md
+folder_path: .
 # CommitChanges Inputs
 disable_branch: false