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

Update test_proxy_api.py

Browse source
This commit is contained in:
Ishaan Jaff 2023-08-02 20:52:01 -07:00 • committed by GitHub
parent a964e326f1
commit bf872fc14b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

24
litellm/tests/test_proxy_api.py
View file

@ -1,15 +1,15 @@
import sys, os
import traceback
sys.path.insert(0, os.path.abspath('../..')) # Adds the parent directory to the system path
import litellm
from litellm import embedding, completion
# import sys, os
# import traceback
# sys.path.insert(0, os.path.abspath('../..')) # Adds the parent directory to the system path
# import litellm
# from litellm import embedding, completion
litellm.api_base = "https://oai.hconeai.com/v1"
litellm.headers = {"Helicone-Auth": f"Bearer {os.getenv('HELICONE_API_KEY')}"}
# litellm.api_base = "https://oai.hconeai.com/v1"
# litellm.headers = {"Helicone-Auth": f"Bearer {os.getenv('HELICONE_API_KEY')}"}
response = litellm.completion(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": "how does a court case get to the Supreme Court?"}]
)
# response = litellm.completion(
# model="gpt-3.5-turbo",
# messages=[{"role": "user", "content": "how does a court case get to the Supreme Court?"}]
# )
print(response)
# print(response)