scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

fix type errors

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-01 15:02:29 -07:00 • committed by Krrish Dholakia
parent 7e79adc614
commit 520bad0fdb
1 changed files with 15 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
litellm/integrations/gcs_bucket.py
View file

@ -97,20 +97,26 @@ class GCSBucketLogger(CustomLogger):
get_logging_payload,
)
spend_logs_payload: GCSBucketPayload = get_logging_payload(
spend_logs_payload: SpendLogsPayload = get_logging_payload(
kwargs=kwargs,
response_obj=response_obj,
start_time=start_time,
end_time=end_time,
end_user_id=kwargs.get("user"),
)
spend_logs_payload["startTime"] = start_time.isoformat()
spend_logs_payload["endTime"] = end_time.isoformat()
spend_logs_payload["completionStartTime"] = spend_logs_payload[
"completionStartTime"
].isoformat()
object_name = spend_logs_payload["request_id"]
gcs_payload: GCSBucketPayload = GCSBucketPayload(
**spend_logs_payload, messages=None, output=None
)
gcs_payload["messages"] = kwargs.get("messages", None)
gcs_payload["startTime"] = start_time.isoformat()
gcs_payload["endTime"] = end_time.isoformat()
if gcs_payload["completionStartTime"] is not None:
gcs_payload["completionStartTime"] = gcs_payload[ # type: ignore
"completionStartTime" # type: ignore
].isoformat()
output = None
if response_obj is not None and (
kwargs.get("call_type", None) == "embedding"
@ -140,8 +146,8 @@ class GCSBucketLogger(CustomLogger):
):
output = response_obj["text"]
spend_logs_payload["output"] = output
return spend_logs_payload
gcs_payload["output"] = output
return gcs_payload
async def download_gcs_object(self, object_name):
"""