From 8d4c1091715263688aaba8bb7fc0d07ddd1bcce4 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Mon, 9 Oct 2023 13:47:06 -0700
Subject: [PATCH] (test) add async + non stream test for ollama

---
 litellm/tests/test_ollama_local.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/litellm/tests/test_ollama_local.py b/litellm/tests/test_ollama_local.py
index 3962287de2..4e58ed59f7 100644
--- a/litellm/tests/test_ollama_local.py
+++ b/litellm/tests/test_ollama_local.py
@@ -135,6 +135,13 @@
 #     response = await ask_question()
 #     async for chunk in response:
 #         print(chunk)
+    
+#     print("test async completion without streaming")
+#     response = await litellm.acompletion(
+#         model="ollama/llama2",
+#         messages=prepare_messages_for_chat("What is litellm? respond in 2 words"),
+#     )
+#     print("response", response)
 
 # if __name__ == "__main__":
 #     import asyncio