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

(feat) API custom callbacks

Browse source
This commit is contained in:
ishaan-jaff 2024-02-15 16:23:05 -08:00
parent 47b8715d25
commit aa333161a8
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
enterprise/callbacks/example_logging_api.py
View file

@ -7,6 +7,7 @@ app = FastAPI()
@app.post("/log-event") @app.post("/log-event")
async def log_event(request: Request): async def log_event(request: Request):
try: try:
print("Received /log-event request")
# Assuming the incoming request has JSON data # Assuming the incoming request has JSON data
data = await request.json() data = await request.json()
print("Received request data:") print("Received request data:")
@ -18,6 +19,9 @@ async def log_event(request: Request):
return {"message": "Request received successfully"} return {"message": "Request received successfully"}
except Exception as e: except Exception as e:
print(f"Error processing request: {str(e)}") print(f"Error processing request: {str(e)}")
import traceback
traceback.print_exc()
raise HTTPException(status_code=500, detail="Internal Server Error") raise HTTPException(status_code=500, detail="Internal Server Error")

5
enterprise/callbacks/generic_api_callback.py
View file

@ -106,15 +106,14 @@ class GenericAPILogger:
import json import json
payload = json.dumps(payload)
data = { data = {
"data": payload, "data": payload,
} }
data = json.dumps(data)
print_verbose(f"\nGeneric Logger - Logging payload = {data}") print_verbose(f"\nGeneric Logger - Logging payload = {data}")
# make request to endpoint with payload # make request to endpoint with payload
response = requests.post(self.endpoint, data=data, headers=self.headers) response = requests.post(self.endpoint, json=data, headers=self.headers)
response_status = response.status_code response_status = response.status_code
response_text = response.text response_text = response.text