diff --git a/petercat_utils/rag_helper/git_doc_task.py b/petercat_utils/rag_helper/git_doc_task.py index b3133bc1..7befb8fd 100644 --- a/petercat_utils/rag_helper/git_doc_task.py +++ b/petercat_utils/rag_helper/git_doc_task.py @@ -93,7 +93,7 @@ def handle_tree_node(self): task_list = list( filter( - lambda item: item["path"].endswith(".md") or item["node_type"] == "tree", + lambda item: item["path"].endswith(".md") or item["node_type"] == GitDocTaskNodeType.TREE.value, map( lambda item: { "repo_name": self.repo_name, diff --git a/pyproject.toml b/pyproject.toml index 369a4561..808ee806 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "petercat_utils" -version = "0.1.27" +version = "0.1.28" description = "" authors = ["raoha.rh "] readme = "README.md"