Files
xiaozhi-esp32-server/core/utils/llm.py
T

24 lines
832 B
Python
Raw Normal View History

import os
import sys
# 添加项目根目录到Python路径
current_dir = os.path.dirname(os.path.abspath(__file__))
project_root = os.path.abspath(os.path.join(current_dir, "..", ".."))
sys.path.insert(0, project_root)
2025-02-18 00:07:19 +08:00
from config.logger import setup_logging
import importlib
2025-02-02 23:01:14 +08:00
2025-02-18 00:07:19 +08:00
logger = setup_logging()
2025-02-02 23:01:14 +08:00
def create_instance(class_name, *args, **kwargs):
# 创建LLM实例
if os.path.exists(os.path.join('core', 'providers', 'llm', class_name, f'{class_name}.py')):
lib_name = f'core.providers.llm.{class_name}.{class_name}'
if lib_name not in sys.modules:
sys.modules[lib_name] = importlib.import_module(f'{lib_name}')
return sys.modules[lib_name].LLMProvider(*args, **kwargs)
2025-02-02 23:01:14 +08:00
raise ValueError(f"不支持的LLM类型: {class_name},请检查该配置的type是否设置正确")