diff --git a/app/config/ai_configs.py b/app/config/ai_configs.py index d75bc8d..071de55 100644 --- a/app/config/ai_configs.py +++ b/app/config/ai_configs.py @@ -49,6 +49,11 @@ ai_configs = { 'url': 'https://ark.cn-beijing.volces.com/api/v3/chat/completions', 'key': env.llm_key_huoshan, }, + 'doubao-vision-lite': { + 'model': 'doubao-1.5-vision-lite-250315', + 'url': 'https://ark.cn-beijing.volces.com/api/v3/chat/completions', + 'key': env.llm_key_huoshan, + }, 'huoshan-embedding-240715': { "model": 'doubao-embedding-text-240715', diff --git a/app/server.py b/app/server.py index 46ce517..3c764b8 100644 --- a/app/server.py +++ b/app/server.py @@ -89,9 +89,9 @@ add_routes( add_routes( app=app, - runnable=RunnableLambda(lambda x: x['messages']) | create_llm("huoshan-doubao-seed") | StrOutputParser(), + runnable=RunnableLambda(lambda x: x['messages']) | create_llm("doubao-vision-lite") | StrOutputParser(), input_type=ModelInputSchema, - path="/huoshan-doubao-seed" + path="/doubao-vision-lite" ) add_async_route(