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

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