From aa333161a87187c7668b0ee1bdb8710e3b657cd0 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 15 Feb 2024 16:23:05 -0800 Subject: [PATCH] (feat) API custom callbacks --- enterprise/callbacks/example_logging_api.py | 4 ++++ enterprise/callbacks/generic_api_callback.py | 5 ++--- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/enterprise/callbacks/example_logging_api.py b/enterprise/callbacks/example_logging_api.py index f3c16299a..57ea99a67 100644 --- a/enterprise/callbacks/example_logging_api.py +++ b/enterprise/callbacks/example_logging_api.py @@ -7,6 +7,7 @@ app = FastAPI() @app.post("/log-event") async def log_event(request: Request): try: + print("Received /log-event request") # Assuming the incoming request has JSON data data = await request.json() print("Received request data:") @@ -18,6 +19,9 @@ async def log_event(request: Request): return {"message": "Request received successfully"} except Exception as e: print(f"Error processing request: {str(e)}") + import traceback + + traceback.print_exc() raise HTTPException(status_code=500, detail="Internal Server Error") diff --git a/enterprise/callbacks/generic_api_callback.py b/enterprise/callbacks/generic_api_callback.py index 309001e1b..076c13d5e 100644 --- a/enterprise/callbacks/generic_api_callback.py +++ b/enterprise/callbacks/generic_api_callback.py @@ -106,15 +106,14 @@ class GenericAPILogger: import json - payload = json.dumps(payload) data = { "data": payload, } - + data = json.dumps(data) print_verbose(f"\nGeneric Logger - Logging payload = {data}") # 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_text = response.text