Add cohere /v2/chat pass-through cost tracking support

[krrishdholakia]

Title

Ensures pass-through calls for cohere have cost tracking

from langchain_cohere import ChatCohere # pip install langchain-cohere~=0.4.1

base_url = "http://0.0.0.0:4000/" # 👈 PROXY BASE URL
token = "sk-1234"

llm = ChatCohere(
    model="command-r",
    cohere_api_key=token,
    base_url=f"{base_url}/cohere",
)

llm.invoke("hola")

Relevant issues

Type

🆕 New Feature
🐛 Bug Fix
🧹 Refactoring
📖 Documentation
🚄 Infrastructure
✅ Test

Changes

[REQUIRED] Testing - Attach a screenshot of any new tests passing locally

If UI changes, send a screenshot/GIF of working UI fixes

[vercel]

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name	Status	Preview	Comments	Updated (UTC)
litellm	✅ Ready (Inspect)	Visit Preview	💬 Add feedback	Feb 5, 2025 6:09am