(test) claude fixes

This commit is contained in:
ishaan-jaff 2024-03-04 08:20:26 -08:00
parent d179ae376e
commit ba460e6838
3 changed files with 18 additions and 8 deletions

View file

@ -47,8 +47,9 @@ test_function_call_non_openai_model()
## case 2: add_function_to_prompt set
def test_function_call_non_openai_model_litellm_mod_set():
litellm.add_function_to_prompt = True
litellm.set_verbose = True
try:
model = "claude-instant-1"
model = "claude-instant-1.2"
messages = [{"role": "user", "content": "what's the weather in sf?"}]
functions = [
{