fix None openaiMetadta

bdqfork · Feb 4, 2024 · 0428d89 · 0428d89
1 parent 96406aa
commit 0428d89
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/libs/superagent/app/api/agents.py b/libs/superagent/app/api/agents.py
@@ -222,6 +222,8 @@ async def delete(agent_id: str, api_user=Depends(get_current_api_user)):
         if SEGMENT_WRITE_KEY:
             analytics.track(api_user.id, "Deleted Agent")
         deleted = await prisma.agent.delete(where={"id": agent_id})
+        if deleted is None or deleted.openaiMetadata is None:
+            return {"success": True, "data": None}
         openaiMetadata = json.loads(deleted.openaiMetadata)
         if openaiMetadata.get("id", None):
             llm = await prisma.llm.find_first_or_raise(
@@ -254,6 +256,8 @@ async def update(
                 "apiUserId": api_user.id,
             },
         )
+        if data.openaiMetadata is None:
+            return {"success": True, "data": data}
         openaiMetadata = json.loads(data.openaiMetadata)
         if openaiMetadata:
             llm = await prisma.llm.find_first_or_raise(
@@ -427,7 +431,8 @@ def track_agent_invocation(result):
             },
         )
 
-    costCallback = CostCalcAsyncHandler(model=LLM_MAPPING[agent_config.llmModel])
+    costCallback = CostCalcAsyncHandler(
+        model=LLM_MAPPING[agent_config.llmModel])
 
     agentCallbacks = [costCallback]
 
@@ -481,7 +486,8 @@ async def send_message(
             logging.error(f"Error in send_message: {error}")
             if SEGMENT_WRITE_KEY:
                 try:
-                    track_agent_invocation({"error": str(error), "status_code": 500})
+                    track_agent_invocation(
+                        {"error": str(error), "status_code": 500})
                 except Exception as e:
                     logging.error(f"Error tracking agent invocation: {e}")
             yield ("event: error\n" f"data: {error}\n\n")