From 0b8d9d177b3dfcf65b385de28ab6d0c7c024ef82 Mon Sep 17 00:00:00 2001
From: ishaan-jaff <ishaanjaffer0324@gmail.com>
Date: Tue, 26 Dec 2023 20:01:48 +0530
Subject: [PATCH] (test) ollama_chat acompletion without stream

---
 litellm/tests/test_ollama_local_chat.py | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/litellm/tests/test_ollama_local_chat.py b/litellm/tests/test_ollama_local_chat.py
index 08f32b0783..897539d4ae 100644
--- a/litellm/tests/test_ollama_local_chat.py
+++ b/litellm/tests/test_ollama_local_chat.py
@@ -54,7 +54,7 @@
 #         print(e)
 
 
-# test_ollama_streaming()
+# # test_ollama_streaming()
 
 
 # async def test_async_ollama_streaming():
@@ -71,7 +71,21 @@
 #         print(e)
 
 
-# asyncio.run(test_async_ollama_streaming())
+# # asyncio.run(test_async_ollama_streaming())
+
+# async def test_async_ollama():
+#     try:
+#         litellm.set_verbose = True
+#         response = await litellm.acompletion(
+#             model="ollama_chat/llama2",
+#             messages=[{"role": "user", "content": "Hey, how's it going?"}],
+#         )
+#         print("\n response", response)
+#     except Exception as e:
+#         print(e)
+
+
+# asyncio.run(test_async_ollama())
 
 
 # def test_completion_ollama():