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

feat(google_ai_studio_endpoints.py): support pass-through endpoint for all google ai studio requests

Browse source 
New Feature
This commit is contained in:
Krrish Dholakia 2024-08-17 10:46:59 -07:00
parent 668ea6cbc7
commit 29bedae79f
6 changed files with 186 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/litellm_core_utils/litellm_logging.py
View file

@ -274,6 +274,7 @@ class Logging:
headers = {}
data = additional_args.get("complete_input_dict", {})
api_base = str(additional_args.get("api_base", ""))
query_params = additional_args.get("query_params", {})
if "key=" in api_base:
# Find the position of "key=" in the string
key_index = api_base.find("key=") + 4
@ -2362,7 +2363,7 @@ def get_standard_logging_object_payload(
return payload
except Exception as e:
verbose_logger.error(
verbose_logger.exception(
"Error creating standard logging object - {}".format(str(e))
)
return None