add ai image base
This commit is contained in:
34
ai_service/llm/factory.py
Normal file
34
ai_service/llm/factory.py
Normal file
@@ -0,0 +1,34 @@
|
||||
from .adapter.deepseek import DeepSeekAdapter
|
||||
from .adapter.genai import GoogleGenAIAdapter
|
||||
from .adapter.openai import OpenAIAdapter
|
||||
from .adapter.tongyi import TongYiAdapter
|
||||
|
||||
|
||||
def get_adapter(provider, api_key, model, **kwargs):
|
||||
if provider == 'deepseek':
|
||||
return DeepSeekAdapter(api_key, model, **kwargs)
|
||||
elif provider == 'tongyi':
|
||||
return TongYiAdapter(api_key, model, **kwargs)
|
||||
elif provider == 'openai':
|
||||
return OpenAIAdapter(api_key, model, **kwargs)
|
||||
elif provider == 'google-genai':
|
||||
return GoogleGenAIAdapter(api_key, model, **kwargs)
|
||||
else:
|
||||
raise ValueError('不支持的服务商')
|
||||
|
||||
# 使用示例
|
||||
# adapter = get_adapter('tongyi', api_key='xxx', model='wanx_v1')
|
||||
|
||||
# 对话
|
||||
# try:
|
||||
# result = await adapter.chat(messages)
|
||||
# except NotImplementedError:
|
||||
# print("该服务商不支持对话能力")
|
||||
|
||||
# # 图片生成
|
||||
# try:
|
||||
# task = adapter.create_image_task(prompt="一只猫")
|
||||
# status = adapter.fetch_image_task_status(task)
|
||||
# result = adapter.fetch_image_result(task)
|
||||
# except NotImplementedError:
|
||||
# print("该服务商不支持图片生成")
|
||||
Reference in New Issue
Block a user