mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-25 00:36:31 +02:00
chore: updated litellm and langchain-litellm
- Fixed CometAPI map
This commit is contained in:
parent
eafa837320
commit
6ebad8936c
3 changed files with 57 additions and 10 deletions
|
|
@ -1,6 +1,6 @@
|
|||
import logging
|
||||
|
||||
from langchain_community.chat_models import ChatLiteLLM
|
||||
from langchain_litellm import ChatLiteLLM
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from sqlalchemy.future import select
|
||||
|
||||
|
|
@ -81,7 +81,7 @@ async def get_user_llm_instance(
|
|||
"MISTRAL": "mistral",
|
||||
"AZURE_OPENAI": "azure",
|
||||
"OPENROUTER": "openrouter",
|
||||
"COMETAPI": "openai",
|
||||
"COMETAPI": "cometapi",
|
||||
# Add more mappings as needed
|
||||
}
|
||||
provider_prefix = provider_map.get(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue