Update app.py

app.py

@@ -65,6 +65,30 @@ class PersonalAIResearchAssistant:
         with open(self.knowledge_base_path, 'w') as f:
             json.dump(kb, f, indent=2)
     
+    def _extract_tool_info(self, response) -> Dict:
+        """
+        Extract tool usage information in a JSON serializable format
+        """
+        tool_info = None
+        if hasattr(response.choices[0].message, 'executed_tools'):
+            # Convert ExecutedTool objects to dictionaries
+            tools = response.choices[0].message.executed_tools
+            if tools:
+                tool_info = []
+                for tool in tools:
+                    # Extract only serializable data
+                    tool_dict = {
+                        "tool_type": getattr(tool, "type", "unknown"),
+                        "tool_name": getattr(tool, "name", "unknown"),
+                    }
+                    # Add any other relevant attributes in a serializable form
+                    if hasattr(tool, "input"):
+                        tool_dict["input"] = str(tool.input)
+                    if hasattr(tool, "output"):
+                        tool_dict["output"] = str(tool.output)
+                    tool_info.append(tool_dict)
+        return tool_info
+    
     def research_digest(self, topic: str, 
                         include_domains: List[str] = None,
                         exclude_domains: List[str] = None,
@@ -113,10 +137,8 @@ class PersonalAIResearchAssistant:
         response = self.client.chat.completions.create(**params)
         content = response.choices[0].message.content
         
-        # Extract tool usage information if available
-        tool_info = None
-        if hasattr(response.choices[0].message, 'executed_tools'):
-            tool_info = response.choices[0].message.executed_tools
+        # Extract tool usage information in a serializable format
+        tool_info = self._extract_tool_info(response)
         
         # Create digest entry
         digest = {
@@ -190,10 +212,8 @@ class PersonalAIResearchAssistant:
         
         content = response.choices[0].message.content
         
-        # Extract tool usage information if available
-        tool_info = None
-        if hasattr(response.choices[0].message, 'executed_tools'):
-            tool_info = response.choices[0].message.executed_tools
+        # Extract tool usage information in a serializable format
+        tool_info = self._extract_tool_info(response)
         
         # Create code analysis entry
         analysis = {
@@ -249,10 +269,8 @@ class PersonalAIResearchAssistant:
         
         content = response.choices[0].message.content
         
-        # Extract tool usage information if available
-        tool_info = None
-        if hasattr(response.choices[0].message, 'executed_tools'):
-            tool_info = response.choices[0].message.executed_tools
+        # Extract tool usage information in a serializable format
+        tool_info = self._extract_tool_info(response)
         
         # Create connection entry
         connection = {
@@ -491,7 +509,7 @@ def research_topic(topic, include_domains, exclude_domains):
         
         # Add tool usage info if available
         if result.get("tool_usage"):
-            response += f"\n\n*Tool Usage: {result['tool_usage']}*"
+            response += f"\n\n*Tool Usage Information Available*"
             
         return response
     except Exception as e:
@@ -520,7 +538,7 @@ def analyze_code(code_snippet, language, analysis_type):
         
         # Add tool usage info if available
         if result.get("tool_usage"):
-            response += f"\n\n*Tool Usage: {result['tool_usage']}*"
+            response += f"\n\n*Tool Usage Information Available*"
             
         return response
     except Exception as e:
@@ -548,7 +566,7 @@ def connect_concepts_handler(concept_a, concept_b):
         
         # Add tool usage info if available
         if result.get("tool_usage"):
-            response += f"\n\n*Tool Usage: {result['tool_usage']}*"
+            response += f"\n\n*Tool Usage Information Available*"
             
         return response
     except Exception as e: