fix: bot builder (#335)

- 修 bot builder

server/agent/bot/bot_builder.py

@@ -1,12 +1,11 @@
 from typing import AsyncIterator
-from agent.bot.get_bot import Bot
+from agent.llm.openai import OpenAIClient
 from petercat_utils.data_class import ChatData
 
 from agent.base import AgentBuilder
 from agent.prompts.bot_builder import generate_prompt_by_user_id
 from agent.tools import bot_builder
 
-
 TOOL_MAPPING = {
     "create_bot": bot_builder.create_bot,
     "edit_bot": bot_builder.edit_bot,
@@ -16,11 +15,11 @@
 def agent_stream_chat(
     input_data: ChatData, 
     user_id: str,
-    bot: Bot,
+    bot_id: str,
 ) -> AsyncIterator[str]:
-    prompt = generate_prompt_by_user_id(user_id, bot.id)
+    prompt = generate_prompt_by_user_id(user_id, bot_id)
     agent = AgentBuilder(
-        chat_model=bot.llm,
+        chat_model=OpenAIClient(),
         prompt=prompt, tools=TOOL_MAPPING, enable_tavily=False
     )
     return agent.run_stream_chat(input_data)

server/agent/prompts/bot_builder.py

@@ -1,3 +1,6 @@
+from typing import Optional
+
+
 CREATE_PROMPT = """
 ## Role:
 You are a GitHub Answering Bot Creation Assistant. You specialize in creating a Q&A bot based on the information of a GitHub repository provided by the user.
@@ -45,7 +48,7 @@
 """
 
 
-def generate_prompt_by_user_id(user_id: str, bot_id: str):
+def generate_prompt_by_user_id(user_id: str, bot_id: Optional[str]):
     if bot_id:
       return EDIT_PROMPT.format(bot_id=bot_id, user_id=user_id)
     else: