save

Author: CTY-git

patchwork/steps/ReadFile/ReadFile.py

@@ -0,0 +1,19 @@
+from patchwork.common.utils.utils import open_with_chardet
+from patchwork.step import Step
+from patchwork.steps.ReadFile.typed import ReadFileInputs
+
+
+class ReadFile(Step):
+    def __init__(self, inputs):
+        super().__init__(inputs)
+        missing_keys = ReadFileInputs.__required_keys__.difference(inputs.keys())
+        if len(missing_keys) > 0:
+            raise ValueError(f"Missing required data: {missing_keys}")
+
+        self.file = inputs["file_path"]
+
+    def run(self):
+        with open_with_chardet("r", self.file) as f:
+            file_contents = f.read()
+
+        return dict(file_content=file_contents)

patchwork/steps/ReadFile/__init__.py

@@ -0,0 +1,9 @@
+from typing_extensions import TypedDict
+
+
+class ReadFileInputs(TypedDict):
+    file_path: str
+
+
+class ReadFileOutputs(TypedDict):
+    file_content: str

patchwork/steps/ReadFile/typed.py

@@ -43,7 +43,7 @@
     "CallAPI",
     "CallCode2Prompt",
     "CallLLM",
-    "Combine"
+    "Combine",
     "CommitChanges",
     "CreateIssue",
     "CreateIssueComment",
@@ -63,6 +63,7 @@
     "PreparePR",
     "PreparePrompt",
     "QueryEmbeddings",
+    "ReadFile",
     "ReadIssues",
     "ReadPRDiffs",
     "ScanDepscan",

patchwork/steps/__init__.py

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