From 64a64c68c5d2e511cd1c8db1b7840f09cffca12d Mon Sep 17 00:00:00 2001 From: Mehmet Bektas Date: Sun, 5 May 2024 20:25:43 -0700 Subject: [PATCH] update ollama test file --- litellm/tests/test_ollama_local.py | 8 ++++++++ 1 file changed, 8 insertions(+) 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