chore (backend): ensure date strings are converted to datetime before…

… seconds computation
Agenta-AI · Dec 6, 2024 · f6a50dc · f6a50dc
1 parent adc50f3
commit f6a50dc
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/agenta-backend/agenta_backend/services/llm_apps_service.py b/agenta-backend/agenta_backend/services/llm_apps_service.py
@@ -58,15 +58,20 @@ def extract_result_from_response(response: dict):
             if "tree" in response:
                 trace_tree = response.get("tree", {}).get("nodes", [])[0]
 
-                duration_ms = get_nested_value(trace_tree, ["metrics", "acc", "duration", "total"])
+                duration_ms = get_nested_value(
+                    trace_tree, ["metrics", "acc", "duration", "total"]
+                )
                 if duration_ms:
                     duration_seconds = duration_ms / 1000
                 else:
                     start_time = get_nested_value(trace_tree, ["time", "start"])
                     end_time = get_nested_value(trace_tree, ["time", "end"])
 
                     if start_time and end_time:
-                        duration_seconds = (datetime(end_time) - datetime(start_time)).total_seconds()
+                        duration_seconds = (
+                            datetime.fromisoformat(end_time)
+                            - datetime.fromisoformat(start_time)
+                        ).total_seconds()
                     else:
                         duration_seconds = None