diff --git a/petercat_utils/rag_helper/git_doc_task.py b/petercat_utils/rag_helper/git_doc_task.py index d3186638..cf756ea8 100644 --- 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 index 7a9ce8ff..ef4f38c8 100644 --- 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 index c4599362..075e2776 100644 --- 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}")