phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

feat(cohere/chat.py): return citations in model response

Browse source 
Closes https://github.com/BerriAI/litellm/issues/6814
This commit is contained in:
Krrish Dholakia 2024-11-30 13:59:57 -08:00
parent bd59f18809
commit 2fbc71a62c
6 changed files with 310 additions and 254 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/local_testing/test_function_calling.py
View file

@ -46,11 +46,12 @@ def get_current_weather(location, unit="fahrenheit"):
"model",
[
"gpt-3.5-turbo-1106",
# "mistral/mistral-large-latest",
"mistral/mistral-large-latest",
"claude-3-haiku-20240307",
"gemini/gemini-1.5-pro",
"anthropic.claude-3-sonnet-20240229-v1:0",
# "groq/llama3-8b-8192",
"groq/llama3-8b-8192",
"cohere_chat/command-r",
],
)
@pytest.mark.flaky(retries=3, delay=1)