You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
|
|
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)
|
|
|
|
|
|
|
|
|
|
|
|
from config.logger import setup_logging
|
|
|
|
|
|
import importlib
|
|
|
|
|
|
|
|
|
|
|
|
logger = setup_logging()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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)
|
|
|
|
|
|
|
|
|
|
|
|
raise ValueError(f"不支持的LLM类型: {class_name}，请检查该配置的type是否设置正确")
|
