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

call spend logs endpoint

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-30 16:35:07 -07:00
parent c86d1cb391
commit 414d2dcb52
2 changed files with 21 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/example_config_yaml/pass_through_config.yaml
View file

@ -6,4 +6,4 @@ model_list:
api_base: https://exampleopenaiendpoint-production.up.railway.app/
general_settings:
master_key: sk-1234
custom_auth: example_config_yaml.custom_auth.user_api_key_auth
custom_auth: custom_auth_basic.user_api_key_auth

20
tests/pass_through_tests/test_vertex_ai.py
View file

@ -57,6 +57,23 @@ def load_vertex_ai_credentials():
os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = os.path.abspath(temp_file.name)
async def call_spend_logs_endpoint():
"""
Call this
curl -X GET "http://0.0.0.0:4000/spend/logs?start_date={}" -H "Authorization: Bearer sk-1234"
"""
import datetime
import requests
todays_date = datetime.datetime.now().strftime("%Y-%m-%d")
url = f"http://0.0.0.0:4000/spend/logs?start_date={todays_date}"
headers = {"Authorization": f"Bearer sk-1234"}
response = requests.get(url, headers=headers)
print("response from call_spend_logs_endpoint", response)
return response
LITE_LLM_ENDPOINT = "http://localhost:4000"
@ -74,4 +91,7 @@ async def test_basic_vertex_ai_pass_through_with_spendlog():
print("response", response)
_spend_logs_response = await call_spend_logs_endpoint()
print("spend logs response", _spend_logs_response)
pass