also add error messages to the apicalls table

matthewlouisbrockman · matthewlouisbrockman · commit f6f666dd91a2 · 2023-04-15T18:07:32.000-07:00
diff --git a/api/app/api/utils/caesar_logging.py b/api/app/api/utils/caesar_logging.py
@@ -3,7 +3,7 @@
 
 from app.config import EVENTS_ENGINE
 
-def log_apicall(duration, provider, model, input_tokens, output_tokens, service, purpose, session_id=None, success=True):
+def log_apicall(duration, provider, model, input_tokens, output_tokens, service, purpose, session_id=None, success=True, log_message=None):
     if not EVENTS_ENGINE:
         return {"status": "no engine"}
     cost = calculate_cost(model, input_tokens, output_tokens)
@@ -19,11 +19,12 @@ def log_apicall(duration, provider, model, input_tokens, output_tokens, service,
         "cost": cost,
         "success": success,
         "session_id": session_id,
+        "log_message": log_message,
     }
 
     insert_query = text("""
-        INSERT INTO apicalls (duration, provider, model, input_tokens, output_tokens, service, purpose, cost, success, session_id)
-        VALUES (:duration, :provider, :model, :input_tokens, :output_tokens, :service, :purpose, :cost, :success, :session_id)
+        INSERT INTO apicalls (duration, provider, model, input_tokens, output_tokens, service, purpose, cost, success, session_id, log_message)
+        VALUES (:duration, :provider, :model, :input_tokens, :output_tokens, :service, :purpose, :cost, :success, :session_id, :log_message)
     """)
 
     with EVENTS_ENGINE.connect() as conn:
diff --git a/api/app/api/utils/messages.py b/api/app/api/utils/messages.py
@@ -42,6 +42,7 @@ def get_assistant_message_from_openai(
             purpose,
             session_id = session_id,
             success=False,
+            log_message = str(e),
         )
         raise e
     duration = time.time() - start
@@ -95,6 +96,7 @@ def call_chat(
             purpose,
             session_id = session_id,
             success=False,
+            log_message = str(e),
         )
         raise e
 

Original file line number	Diff line number	Diff line change
`@@ -42,6 +42,7 @@ def get_assistant_message_from_openai(`
`42`	`42`	`purpose,`
`43`	`43`	`session_id = session_id,`
`44`	`44`	`success=False,`
	`45`	`+ log_message = str(e),`
`45`	`46`	`)`
`46`	`47`	`raise e`
`47`	`48`	`duration = time.time() - start`
`@@ -95,6 +96,7 @@ def call_chat(`
`95`	`96`	`purpose,`
`96`	`97`	`session_id = session_id,`
`97`	`98`	`success=False,`
	`99`	`+ log_message = str(e),`
`98`	`100`	`)`
`99`	`101`	`raise e`
`100`	`102`