tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-07 04:24:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

(test) ollama_chat acompletion without stream

Browse Source
This commit is contained in:
ishaan-jaff
2023-12-26 20:01:51 +05:30
parent 3839213d28
commit 0b8d9d177b
1 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_ollama_local_chat.py
+16 -2
View File
@@ -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():