Fix parsed code bug & version bump (#22)

* Fix * Bump version
shruti222patel · Aug 22, 2023 · a3ed389 · a3ed389
1 parent d786231
commit a3ed389
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/.repo_gpt/code_embeddings.pkl b/.repo_gpt/code_embeddings.pkl
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "repo-gpt"
-version = "0.1.2"
+version = "0.1.3"
 description = "Search your code repository using GPT3.5 or GPT4."
 authors = ["Shruti Patel <[email protected]>"]
 license = "Apache License 2.0"

diff --git a/src/repo_gpt/code_manager/code_extractor.py b/src/repo_gpt/code_manager/code_extractor.py
@@ -138,9 +138,11 @@ def extract_functions_from_file(
         self, filepath: str, file_checksum: str
     ) -> List[ParsedCode]:
         handler = self.get_handler(filepath)
+        code_blocks = []
         if handler:
-            parsed_code = handler().extract_code(filepath)
-            for code in parsed_code:
+            code_blocks = handler().extract_code(filepath)
+            for code in code_blocks:
                 code.filepath = filepath
                 code.file_checksum = file_checksum
-        return parsed_code
+
+        return code_blocks