admin/django-vue3-admin-gd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修改LLMProvider 类型

Browse Source
This commit is contained in:
XIE7654
2025-08-11 10:26:42 +08:00
parent e21a1ac716
commit a88f272c19
7 changed files with 54 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
ai_service/llm/factory.py
View File

@@ -2,22 +2,23 @@ from .adapter.deepseek import DeepSeekAdapter
from .adapter.genai import GoogleGenAIAdapter
from .adapter.openai import OpenAIAdapter
from .adapter.tongyi import TongYiAdapter
from .enums import LLMProvider
def get_adapter(provider, api_key, model, **kwargs):
if provider == 'deepseek':
def get_adapter(provider: LLMProvider, api_key, model, **kwargs):
if provider == LLMProvider.DEEPSEEK:
return DeepSeekAdapter(api_key, model, **kwargs)
elif provider == 'tongyi':
elif provider == LLMProvider.TONGYI:
return TongYiAdapter(api_key, model, **kwargs)
elif provider == 'openai':
elif provider == LLMProvider.OPENAI:
return OpenAIAdapter(api_key, model, **kwargs)
elif provider == 'google-genai':
elif provider == LLMProvider.GOOGLE_GENAI:
return GoogleGenAIAdapter(api_key, model, **kwargs)
else:
raise ValueError('不支持的服务商')
# 使用示例
# adapter = get_adapter('tongyi', api_key='xxx', model='wanx_v1')
# adapter = get_adapter(LLMProvider.TONGYI, api_key='xxx', model='wanx_v1')
# 对话
# try: