From 808886dd47210739629a6775081147413f658e9d Mon Sep 17 00:00:00 2001
From: changshan <chgaowei@gmail.com>
Date: Sun, 27 Oct 2024 21:29:04 +0800
Subject: [PATCH] add tool: agenti-connect

---
 .../templates/crewai/tools/agent-connect.py   | 95 +++++++++++++++++++
 agentstack/tools/agent-connect.json           |  6 ++
 agentstack/tools/tools.json                   |  6 +-
 3 files changed, 106 insertions(+), 1 deletion(-)
 create mode 100644 agentstack/templates/crewai/tools/agent-connect.py
 create mode 100644 agentstack/tools/agent-connect.json

diff --git a/agentstack/templates/crewai/tools/agent-connect.py b/agentstack/templates/crewai/tools/agent-connect.py
new file mode 100644
index 00000000..f90a49c6
--- /dev/null
+++ b/agentstack/templates/crewai/tools/agent-connect.py
@@ -0,0 +1,95 @@
+from crewai_tools import tool
+from dotenv import load_dotenv
+import os
+
+from agent_connect.simple_node import SimpleNode
+import json
+
+load_dotenv()
+
+# An HTTP and WS service will be started in agent-connect
+# It can be an IP address or a domain name
+host_domain = os.getenv("HOST_DOMAIN")
+# Host port, default is 80
+host_port = os.getenv("HOST_PORT")
+# WS path, default is /ws
+host_ws_path = os.getenv("HOST_WS_PATH")
+# Path to store DID document
+did_document_path = os.getenv("DID_DOCUMENT_PATH")
+# SSL certificate path, if using HTTPS, certificate and key need to be provided
+ssl_cert_path = os.getenv("SSL_CERT_PATH")
+ssl_key_path = os.getenv("SSL_KEY_PATH")
+
+
+def generate_did_info(node: SimpleNode, did_document_path: str) -> None:
+    """
+    Generate or load DID information for a node.
+    
+    Args:
+        node: SimpleNode instance
+        did_document_path: Path to store/load DID document
+    """
+    if os.path.exists(did_document_path):
+        print(f"Loading existing DID information from {did_document_path}")
+        with open(did_document_path, "r") as f:
+            did_info = json.load(f)
+        node.set_did_info(
+            did_info["private_key_pem"],
+            did_info["did"],
+            did_info["did_document_json"]
+        )
+    else:
+        print("Generating new DID information")
+        private_key_pem, did, did_document_json = node.generate_did_document()
+        node.set_did_info(private_key_pem, did, did_document_json)
+        
+        # Save DID information
+        os.makedirs(os.path.dirname(did_document_path), exist_ok=True)
+        with open(did_document_path, "w") as f:
+            json.dump({
+                "private_key_pem": private_key_pem,
+                "did": did,
+                "did_document_json": did_document_json
+            }, f, indent=2)
+        print(f"DID information saved to {did_document_path}")
+
+agent_connect_simple_node = SimpleNode(host_domain, host_port, host_ws_path)
+generate_did_info(agent_connect_simple_node, did_document_path)
+agent_connect_simple_node.run()
+
+@tool("Send Message to Agent by DID")
+async def send_message(message: str, destination_did: str) -> bool:
+    """
+    Send a message through agent-connect node.
+    
+    Args:
+        message: Message content to be sent
+        destination_did: DID of the recipient agent
+    Returns:
+        bool: True if message was sent successfully, False otherwise
+    """
+    try:
+        await agent_connect_simple_node.send_message(message, destination_did)
+        print(f"Successfully sent message: {message}")
+        return True
+    except Exception as e:
+        print(f"Failed to send message: {e}")
+        return False
+
+@tool("Receive Message from Agent")
+async def receive_message() -> tuple[str, str]:
+    """
+    Receive message from agent-connect node.
+    
+    Returns:
+        tuple[str, str]: Sender DID and received message content, empty string if no message or error occurred
+    """
+    try:
+        sender_did, message = await agent_connect_simple_node.receive_message()
+        if message:
+            print(f"Received message from {sender_did}: {message}")
+            return sender_did, message
+        return "", ""
+    except Exception as e:
+        print(f"Failed to receive message: {e}")
+        return "", ""
diff --git a/agentstack/tools/agent-connect.json b/agentstack/tools/agent-connect.json
new file mode 100644
index 00000000..43c87b49
--- /dev/null
+++ b/agentstack/tools/agent-connect.json
@@ -0,0 +1,6 @@
+{
+    "name": "agent-connect",
+    "package": "poetry add agent-connect",
+    "env": "HOST_DOMAIN=...\nHOST_PORT=\"80\"\nHOST_WS_PATH=\"/ws\"\nDID_DOCUMENT_PATH=...\nSSL_CERT_PATH=...\nSSL_KEY_PATH=...",
+    "tools": ["send_message", "receive_message"]
+}
diff --git a/agentstack/tools/tools.json b/agentstack/tools/tools.json
index aefb217b..ad7642ad 100644
--- a/agentstack/tools/tools.json
+++ b/agentstack/tools/tools.json
@@ -28,5 +28,9 @@
   },{
     "name": "ftp",
     "url": "AgentStack custom tool"
+  }],
+  "network-protocols": [{
+    "name": "agent-connect",
+    "url": "https://github.com/chgaowei/AgentConnect"
   }]
-}
\ No newline at end of file
+}