typo

Author: phact

client/examples/chat_completion.py

@@ -51,5 +51,5 @@ def print_chat_completion(model):
 model="anthropic.claude-v2"
 print_chat_completion(model)
 
-model=gemini/gemini-1.5-flash
+model="gemini/gemini-1.5-flash"
 print_chat_completion(model)

client/examples/run.py

@@ -90,6 +90,6 @@ def test_run_with_assistant(assistant, client):
 gemini_assistant = client.beta.assistants.create(
     name="Gemini Animal Tutor",
     instructions=instructions,
-    model=gemini/gemini-1.5-flash,
+    model="gemini/gemini-1.5-flash",
 )
 test_run_with_assistant(gemini_assistant, client)

client/examples/run_retreival.py

@@ -108,7 +108,7 @@ def run_with_assistant(assistant, client):
 )
 run_with_assistant(claude_assistant, client)
 
-model = gemini/gemini-1.5-flash
+model = "gemini/gemini-1.5-flash"
 name = f"{model} Math Tutor"
 
 gemini_assistant = client.beta.assistants.create(

client/tests/astra-assistants/test_chat_completion.py

@@ -98,5 +98,5 @@ def test_chat_completion_claude(patched_openai_client):
 
 @pytest.mark.skip(reason="Tool choice not supporeted / working consistently")
 def test_chat_completion_gemini_pro(patched_openai_client):
-    model=gemini/gemini-1.5-flash
+    model="gemini/gemini-1.5-flash"
     print_chat_completion(model, patched_openai_client)

client/tests/astra-assistants/test_function_calling_v2.py

@@ -38,7 +38,7 @@ def test_function_calling_claude(patched_openai_client):
 
 @pytest.mark.skip(reason="gemini does not consistently work with function calling, skip")
 def test_function_calling_gemini(patched_openai_client):
-    model=gemini/gemini-1.5-flash
+    model="gemini/gemini-1.5-flash"
     function_calling(model, patched_openai_client)
 
 @pytest.mark.skip(reason="llama does not consistently work with function calling, skip")

client/tests/astra-assistants/test_run_retreival_v2.py

@@ -160,7 +160,7 @@ def test_run_claude(patched_openai_client):
     run_with_assistant(claude_assistant, patched_openai_client, file1_path, embedding_model1)
 
 def test_run_gemini(patched_openai_client):
-    model = gemini/gemini-1.5-flash
+    model = "gemini/gemini-1.5-flash"
     name = f"{model} Math Tutor"
 
     gemini_assistant = patched_openai_client.beta.assistants.create(

client/tests/astra-assistants/test_run_v2.py

@@ -89,6 +89,6 @@ def test_run_gemini(patched_openai_client):
     gemini_assistant = patched_openai_client.beta.assistants.create(
         name="Gemini Animal Tutor",
         instructions=instructions,
-        model=gemini/gemini-1.5-flash,
+        model="gemini/gemini-1.5-flash",
     )
     run_with_assistant(gemini_assistant, patched_openai_client)

client/tests/astra-assistants/test_streaming_run_function_calling_v2.py

@@ -36,7 +36,7 @@ def test_function_calling_claude(patched_openai_client):
 
 @pytest.mark.skip(reason="litellm does not use the latest gemini tool support yet and gemini refuses without it, skip")
 def test_function_calling_gemini(patched_openai_client):
-    model=gemini/gemini-1.5-flash
+    model="gemini/gemini-1.5-flash"
     function_calling(model, patched_openai_client)
 
 def test_function_calling_groq_llama3(patched_openai_client):

client/tests/streaming-assistants/test_function_calling_v1.py

@@ -38,7 +38,7 @@ def test_function_calling_claude(streaming_assistants_openai_client):
 
 @pytest.mark.skip(reason="gemini does not consistently work with function calling, skip")
 def test_function_calling_gemini(streaming_assistants_openai_client):
-    model=gemini/gemini-1.5-flash
+    model="gemini/gemini-1.5-flash"
     function_calling(model, streaming_assistants_openai_client)
 
 @pytest.mark.skip(reason="llama does not consistently work with function calling, skip")

client/tests/streaming-assistants/test_run_retreival_v1.py

@@ -109,7 +109,7 @@ def test_run_claude(streaming_assistants_openai_client):
 
 @pytest.mark.skip(reason="flaky")
 def test_run_gemini(streaming_assistants_openai_client):
-    model = gemini/gemini-1.5-flash
+    model = "gemini/gemini-1.5-flash"
     name = f"{model} Math Tutor"
 
     gemini_assistant = streaming_assistants_openai_client.beta.assistants.create(