fix: fix doc task problem (#217)

petercat-ai · Aug 18, 2024 · 75e23bf · 75e23bf
2 parents 51880cb + 9f1b24c
commit 75e23bf
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/petercat_utils/rag_helper/git_doc_task.py b/petercat_utils/rag_helper/git_doc_task.py
@@ -72,7 +72,7 @@ def __init__(self,
         super().__init__(bot_id=bot_id, type=TaskType.GIT_DOC, from_id=from_id, id=id, status=status,
                          repo_name=repo_name)
         self.commit_id = commit_id
-        self.node_type = node_type
+        self.node_type = GitDocTaskNodeType(node_type)
         self.sha = sha
         self.path = path
 

diff --git a/petercat_utils/rag_helper/git_issue_task.py b/petercat_utils/rag_helper/git_issue_task.py
@@ -38,7 +38,7 @@ def __init__(self,
         super().__init__(bot_id=bot_id, type=TaskType.GIT_ISSUE, from_id=from_id, id=id, status=status,
                          repo_name=repo_name)
         self.issue_id = issue_id
-        self.node_type = node_type
+        self.node_type = GitIssueTaskNodeType(node_type)
 
     def extra_save_data(self):
         return {

diff --git a/server/bot/builder.py b/server/bot/builder.py
@@ -80,7 +80,7 @@ async def bot_builder(
         response = supabase.table("bots").insert(bot_data).execute()
         if response:
             bot_id = response.data[0]["id"]
-            trigger_rag_task(repo_name, bot_id=bot_id)
+            trigger_rag_task(repo_name=repo_name, bot_id=bot_id)
         return response
     except Exception as e:
         print(f"An error occurred: {e}")