chore: updated litellm and langchain-litellm

- Fixed CometAPI map
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2025-10-04 12:07:36 -07:00
parent eafa837320
commit 6ebad8936c
3 changed files with 57 additions and 10 deletions

View file

@ -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(