fix-openaichat-model-metrics-phi-1233

phidatahq · Sep 19, 2024 · cfcaad3 · cfcaad3
1 parent 814903f
commit cfcaad3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/phi/agent/agent.py b/phi/agent/agent.py
@@ -836,7 +836,10 @@ def _run(
         12. Log Agent Run
         """
         # Create the run_response object
-        run_response = RunResponse(run_id=str(uuid4()), model=self.model.model if self.model is not None else None)
+        run_response = RunResponse(
+            run_id=str(uuid4()),
+            model=self.model.model if self.model is not None else None,
+        )
 
         logger.debug(f"*********** Agent Run Start: {run_response.run_id} ***********")
         # 1. Read existing session from storage
@@ -942,6 +945,9 @@ def _run(
             run_response.content = model_response.content
             run_response.messages = run_messages
 
+        # Add the model metrics to the run_response
+        run_response.metrics = self.model.metrics if self.model else {}
+
         # 9. Update Memory
         # Add the user message to memory
         if message is not None:

diff --git a/phi/playground/playground.py b/phi/playground/playground.py
@@ -96,7 +96,9 @@ def agent_get():
                         ),
                         enable_rag=agent.enable_rag,
                         tools=formatted_tools,
-                        memory={"name": agent.memory.db.__class__.__name__} if agent.memory and agent.memory.db else None,
+                        memory={"name": agent.memory.db.__class__.__name__}
+                        if agent.memory and agent.memory.db
+                        else None,
                         storage={"name": agent.storage.__class__.__name__} if agent.storage else None,
                         knowledge={"name": agent.knowledge.__class__.__name__} if agent.knowledge else None,
                         description=agent.description,