@@ -67,7 +67,7 @@ class PluginManager(Singleton):
|
|||||||
current_dir = os.path.dirname(os.path.abspath(__file__))
|
current_dir = os.path.dirname(os.path.abspath(__file__))
|
||||||
# 回退三级到项目根目录 (core/managers -> core -> neobot -> src)
|
# 回退三级到项目根目录 (core/managers -> core -> neobot -> src)
|
||||||
root_dir = os.path.dirname(os.path.dirname(os.path.dirname(current_dir)))
|
root_dir = os.path.dirname(os.path.dirname(os.path.dirname(current_dir)))
|
||||||
plugin_dir = os.path.join(root_dir, "src", "neobot", "plugins")
|
plugin_dir = os.path.join(root_dir, "neobot", "plugins")
|
||||||
|
|
||||||
# 使用完整的包名:neobot.plugins
|
# 使用完整的包名:neobot.plugins
|
||||||
package_name = "neobot.plugins"
|
package_name = "neobot.plugins"
|
||||||
|
|||||||
@@ -4,7 +4,10 @@ AI 聊天插件,支持向量数据库记忆功能
|
|||||||
"""
|
"""
|
||||||
import time
|
import time
|
||||||
import uuid
|
import uuid
|
||||||
|
<<<<<<< HEAD
|
||||||
|
=======
|
||||||
<<<<<<< HEAD:src/neobot/plugins/ai_chat.py
|
<<<<<<< HEAD:src/neobot/plugins/ai_chat.py
|
||||||
|
>>>>>>> eb9079744c82f8e254de084a3a089ef91c37e9dc
|
||||||
import os
|
import os
|
||||||
import base64
|
import base64
|
||||||
from neobot.core.managers.command_manager import matcher
|
from neobot.core.managers.command_manager import matcher
|
||||||
@@ -13,6 +16,8 @@ from neobot.core.managers.vectordb_manager import vectordb_manager
|
|||||||
from neobot.core.managers.image_manager import image_manager
|
from neobot.core.managers.image_manager import image_manager
|
||||||
from neobot.core.utils.logger import ModuleLogger
|
from neobot.core.utils.logger import ModuleLogger
|
||||||
from neobot.core.config_loader import global_config
|
from neobot.core.config_loader import global_config
|
||||||
|
<<<<<<< HEAD
|
||||||
|
=======
|
||||||
=======
|
=======
|
||||||
import markdown
|
import markdown
|
||||||
from core.managers.command_manager import matcher
|
from core.managers.command_manager import matcher
|
||||||
@@ -23,6 +28,7 @@ from core.managers.image_manager import image_manager
|
|||||||
from core.utils.logger import ModuleLogger
|
from core.utils.logger import ModuleLogger
|
||||||
from core.config_loader import global_config
|
from core.config_loader import global_config
|
||||||
>>>>>>> origin/main:plugins/ai_chat.py
|
>>>>>>> origin/main:plugins/ai_chat.py
|
||||||
|
>>>>>>> eb9079744c82f8e254de084a3a089ef91c37e9dc
|
||||||
|
|
||||||
logger = ModuleLogger("AIChat")
|
logger = ModuleLogger("AIChat")
|
||||||
|
|
||||||
@@ -43,10 +49,13 @@ async def get_ai_response(user_id: int, group_id: int, user_message: str) -> str
|
|||||||
if not OPENAI_AVAILABLE:
|
if not OPENAI_AVAILABLE:
|
||||||
return "请先安装 openai 库: pip install openai"
|
return "请先安装 openai 库: pip install openai"
|
||||||
|
|
||||||
|
<<<<<<< HEAD
|
||||||
|
=======
|
||||||
<<<<<<< HEAD:src/neobot/plugins/ai_chat.py
|
<<<<<<< HEAD:src/neobot/plugins/ai_chat.py
|
||||||
=======
|
=======
|
||||||
# 从配置中获取 DeepSeek API 配置(复用跨平台插件的配置或全局配置)
|
# 从配置中获取 DeepSeek API 配置(复用跨平台插件的配置或全局配置)
|
||||||
>>>>>>> origin/main:plugins/ai_chat.py
|
>>>>>>> origin/main:plugins/ai_chat.py
|
||||||
|
>>>>>>> eb9079744c82f8e254de084a3a089ef91c37e9dc
|
||||||
api_key = getattr(global_config.cross_platform, 'deepseek_api_key', None) or "sk-f71322a9fbba4b05a7df969cb4004f06"
|
api_key = getattr(global_config.cross_platform, 'deepseek_api_key', None) or "sk-f71322a9fbba4b05a7df969cb4004f06"
|
||||||
api_url = getattr(global_config.cross_platform, 'deepseek_api_url', "https://api.deepseek.com/v1")
|
api_url = getattr(global_config.cross_platform, 'deepseek_api_url', "https://api.deepseek.com/v1")
|
||||||
model = getattr(global_config.cross_platform, 'deepseek_model', "deepseek-chat")
|
model = getattr(global_config.cross_platform, 'deepseek_model', "deepseek-chat")
|
||||||
@@ -158,7 +167,10 @@ async def chat_command(event: GroupMessageEvent | PrivateMessageEvent, args: lis
|
|||||||
await event.reply("正在思考中...")
|
await event.reply("正在思考中...")
|
||||||
reply = await get_ai_response(user_id, group_id, user_message)
|
reply = await get_ai_response(user_id, group_id, user_message)
|
||||||
|
|
||||||
|
<<<<<<< HEAD
|
||||||
|
=======
|
||||||
<<<<<<< HEAD:src/neobot/plugins/ai_chat.py
|
<<<<<<< HEAD:src/neobot/plugins/ai_chat.py
|
||||||
|
>>>>>>> eb9079744c82f8e254de084a3a089ef91c37e9dc
|
||||||
try:
|
try:
|
||||||
image_base64 = await generate_chat_image_base64(
|
image_base64 = await generate_chat_image_base64(
|
||||||
user_name=str(event.user_id),
|
user_name=str(event.user_id),
|
||||||
@@ -174,6 +186,8 @@ async def chat_command(event: GroupMessageEvent | PrivateMessageEvent, args: lis
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.error(f"生成聊天图片失败: {e}")
|
logger.error(f"生成聊天图片失败: {e}")
|
||||||
await event.reply(reply)
|
await event.reply(reply)
|
||||||
|
<<<<<<< HEAD
|
||||||
|
=======
|
||||||
=======
|
=======
|
||||||
# 将 Markdown 转换为 HTML
|
# 将 Markdown 转换为 HTML
|
||||||
try:
|
try:
|
||||||
@@ -206,3 +220,4 @@ async def chat_command(event: GroupMessageEvent | PrivateMessageEvent, args: lis
|
|||||||
logger.error(f"渲染聊天图片失败: {e}")
|
logger.error(f"渲染聊天图片失败: {e}")
|
||||||
await event.reply("图片生成失败,返回文本:\n" + reply)
|
await event.reply("图片生成失败,返回文本:\n" + reply)
|
||||||
>>>>>>> origin/main:plugins/ai_chat.py
|
>>>>>>> origin/main:plugins/ai_chat.py
|
||||||
|
>>>>>>> eb9079744c82f8e254de084a3a089ef91c37e9dc
|
||||||
|
|||||||
@@ -17,11 +17,7 @@ class CrossPlatformConfig:
|
|||||||
self.ENABLE_CROSS_PLATFORM = True
|
self.ENABLE_CROSS_PLATFORM = True
|
||||||
|
|
||||||
# DeepSeek API 配置 - 从环境变量或配置文件加载
|
# DeepSeek API 配置 - 从环境变量或配置文件加载
|
||||||
<<<<<<< HEAD:src/neobot/plugins/discord-cross/config.py
|
|
||||||
self.DEEPSEEK_API_KEY = os.environ.get("DEEPSEEK_API_KEY", "sk-28b794e08e184f868d6c0107a46e0c3e")
|
|
||||||
=======
|
|
||||||
self.DEEPSEEK_API_KEY = os.environ.get("DEEPSEEK_API_KEY", "sk-f71322a9fbba4b05a7df969cb4004f06")
|
self.DEEPSEEK_API_KEY = os.environ.get("DEEPSEEK_API_KEY", "sk-f71322a9fbba4b05a7df969cb4004f06")
|
||||||
>>>>>>> origin/main:plugins/discord-cross/config.py
|
|
||||||
self.DEEPSEEK_API_URL = os.environ.get("DEEPSEEK_API_URL", "https://api.deepseek.com/v1/chat/completions")
|
self.DEEPSEEK_API_URL = os.environ.get("DEEPSEEK_API_URL", "https://api.deepseek.com/v1/chat/completions")
|
||||||
self.DEEPSEEK_MODEL = os.environ.get("DEEPSEEK_MODEL", "deepseek-chat")
|
self.DEEPSEEK_MODEL = os.environ.get("DEEPSEEK_MODEL", "deepseek-chat")
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user