scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

(test) ollama_chat acompletion without stream

Browse source
This commit is contained in:
ishaan-jaff 2023-12-26 20:01:48 +05:30
parent 3839213d28
commit 0b8d9d177b
1 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
litellm/tests/test_ollama_local_chat.py
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():