Update app.py

app.py

@@ -540,7 +540,7 @@ class VisualConsensusEngine:
 
 
     def _execute_research_function(self, function_name: str, arguments: dict, requesting_model_name: str = None) -> str:
-        """Execute research function with REAL-TIME visual feedback and progress indicators"""
+        """Execute research function using proper OpenFloor protocol messaging"""
         
         query_param = arguments.get("query") or arguments.get("topic") or arguments.get("technology") or arguments.get("company")
         
@@ -549,74 +549,106 @@ class VisualConsensusEngine:
             self.show_research_starting(function_name, query_param)
         
         try:
-            # Actually execute the research with detailed progress indicators
+            # Map function names to research agents
+            function_to_agent = {
+                "search_web": "web_search",
+                "search_wikipedia": "wikipedia", 
+                "search_academic": "arxiv",
+                "search_technology_trends": "github",
+                "search_financial_data": "sec_edgar"
+            }
+            
             result = ""
             
-            if function_name == "search_web":
-                self.update_research_progress("Initializing web search engines...")
-                depth = arguments.get("depth", "standard")
+            if function_name in function_to_agent:
+                agent_name = function_to_agent[function_name]
+                research_agent = self.research_agents[agent_name]
                 
-                if depth == "deep":
-                    self.update_research_progress("Performing deep web search (multiple sources)...")
-                else:
-                    self.update_research_progress("Searching web databases...")
+                self.update_research_progress(f"Sending OpenFloor research request...")
                 
-                result = self.search_agent.search(arguments["query"], depth)
-                self.update_research_progress(f"Web search complete - found {len(result)} characters of data")
+                # Create OpenFloor conversation and envelope
+                conversation = Conversation()
                 
-            elif function_name == "search_wikipedia":
-                self.update_research_progress("Connecting to Wikipedia API...")
-                self.update_research_progress("Searching Wikipedia articles...")
-                result = self.search_agent.search_wikipedia(arguments["topic"])
-                self.update_research_progress(f"Wikipedia search complete - found {len(result)} characters")
+                # Create research request utterance
+                request_dialog = DialogEvent(
+                    speakerUri=f"tag:consilium.ai,2025:{requesting_model_name or 'expert'}",
+                    features={
+                        "text": TextFeature(values=[query_param])
+                    }
+                )
                 
-            elif function_name == "search_academic":
-                self.update_research_progress("Connecting to arXiv preprint server...")
-                self.update_research_progress("Searching academic papers on arXiv...")
-                result = self.search_agent.tools['arxiv'].search(arguments["query"])
-                self.update_research_progress(f"arXiv search complete - found {len(result)} characters")
-                    
-            elif function_name == "search_technology_trends":
-                self.update_research_progress("Connecting to GitHub API...")
-                self.update_research_progress("Analyzing technology trends and repository data...")
-                result = self.search_agent.tools['github'].search(arguments["technology"])
-                self.update_research_progress(f"Technology trends analysis complete - found {len(result)} characters")
+                # Create request envelope using OpenFloor protocol
+                request_envelope = Envelope(
+                    conversation=conversation,
+                    sender=Sender(speakerUri=f"tag:consilium.ai,2025:{requesting_model_name or 'expert'}"),
+                    events=[
+                        UtteranceEvent(
+                            dialogEvent=request_dialog,
+                            to=To(speakerUri=research_agent.manifest.identification.speakerUri)
+                        )
+                    ]
+                )
                 
-            elif function_name == "search_financial_data":
-                self.update_research_progress("Connecting to SEC EDGAR database...")
-                self.update_research_progress("Searching financial filings and reports...")
-                result = self.search_agent.tools['sec'].search(arguments["company"])
-                self.update_research_progress(f"Financial data search complete - found {len(result)} characters")
+                self.update_research_progress(f"Processing research via OpenFloor protocol...")
+                
+                # Send request to research agent using OpenFloor protocol
+                response_envelope = research_agent.handle_utterance_event(request_envelope)
+                
+                # Extract result from OpenFloor response
+                result = self._extract_research_result_from_envelope(response_envelope)
+                
+                self.update_research_progress(f"OpenFloor research complete - found {len(result)} characters")
                 
             elif function_name == "multi_source_research":
-                self.update_research_progress("Initializing multi-source deep research...")
-                self.update_research_progress("Phase 1: Web search...")
+                self.update_research_progress("Initializing multi-source OpenFloor research...")
                 
