update ollama test file

2024-05-05 20:25:43 -07:00 · 2024-05-05 20:25:43 -07:00 · 64a64c68c5
commit 64a64c68c5
parent 3acad270e5
1 changed files with 8 additions and 0 deletions
--- 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