a2a structure agent

README.md

@@ -59,10 +59,11 @@ The script automatically handles port assignments and logging for each server, m
 Populate `agents/MatCreator/.env` with your model and Bohrium credentials (if using `dp` model).
 
 ```env
-LLM_API_KEY="API_KEYS",
-LLM_BASE_URL="API_KEYS",
-BOHRIUM_USERNAME="",
-BOHRIUM_PASSWORD="",
+LLM_MODEL=YOUR_LLM_MODEL
+LLM_API_KEY=YOUR_API_KEYS
+LLM_BASE_URL=YOUR_BASE_URL
+BOHRIUM_USERNAME=""
+BOHRIUM_PASSWORD=""
 BOHRIUM_PROJECT_ID=11111
 ```
 

agents/MatCreator/structure_agent/agent.py

@@ -1,4 +1,6 @@
 from google.adk.agents import Agent
+from google.adk.agents.remote_a2a_agent import RemoteA2aAgent
+from google.adk.agents.remote_a2a_agent import AGENT_CARD_WELL_KNOWN_PATH
 from google.adk.models.lite_llm import LiteLlm
 from google.adk.tools.agent_tool import AgentTool
 from google.adk.tools.mcp_tool import MCPToolset
@@ -27,6 +29,7 @@
 - Inspect structure files to report frame counts, formulas, atom counts,
   cells, periodicity flags, and available per-frame/per-atom properties.
 - Apply entropy-based configuration selection to curate diverse structure sets.
+- Build complex structures by calling the flexible_structure_agent when needed.
 """
 
 instruction = """
@@ -64,9 +67,12 @@
   warnings or anomalies.
 - When chaining tools (e.g., build_bulk_crystal → perturb_atoms), clearly
   indicate which artifact from the previous step you are using.
+- If the user request contains complex structure manipulations beyond your
+  tools' capabilities, recognize the limitation gracefully and, transfer to the
+  flexible_structure_agent with a brief explanation.
 
 Response format
-- Plan: 1–3 bullets describing the immediate next step(s).
+- Plan: 1-3 bullets describing the immediate next step(s).
 - Action: the exact tool you will call.
 - Result: brief summary with key outputs and absolute paths.
 - Next: the immediate follow-up or final recap. Transfer back to parent agent if done.
@@ -80,6 +86,14 @@
     )
 )
 
+remote_a2a_url = "http://localhost:8001"
+flexible_structure_agent = RemoteA2aAgent(
+    name='flexible_structure_agent',
+    description="A structure agent that can handle very complex structure-related tasks. " \
+    "This is a remote agent, so always remember to transfer the outputs back to the parent agent after done.",
+    agent_card=f"{remote_a2a_url}/{AGENT_CARD_WELL_KNOWN_PATH}"
+)
+
 structure_agent = Agent(
     name='structure_agent',
     model=LiteLlm(
@@ -94,5 +108,6 @@
     tools=[
         toolset,
     ],
-    after_tool_callback=after_tool_callback
+    after_tool_callback=after_tool_callback,
+    sub_agents=[flexible_structure_agent],  # Enable transfer to flexible_structure_agent
 )

tools/quest/server.py

@@ -1130,6 +1130,45 @@ def inspect_structure(
 	)
 
 
+## ================================
+## Tools for file transfer between local and remote structure agents
+## ===============================
+
+import requests
+
+@mcp.tool()
+def get_remote_files(file_urls: list[str], local_save_path: str="downloads") -> list[dict]:
+    """
+    Downloads files from the remote agent to local storage.
+    Args:
+        file_urls: List of full URLs of files to download from the remote agent.
+        local_save_path: Local directory path to save the downloaded files.
+    """
+    os.makedirs(local_save_path, exist_ok=True)
+    results: list[dict] = []
+    for file_url in file_urls:
+        # Extract filename from URL (the last part)
+        file_name = file_url.split('/')[-1]
+        local_file_path = os.path.join(local_save_path, file_name)
+
+        # get the file contents from the remote URL
+        response = requests.get(file_url)
+        response.raise_for_status()
+
+        # Write the raw content to file
+        with open(local_file_path, 'wb') as f:
+            f.write(response.content)
+
+        results.append({
+            "result": "File downloaded successfully",
+            "file_name": file_name,
+            "local_path": local_file_path
+        })
+
+    return results
+
+
+
 if __name__ == "__main__":
 	create_workpath()
 	mcp.run(transport=args.transport)