phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test: commenting out local test

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-02 20:53:24 -08:00
parent bb4f82066a
commit 3b70a6864f
1 changed files with 26 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

52
litellm/tests/test_ollama_local.py
View file

@ -1,35 +1,35 @@
##### THESE TESTS CAN ONLY RUN LOCALLY WITH THE OLLAMA SERVER RUNNING ######
# https://ollama.ai/
# ##### THESE TESTS CAN ONLY RUN LOCALLY WITH THE OLLAMA SERVER RUNNING ######
# # https://ollama.ai/
import sys, os
import traceback
from dotenv import load_dotenv
load_dotenv()
import os
sys.path.insert(0, os.path.abspath('../..')) # Adds the parent directory to the system path
import pytest
import litellm
from litellm import embedding, completion
import asyncio
# import sys, os
# import traceback
# from dotenv import load_dotenv
# load_dotenv()
# import os
# sys.path.insert(0, os.path.abspath('../..')) # Adds the parent directory to the system path
# import pytest
# import litellm
# from litellm import embedding, completion
# import asyncio
user_message = "respond in 20 words. who are you?"
messages = [{ "content": user_message,"role": "user"}]
# user_message = "respond in 20 words. who are you?"
# messages = [{ "content": user_message,"role": "user"}]
def test_completion_ollama():
try:
response = completion(
model="ollama/llama2",
messages=messages,
max_tokens=200,
request_timeout = 10,
# def test_completion_ollama():
# try:
# response = completion(
# model="ollama/llama2",
# messages=messages,
# max_tokens=200,
# request_timeout = 10,
)
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# )
# print(response)
# except Exception as e:
# pytest.fail(f"Error occurred: {e}")
test_completion_ollama()
# test_completion_ollama()
# def test_completion_ollama_with_api_base():
# try: