Fix tool use frontend implementation

src/app.py

@@ -47,7 +47,7 @@ def respond_for_chat_interface(message: str, history: list, api_key_input: str =
     """Enhanced response function for gr.ChatInterface with Fed AI Savant capabilities"""
 
     api_key = api_key_input.strip() if api_key_input else os.getenv("FIREWORKS_API_KEY", "")
-    
+
     if not api_key:
         yield "❌ Please enter your Fireworks AI API key in the configuration section above."
         return
@@ -63,30 +63,14 @@ def respond_for_chat_interface(message: str, history: list, api_key_input: str =
                 history=message_with_history
         ):
             if isinstance(messages, list) and len(messages) > 0:
-                formatted_content = ""
-
-                for i, msg in enumerate(messages):
-                    if hasattr(msg, 'content'):
-                        content = msg.content
-
-                        if i > 0:
-                            formatted_content += "\n\n"
-
-                        formatted_content += content
-                    else:
-                        formatted_content += str(msg)
-
-                final_response = formatted_content
-                yield formatted_content
+                yield messages
             else:
-                final_response = str(messages)
                 yield str(messages)
 
     except Exception as e:
         error_msg = f"❌ Error: {str(e)}"
         yield error_msg
 
-
 # Function to create searchable FOMC meetings accordion
 def create_fomc_meetings_accordion():
     """Create searchable accordion for FOMC meetings"""

src/modules/llm_completions.py

@@ -151,68 +151,37 @@ def execute_tool_calls(tool_calls: List[Any], fed_tools: Dict[str, callable]) ->
 
     return results
 
-def create_planning_message_from_response(orchestrator_message: Any) -> ChatMessage:
-    """Create the planning/analysis message from orchestrator response"""
-    content = orchestrator_message.content or "Analyzing Fed data requirements..."
-
-    if orchestrator_message.tool_calls:
-        tool_summary = f"\n\nTools to execute: {len(orchestrator_message.tool_calls)}"
-        for i, tool_call in enumerate(orchestrator_message.tool_calls, 1):
-            tool_summary += f"\n{i}. {tool_call.function.name}"
-        content += tool_summary
-
-    return ChatMessage(
-        role="assistant",
-        content=content,
-        metadata={"title": "🧠 Planning", "status": "done"}
-    )
-
-
-def create_tool_messages_from_calls(tool_calls: List[Any]) -> List[ChatMessage]:
-    """Create initial tool execution messages from tool calls"""
-    tool_messages = []
-
-    for i, tool_call in enumerate(tool_calls):
-        try:
-            parameters = json.loads(tool_call.function.arguments)
-            params_str = ', '.join([f'{k}={v}' for k, v in parameters.items()])
-        except json.JSONDecodeError:
-            params_str = tool_call.function.arguments
-
-        tool_msg = ChatMessage(
-            role="assistant",
-            content=f"Executing: {tool_call.function.name}({params_str})\n\nTool Call ID: {tool_call.id}",
-            metadata={"title": f"🔧 Tool {i + 1}: {tool_call.function.name}", "status": "pending"}
-        )
-        tool_messages.append(tool_msg)
-
-    return tool_messages
 
 def extract_citations_from_tool_results(tool_results: List[Dict[str, Any]]) -> List[Dict[str, str]]:
     """Extract unique citations from tool results"""
     citations = []
-    
+
     for result in tool_results:
         if result["success"] and result["result"].get("success"):
             # Check if result has meeting data with URLs
-            if "meeting" in result["result"]:
-                meeting = result["result"]["meeting"]
-                if meeting.get("url"):
-                    citations.append({
-                        "date": meeting.get("date", "Unknown date"),
-                        "url": meeting["url"],
-                        "title": meeting.get("title", f"FOMC Meeting {meeting.get('date', '')}")
-                    })
+            meeting_data = result["result"].get("meeting")
+            if meeting_data:
+                # Handle both single meeting object and list of meetings
+                meetings_to_process = meeting_data if isinstance(meeting_data, list) else [meeting_data]
+
+                for meeting in meetings_to_process:
+                    if isinstance(meeting, dict) and meeting.get("url"):
+                        citations.append({
+                            "date": meeting.get("date", "Unknown date"),
+                            "url": meeting["url"],
+                            "title": meeting.get("title", f"FOMC Meeting {meeting.get('date', '')}")
+                        })
+
+            # Handle search results
             elif "results" in result["result"]:
-                # Handle search results
                 for meeting in result["result"]["results"]:
-                    if meeting.get("url"):
+                    if isinstance(meeting, dict) and meeting.get("url"):
                         citations.append({
                             "date": meeting.get("date", "Unknown date"),
                             "url": meeting["url"],
                             "title": meeting.get("title", f"FOMC Meeting {meeting.get('date', '')}")
                         })
-    
+
     # Remove duplicate citations
     unique_citations = []
     seen_urls = set()
@@ -220,7 +189,7 @@ def extract_citations_from_tool_results(tool_results: List[Dict[str, Any]]) -> L
         if citation["url"] not in seen_urls:
             unique_citations.append(citation)
             seen_urls.add(citation["url"])
-    
+
     return unique_citations
 
 def format_response_with_citations(response: str, citations: List[Dict[str, str]]) -> str: