Skip to content

Commit 3f5590c

Browse files
committed
Add langtrace instrumentation for cohere
1 parent 3db8912 commit 3f5590c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

example-apps/chatbot-rag-app/api/llm_integrations.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,10 @@ def init_mistral_chat(temperature):
5050

5151

5252
def init_cohere_chat(temperature):
53+
# Cohere is not yet in EDOT. Use the Langtrace Python SDK instead
54+
from langtrace_python_sdk.instrumentation import CohereInstrumentation
55+
56+
CohereInstrumentation().instrument()
5357
return ChatCohere(model=os.getenv("CHAT_MODEL"), temperature=temperature)
5458

5559

0 commit comments

Comments
 (0)