From 9dfcea1dde3b30786516218f65deb425d2bd2399 Mon Sep 17 00:00:00 2001 From: MadratJerry Date: Sun, 18 Aug 2024 16:36:49 +0800 Subject: [PATCH] fix: fix incorrect module import --- server/bot/builder.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/server/bot/builder.py b/server/bot/builder.py index dd60e5b1..c4599362 100644 --- a/server/bot/builder.py +++ b/server/bot/builder.py @@ -3,7 +3,7 @@ from github import Github from petercat_utils import get_client from petercat_utils.data_class import RAGGitDocConfig -from petercat_utils.rag_helper.git_doc_task import add_rag_git_doc_task +from petercat_utils import git_doc_task from prompts.bot_template import generate_prompt_by_repo_name g = Github() @@ -53,7 +53,7 @@ def trigger_rag_task(repo_name: str, bot_id: str): file_path="", commit_id="", ) - add_rag_git_doc_task(config) + git_doc_task.add_rag_git_doc_task(config) except Exception as e: print(f"trigger_rag_task error: {e}")