update ollama test file

This commit is contained in:
Mehmet Bektas
2024-05-05 20:25:43 -07:00
parent 3acad270e5
commit 64a64c68c5
+8
View File
@@ -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