From db90ccd19f249ae721c7bf22247307eae5925e93 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Tue, 26 Dec 2023 20:11:14 +0530 Subject: [PATCH] (test) local ollama_chat --- litellm/tests/test_ollama_local_chat.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/tests/test_ollama_local_chat.py b/litellm/tests/test_ollama_local_chat.py index 897539d4a..cca319428 100644 --- a/litellm/tests/test_ollama_local_chat.py +++ b/litellm/tests/test_ollama_local_chat.py @@ -85,7 +85,7 @@ # print(e) -# asyncio.run(test_async_ollama()) +# # asyncio.run(test_async_ollama()) # def test_completion_ollama(): @@ -130,7 +130,7 @@ # } # ] # response = completion( -# model="ollama/mistral", +# model="ollama_chat/mistral", # messages=messages, # functions=functions, # max_tokens=200, @@ -143,7 +143,7 @@ # pytest.fail(f"Error occurred: {e}") -# # test_completion_ollama_function_calling() +# test_completion_ollama_function_calling() # async def async_test_completion_ollama_function_calling():