Update app.py

app.py

@@ -11,6 +11,54 @@ from openfloor import (
 from openfloor.manifest import *
 from openfloor.envelope import *
 
+# Function to build the envelope for a manifest request using OpenFloor library
+def build_manifest_request_envelope(agent_url):
+    # Create conversation and sender
+    conversation = Conversation()
+    sender = Sender(
+        speakerUri="openfloor://localhost/TestClient",
+        serviceUrl="http://localhost"
+    )
+    
+    # Create envelope
+    envelope = Envelope(conversation=conversation, sender=sender)
+    
+    # Add get manifests event
+    manifest_event = GetManifestsEvent(
+        to=To(serviceUrl=agent_url, private=False)
+    )
+    envelope.events.append(manifest_event)
+    
+    return envelope
+
+# Function to build an utterance envelope using OpenFloor library
+def build_utterance_envelope(agent_url, message_text):
+    # Create conversation and sender
+    conversation = Conversation()
+    sender = Sender(
+        speakerUri="openfloor://localhost/TestClient",
+        serviceUrl="http://localhost"
+    )
+    
+    # Create envelope
+    envelope = Envelope(conversation=conversation, sender=sender)
+    
+    # Create dialog event with text feature
+    dialog_event = DialogEvent(
+        speakerUri="openfloor://localhost/TestClient",
+        features={"text": TextFeature(values=[message_text])}
+    )
+    
+    # Create utterance event
+    utterance_event = UtteranceEvent(
+        dialogEvent=dialog_event,
+        to=To(serviceUrl=agent_url, private=False)
+    )
+    
+    envelope.events.append(utterance_event)
+    
+    return envelope
+
 class OpenFloorAgent:
     def __init__(self, url: str, name: str = None):
         self.url = url
@@ -21,22 +69,14 @@ class OpenFloorAgent:
     def get_manifest(self):
         """Get the agent's manifest and capabilities"""
         try:
-            conversation = Conversation()
-            sender = Sender(
-                speakerUri="openfloor://mcp-bridge/client",
-                serviceUrl="http://mcp-bridge"
-            )
-            envelope = Envelope(conversation=conversation, sender=sender)
-            
-            manifest_event = GetManifestsEvent(
-                to=To(serviceUrl=self.url, private=False)
-            )
-            envelope.events.append(manifest_event)
+            envelope = build_manifest_request_envelope(self.url)
             
+            # Convert to JSON using the library's method
             json_str = envelope.to_json()
             envelope_dict = json.loads(json_str)
             payload = {"openFloor": envelope_dict}
             
+            # Send request
             headers = {'Content-Type': 'application/json'}
             response = requests.post(self.url, json=payload, headers=headers, timeout=10)
             
@@ -58,29 +98,14 @@ class OpenFloorAgent:
     def send_utterance(self, message: str) -> str:
         """Send an utterance to the agent and get response"""
         try:
-            conversation = Conversation()
-            sender = Sender(
-                speakerUri="openfloor://mcp-bridge/client",
-                serviceUrl="http://mcp-bridge"
-            )
-            envelope = Envelope(conversation=conversation, sender=sender)
-            
-            dialog_event = DialogEvent(
-                speakerUri="openfloor://mcp-bridge/client",
-                features={"text": TextFeature(values=[message])}
-            )
-            
-            utterance_event = UtteranceEvent(
-                dialogEvent=dialog_event,
-                to=To(serviceUrl=self.url, private=False)
-            )
-            
-            envelope.events.append(utterance_event)
+            envelope = build_utterance_envelope(self.url, message)
             
+            # Convert to JSON using the library's method
             json_str = envelope.to_json()
             envelope_dict = json.loads(json_str)
             payload = {"openFloor": envelope_dict}
             
+            # Send request
             headers = {'Content-Type': 'application/json'}
             response = requests.post(self.url, json=payload, headers=headers, timeout=30)