feat: 修改drawing 为django 接口

This commit is contained in:
XIE7654
2025-10-31 22:16:55 +08:00
parent e4aa6ad18c
commit e5ec2fec56
15 changed files with 401 additions and 34 deletions

35
backend/ai/llm/factory.py Normal file
View File

@@ -0,0 +1,35 @@
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: LLMProvider, api_key, model, **kwargs):
if provider == LLMProvider.DEEPSEEK:
return DeepSeekAdapter(api_key, model, **kwargs)
elif provider == LLMProvider.TONGYI:
return TongYiAdapter(api_key, model, **kwargs)
elif provider == LLMProvider.OPENAI:
return OpenAIAdapter(api_key, model, **kwargs)
elif provider == LLMProvider.GOOGLE_GENAI:
return GoogleGenAIAdapter(api_key, model, **kwargs)
else:
raise ValueError('不支持的服务商')
# 使用示例
# adapter = get_adapter(LLMProvider.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("该服务商不支持图片生成")