chore: the bot directory was changed to models

This commit is contained in:
zhayujie
2026-02-01 15:21:28 +08:00
parent 0e85fcfe51
commit 4a1fae3cb4
33 changed files with 76 additions and 76 deletions

View File

@@ -22,7 +22,7 @@ class MemoryGetTool(BaseTool):
"properties": { "properties": {
"path": { "path": {
"type": "string", "type": "string",
"description": "Relative path to the memory file (e.g., 'MEMORY.md', 'memory/2024-01-29.md')" "description": "Relative path to the memory file (e.g., 'MEMORY.md', 'memory/2026-01-01.md')"
}, },
"start_line": { "start_line": {
"type": "integer", "type": "integer",

View File

@@ -5,7 +5,7 @@ Agent Bridge - Integrates Agent system with existing COW bridge
from typing import Optional, List from typing import Optional, List
from agent.protocol import Agent, LLMModel, LLMRequest from agent.protocol import Agent, LLMModel, LLMRequest
from bot.openai_compatible_bot import OpenAICompatibleBot from models.openai_compatible_bot import OpenAICompatibleBot
from bridge.bridge import Bridge from bridge.bridge import Bridge
from bridge.context import Context from bridge.context import Context
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType

View File

@@ -1,4 +1,4 @@
from bot.bot_factory import create_bot from models.bot_factory import create_bot
from bridge.context import Context from bridge.context import Context
from bridge.reply import Reply from bridge.reply import Reply
from common import const from common import const

View File

@@ -9,9 +9,9 @@ import openai.error
import broadscope_bailian import broadscope_bailian
from broadscope_bailian import ChatQaMessage from broadscope_bailian import ChatQaMessage
from bot.bot import Bot from models.bot import Bot
from bot.ali.ali_qwen_session import AliQwenSession from models.ali.ali_qwen_session import AliQwenSession
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import ContextType from bridge.context import ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger

View File

@@ -1,4 +1,4 @@
from bot.session_manager import Session from models.session_manager import Session
from common.log import logger from common.log import logger
""" """

View File

@@ -2,7 +2,7 @@
import requests import requests
from bot.bot import Bot from models.bot import Bot
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType

View File

@@ -3,13 +3,13 @@
import requests import requests
import json import json
from common import const from common import const
from bot.bot import Bot from models.bot import Bot
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import ContextType from bridge.context import ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger
from config import conf from config import conf
from bot.baidu.baidu_wenxin_session import BaiduWenxinSession from models.baidu.baidu_wenxin_session import BaiduWenxinSession
BAIDU_API_KEY = conf().get("baidu_wenxin_api_key") BAIDU_API_KEY = conf().get("baidu_wenxin_api_key")
BAIDU_SECRET_KEY = conf().get("baidu_wenxin_secret_key") BAIDU_SECRET_KEY = conf().get("baidu_wenxin_secret_key")

View File

@@ -1,4 +1,4 @@
from bot.session_manager import Session from models.session_manager import Session
from common.log import logger from common.log import logger
""" """

View File

@@ -12,64 +12,64 @@ def create_bot(bot_type):
""" """
if bot_type == const.BAIDU: if bot_type == const.BAIDU:
# 替换Baidu Unit为Baidu文心千帆对话接口 # 替换Baidu Unit为Baidu文心千帆对话接口
# from bot.baidu.baidu_unit_bot import BaiduUnitBot # from models.baidu.baidu_unit_bot import BaiduUnitBot
# return BaiduUnitBot() # return BaiduUnitBot()
from bot.baidu.baidu_wenxin import BaiduWenxinBot from models.baidu.baidu_wenxin import BaiduWenxinBot
return BaiduWenxinBot() return BaiduWenxinBot()
elif bot_type == const.CHATGPT: elif bot_type == const.CHATGPT:
# ChatGPT 网页端web接口 # ChatGPT 网页端web接口
from bot.chatgpt.chat_gpt_bot import ChatGPTBot from models.chatgpt.chat_gpt_bot import ChatGPTBot
return ChatGPTBot() return ChatGPTBot()
elif bot_type == const.OPEN_AI: elif bot_type == const.OPEN_AI:
# OpenAI 官方对话模型API # OpenAI 官方对话模型API
from bot.openai.open_ai_bot import OpenAIBot from models.openai.open_ai_bot import OpenAIBot
return OpenAIBot() return OpenAIBot()
elif bot_type == const.CHATGPTONAZURE: elif bot_type == const.CHATGPTONAZURE:
# Azure chatgpt service https://azure.microsoft.com/en-in/products/cognitive-services/openai-service/ # Azure chatgpt service https://azure.microsoft.com/en-in/products/cognitive-services/openai-service/
from bot.chatgpt.chat_gpt_bot import AzureChatGPTBot from models.chatgpt.chat_gpt_bot import AzureChatGPTBot
return AzureChatGPTBot() return AzureChatGPTBot()
elif bot_type == const.XUNFEI: elif bot_type == const.XUNFEI:
from bot.xunfei.xunfei_spark_bot import XunFeiBot from models.xunfei.xunfei_spark_bot import XunFeiBot
return XunFeiBot() return XunFeiBot()
elif bot_type == const.LINKAI: elif bot_type == const.LINKAI:
from bot.linkai.link_ai_bot import LinkAIBot from models.linkai.link_ai_bot import LinkAIBot
return LinkAIBot() return LinkAIBot()
elif bot_type == const.CLAUDEAI: elif bot_type == const.CLAUDEAI:
from bot.claude.claude_ai_bot import ClaudeAIBot from models.claude.claude_ai_bot import ClaudeAIBot
return ClaudeAIBot() return ClaudeAIBot()
elif bot_type == const.CLAUDEAPI: elif bot_type == const.CLAUDEAPI:
from bot.claudeapi.claude_api_bot import ClaudeAPIBot from models.claudeapi.claude_api_bot import ClaudeAPIBot
return ClaudeAPIBot() return ClaudeAPIBot()
elif bot_type == const.QWEN: elif bot_type == const.QWEN:
from bot.ali.ali_qwen_bot import AliQwenBot from models.ali.ali_qwen_bot import AliQwenBot
return AliQwenBot() return AliQwenBot()
elif bot_type == const.QWEN_DASHSCOPE: elif bot_type == const.QWEN_DASHSCOPE:
from bot.dashscope.dashscope_bot import DashscopeBot from models.dashscope.dashscope_bot import DashscopeBot
return DashscopeBot() return DashscopeBot()
elif bot_type == const.GEMINI: elif bot_type == const.GEMINI:
from bot.gemini.google_gemini_bot import GoogleGeminiBot from models.gemini.google_gemini_bot import GoogleGeminiBot
return GoogleGeminiBot() return GoogleGeminiBot()
elif bot_type == const.ZHIPU_AI: elif bot_type == const.ZHIPU_AI:
from bot.zhipuai.zhipuai_bot import ZHIPUAIBot from models.zhipuai.zhipuai_bot import ZHIPUAIBot
return ZHIPUAIBot() return ZHIPUAIBot()
elif bot_type == const.MOONSHOT: elif bot_type == const.MOONSHOT:
from bot.moonshot.moonshot_bot import MoonshotBot from models.moonshot.moonshot_bot import MoonshotBot
return MoonshotBot() return MoonshotBot()
elif bot_type == const.MiniMax: elif bot_type == const.MiniMax:
from bot.minimax.minimax_bot import MinimaxBot from models.minimax.minimax_bot import MinimaxBot
return MinimaxBot() return MinimaxBot()
elif bot_type == const.MODELSCOPE: elif bot_type == const.MODELSCOPE:
from bot.modelscope.modelscope_bot import ModelScopeBot from models.modelscope.modelscope_bot import ModelScopeBot
return ModelScopeBot() return ModelScopeBot()

View File

@@ -7,17 +7,17 @@ import openai
import openai.error import openai.error
import requests import requests
from common import const from common import const
from bot.bot import Bot from models.bot import Bot
from bot.openai_compatible_bot import OpenAICompatibleBot from models.openai_compatible_bot import OpenAICompatibleBot
from bot.chatgpt.chat_gpt_session import ChatGPTSession from models.chatgpt.chat_gpt_session import ChatGPTSession
from bot.openai.open_ai_image import OpenAIImage from models.openai.open_ai_image import OpenAIImage
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import ContextType from bridge.context import ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger
from common.token_bucket import TokenBucket from common.token_bucket import TokenBucket
from config import conf, load_config from config import conf, load_config
from bot.baidu.baidu_wenxin_session import BaiduWenxinSession from models.baidu.baidu_wenxin_session import BaiduWenxinSession
# OpenAI对话模型API (可用) # OpenAI对话模型API (可用)
class ChatGPTBot(Bot, OpenAIImage, OpenAICompatibleBot): class ChatGPTBot(Bot, OpenAIImage, OpenAICompatibleBot):

View File

@@ -1,4 +1,4 @@
from bot.session_manager import Session from models.session_manager import Session
from common.log import logger from common.log import logger
from common import const from common import const

View File

@@ -5,9 +5,9 @@ import time
import requests import requests
from bot.baidu.baidu_wenxin_session import BaiduWenxinSession from models.baidu.baidu_wenxin_session import BaiduWenxinSession
from bot.bot import Bot from models.bot import Bot
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import ContextType from bridge.context import ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common import const from common import const
@@ -16,7 +16,7 @@ from config import conf
# Optional OpenAI image support # Optional OpenAI image support
try: try:
from bot.openai.open_ai_image import OpenAIImage from models.openai.open_ai_image import OpenAIImage
_openai_image_available = True _openai_image_available = True
except Exception as e: except Exception as e:
logger.warning(f"OpenAI image support not available: {e}") logger.warning(f"OpenAI image support not available: {e}")

View File

@@ -1,7 +1,7 @@
# encoding:utf-8 # encoding:utf-8
from bot.bot import Bot from models.bot import Bot
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import ContextType from bridge.context import ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger

View File

@@ -1,4 +1,4 @@
from bot.session_manager import Session from models.session_manager import Session
from common.log import logger from common.log import logger

View File

@@ -9,15 +9,15 @@ Google gemini bot
import json import json
import time import time
import requests import requests
from bot.bot import Bot from models.bot import Bot
import google.generativeai as genai import google.generativeai as genai
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import ContextType, Context from bridge.context import ContextType, Context
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger
from config import conf from config import conf
from bot.chatgpt.chat_gpt_session import ChatGPTSession from models.chatgpt.chat_gpt_session import ChatGPTSession
from bot.baidu.baidu_wenxin_session import BaiduWenxinSession from models.baidu.baidu_wenxin_session import BaiduWenxinSession
from google.generativeai.types import HarmCategory, HarmBlockThreshold from google.generativeai.types import HarmCategory, HarmBlockThreshold

View File

@@ -6,10 +6,10 @@ import time
import requests import requests
import json import json
import config import config
from bot.bot import Bot from models.bot import Bot
from bot.openai_compatible_bot import OpenAICompatibleBot from models.openai_compatible_bot import OpenAICompatibleBot
from bot.chatgpt.chat_gpt_session import ChatGPTSession from models.chatgpt.chat_gpt_session import ChatGPTSession
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import Context, ContextType from bridge.context import Context, ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger

View File

@@ -4,14 +4,14 @@ import time
import openai import openai
import openai.error import openai.error
from bot.bot import Bot from models.bot import Bot
from bot.minimax.minimax_session import MinimaxSession from models.minimax.minimax_session import MinimaxSession
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import Context, ContextType from bridge.context import Context, ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger
from config import conf, load_config from config import conf, load_config
from bot.chatgpt.chat_gpt_session import ChatGPTSession from models.chatgpt.chat_gpt_session import ChatGPTSession
import requests import requests
from common import const from common import const

View File

@@ -1,4 +1,4 @@
from bot.session_manager import Session from models.session_manager import Session
from common.log import logger from common.log import logger
""" """

View File

@@ -4,8 +4,8 @@ import time
import json import json
import openai import openai
import openai.error import openai.error
from bot.bot import Bot from models.bot import Bot
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import ContextType from bridge.context import ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger

View File

@@ -1,4 +1,4 @@
from bot.session_manager import Session from models.session_manager import Session
from common.log import logger from common.log import logger

View File

@@ -4,8 +4,8 @@ import time
import openai import openai
import openai.error import openai.error
from bot.bot import Bot from models.bot import Bot
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import ContextType from bridge.context import ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger

View File

@@ -1,4 +1,4 @@
from bot.session_manager import Session from models.session_manager import Session
from common.log import logger from common.log import logger

View File

@@ -5,11 +5,11 @@ import time
import openai import openai
import openai.error import openai.error
from bot.bot import Bot from models.bot import Bot
from bot.openai_compatible_bot import OpenAICompatibleBot from models.openai_compatible_bot import OpenAICompatibleBot
from bot.openai.open_ai_image import OpenAIImage from models.openai.open_ai_image import OpenAIImage
from bot.openai.open_ai_session import OpenAISession from models.openai.open_ai_session import OpenAISession
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import ContextType from bridge.context import ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger

View File

@@ -1,7 +1,7 @@
import time import time
import openai import openai
from bot.openai.openai_compat import RateLimitError from models.openai.openai_compat import RateLimitError
from common.log import logger from common.log import logger
from common.token_bucket import TokenBucket from common.token_bucket import TokenBucket

View File

@@ -1,4 +1,4 @@
from bot.session_manager import Session from models.session_manager import Session
from common.log import logger from common.log import logger

View File

@@ -1,9 +1,9 @@
# encoding:utf-8 # encoding:utf-8
import requests, json import requests, json
from bot.bot import Bot from models.bot import Bot
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bot.chatgpt.chat_gpt_session import ChatGPTSession from models.chatgpt.chat_gpt_session import ChatGPTSession
from bridge.context import ContextType, Context from bridge.context import ContextType, Context
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger

View File

@@ -1,4 +1,4 @@
from bot.session_manager import Session from models.session_manager import Session
from common.log import logger from common.log import logger

View File

@@ -4,10 +4,10 @@ import time
import openai import openai
import openai.error import openai.error
from bot.bot import Bot from models.bot import Bot
from bot.zhipuai.zhipu_ai_session import ZhipuAISession from models.zhipuai.zhipu_ai_session import ZhipuAISession
from bot.zhipuai.zhipu_ai_image import ZhipuAIImage from models.zhipuai.zhipu_ai_image import ZhipuAIImage
from bot.session_manager import SessionManager from models.session_manager import SessionManager
from bridge.context import ContextType from bridge.context import ContextType
from bridge.reply import Reply, ReplyType from bridge.reply import Reply, ReplyType
from common.log import logger from common.log import logger