diff --git a/litellm/tests/test_ollama_local.py b/litellm/tests/test_ollama_local.py index d4dbc7341b..f5d629140e 100644 --- a/litellm/tests/test_ollama_local.py +++ b/litellm/tests/test_ollama_local.py @@ -24,6 +24,14 @@ # asyncio.run(test_ollama_aembeddings()) +# def test_ollama_embeddings(): +# litellm.set_verbose = True +# input = "The food was delicious and the waiter..." +# response = litellm.embedding(model="ollama/mistral", input=input) +# print(response) + +# test_ollama_embeddings() + # def test_ollama_streaming(): # try: # litellm.set_verbose = False