-                # Show progress for each source in deep research
-                result = ""
-                try:
-                    # Simulate the deep research process with progress updates
-                    self.update_research_progress("Phase 1: Comprehensive web search...")
-                    web_result = self.search_agent.search(arguments["query"], "standard")
-                    self.update_research_progress(f"Web search complete ({len(web_result)} chars) - Phase 2: Academic sources...")
-                    
-                    self.update_research_progress("Phase 2: Searching academic databases...")
-                    # Add small delay to show progress
-                    time.sleep(1)
-                    
-                    self.update_research_progress("Phase 3: Analyzing and synthesizing results...")
-                    result = self.search_agent.search(arguments["query"], "deep")
-                    self.update_research_progress(f"Multi-source research complete - synthesized {len(result)} characters")
-                    
-                except Exception as e:
-                    self.update_research_progress(f"Multi-source research error: {str(e)}")
-                    result = f"Multi-source research encountered an error: {str(e)}"
+                # Use multiple research agents via OpenFloor protocol
+                results = []
+                agents_to_use = ["web_search", "wikipedia", "arxiv"]
                 
+                for i, agent_name in enumerate(agents_to_use, 1):
+                    self.update_research_progress(f"Phase {i}: OpenFloor request to {agent_name.replace('_', ' ').title()}...")
+                    try:
+                        research_agent = self.research_agents[agent_name]
+                        
+                        # Create OpenFloor request for this agent
+                        conversation = Conversation()
+                        request_dialog = DialogEvent(
+                            speakerUri=f"tag:consilium.ai,2025:{requesting_model_name or 'expert'}",
+                            features={"text": TextFeature(values=[query_param])}
+                        )
+                        
+                        request_envelope = Envelope(
+                            conversation=conversation,
+                            sender=Sender(speakerUri=f"tag:consilium.ai,2025:{requesting_model_name or 'expert'}"),
+                            events=[UtteranceEvent(
+                                dialogEvent=request_dialog,
+                                to=To(speakerUri=research_agent.manifest.identification.speakerUri)
+                            )]
+                        )
+                        
+                        # Get OpenFloor response
+                        response_envelope = research_agent.handle_utterance_event(request_envelope)
+                        agent_result = self._extract_research_result_from_envelope(response_envelope)
+                        
+                        if agent_result and len(agent_result) > 100:
+                            results.append(f"**{agent_name.replace('_', ' ').title()} Results:**\n{agent_result}")
+                            
+                    except Exception as e:
+                        self.update_research_progress(f"Phase {i}: {agent_name} OpenFloor error - {str(e)[:50]}...")
+                        continue
+                
+                # Combine results
+                if results:
+                    result = f"**Multi-Source OpenFloor Research for: {query_param}**\n\n" + "\n\n---\n\n".join(results)
+                    self.update_research_progress(f"Multi-source OpenFloor research complete - synthesized {len(result)} characters")
+                else:
+                    result = f"Multi-source OpenFloor research for '{query_param}' encountered errors across all agents."
+                    
             else:
-                self.update_research_progress(f"Unknown research function: {function_name}")
                 result = f"Unknown research function: {function_name}"
             
-            # Phase 3: Show research ACTUALLY complete (after execution)
+            # Phase 3: Show research complete
             if query_param:
                 self.show_research_complete(function_name, query_param, len(result), requesting_model_name)
                 
@@ -626,7 +658,23 @@ class VisualConsensusEngine:
             error_msg = str(e)
             if query_param:
                 self.show_research_error(function_name, query_param, error_msg, requesting_model_name)
-            return f"Research function error: {error_msg}"
+            return f"OpenFloor research error: {error_msg}"
+    
+    def _extract_research_result_from_envelope(self, envelope: Envelope) -> str:
+        """Extract research result from OpenFloor response envelope"""
+        try:
+            for event in envelope.events:
+                if hasattr(event, 'eventType') and event.eventType == 'utterance':
+                    dialog_event = event.parameters.get('dialogEvent')
+                    if dialog_event and hasattr(dialog_event, 'features'):
+                        text_feature = dialog_event.features.get('text')
+                        if text_feature and hasattr(text_feature, 'tokens'):
+                            return ' '.join([token.get('value', '') for token in text_feature.tokens])
+            
+            return "No research result found in OpenFloor response"
+            
+        except Exception as e:
+            return f"Error extracting OpenFloor research result: {str(e)}"
     
     def show_research_starting(self, function: str, query: str):
         """Show research request initiation with enhanced messaging"""