phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix tests

Browse source
This commit is contained in:
ishaan-jaff 2023-09-21 11:29:41 -07:00
parent b718df5e8f
commit 0c895b76fe
1 changed files with 4 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/tests/test_ollama_local.py
View file

@ -27,7 +27,7 @@
# except Exception as e: # except Exception as e:
# pytest.fail(f"Error occurred: {e}") # pytest.fail(f"Error occurred: {e}")
# test_completion_ollama() # # test_completion_ollama()
# def test_completion_ollama_stream(): # def test_completion_ollama_stream():
# user_message = "what is litellm?" # user_message = "what is litellm?"
@ -47,7 +47,7 @@
# except Exception as e: # except Exception as e:
# pytest.fail(f"Error occurred: {e}") # pytest.fail(f"Error occurred: {e}")
# test_completion_ollama_stream() # # test_completion_ollama_stream()
# async def test_completion_ollama_async_stream(): # async def test_completion_ollama_async_stream():
@ -68,10 +68,8 @@
# except Exception as e: # except Exception as e:
# pytest.fail(f"Error occurred: {e}") # pytest.fail(f"Error occurred: {e}")
# import asyncio # # import asyncio
# asyncio.run(test_completion_ollama_async_stream()) # # asyncio.run(test_completion_ollama_async_stream())
# def prepare_messages_for_chat(text: str) -> list: # def prepare_messages_for_chat(text: str) -> list:
# messages = [ # messages = [