bohr-agent-sdk 0.1.118__tar.gz → 0.1.120__tar.gz

{bohr_agent_sdk-0.1.118 → bohr_agent_sdk-0.1.120}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bohr-agent-sdk
-Version: 0.1.118
+Version: 0.1.120
 Summary: SDK for scientific agents
 Home-page: https://github.com/dptech-corp/bohr-agent-sdk/
 Author: DP Technology

{bohr_agent_sdk-0.1.118 → bohr_agent_sdk-0.1.120}/setup.py

@@ -9,7 +9,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="bohr-agent-sdk",
-    version="0.1.118",
+    version="0.1.120",
     description="SDK for science agent and mcp tools",
     long_description=long_description,
     long_description_content_type="text/markdown",

{bohr_agent_sdk-0.1.118 → bohr_agent_sdk-0.1.120}/src/bohr_agent_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bohr-agent-sdk
-Version: 0.1.118
+Version: 0.1.120
 Summary: SDK for scientific agents
 Home-page: https://github.com/dptech-corp/bohr-agent-sdk/
 Author: DP Technology

{bohr_agent_sdk-0.1.118 → bohr_agent_sdk-0.1.120}/src/dp/agent/cli/templates/ui/server/session_manager.py

@@ -632,7 +632,7 @@ class SessionManager:
     ):
         """Process message stream - using ADK native event handling"""
         streaming_text = ""  # Accumulate streaming text
-        
+
         # Run Runner
         async for event in runner.run_async(
             new_message=content,

{bohr_agent_sdk-0.1.118 → bohr_agent_sdk-0.1.120}/src/dp/agent/cli/templates/ui/websocket-server.py

@@ -48,11 +48,13 @@ if __name__ == "__main__":
     
     # uvicorn 始终监听 0.0.0.0 以支持所有配置的主机
     uvicorn.run(
-        app, 
-        host="0.0.0.0", 
+        app,
+        host="0.0.0.0",
         port=port,
         log_level="info",  # 使用 info 级别，过滤掉 warning
         access_log=False,  # 禁用访问日志，减少噪音
+        ws_ping_interval=20,  # Send WebSocket ping every 20s to keep connection alive
+        ws_ping_timeout=30,   # Disconnect if no pong received within 30s
         # 添加自定义的日志配置
         log_config={
             "version": 1,

{bohr_agent_sdk-0.1.118 → bohr_agent_sdk-0.1.120}/src/dp/agent/server/calculation_mcp_server.py

@@ -26,6 +26,7 @@ from .executor import executor_dict
 from .storage import storage_dict
 from .utils import get_logger, JobResult, Tool
 logger = get_logger(__name__)
+CALCULATION_MCP_WORKDIR = os.getenv("CALCULATION_MCP_WORKDIR", os.getcwd())
 
 
 def parse_uri(uri):
@@ -58,6 +59,7 @@ def init_executor(executor_config: Optional[dict] = None):
 @contextmanager
 def set_directory(workdir: str):
     cwd = os.getcwd()
+    workdir = os.path.join(CALCULATION_MCP_WORKDIR, workdir)
     os.makedirs(workdir, exist_ok=True)
     try:
         os.chdir(workdir)
@@ -130,13 +132,15 @@ def handle_input_artifacts(fn, kwargs, storage):
                 kwargs.get(name) is not None):
             uris = kwargs[name]
             new_paths = []
-            for uri in uris:
+            for i, uri in enumerate(uris):
                 scheme, key = parse_uri(uri)
                 if scheme == storage_type:
                     s = storage
                 else:
                     s = storage_dict[scheme]()
-                path = s.download(key, "inputs/%s" % name)
+                dest_dir = Path("inputs") / name / f"item_{i:03d}"
+                dest_dir.mkdir(parents=True, exist_ok=True)
+                path = s.download(key, str(dest_dir))
                 new_paths.append(Path(path))
                 logger.info("Artifact %s downloaded to %s" % (
                     uri, path))
@@ -172,13 +176,15 @@ def handle_input_artifacts(fn, kwargs, storage):
             new_paths_dict = {}
             for key_name, uris in uris_dict.items():
                 new_paths = []
-                for uri in uris:
+                for i, uri in enumerate(uris):
                     scheme, key = parse_uri(uri)
                     if scheme == storage_type:
                         s = storage
                     else:
                         s = storage_dict[scheme]()
-                    path = s.download(key, f"inputs/{name}/{key_name}")
+                    dest_dir = Path("inputs") / name / key_name / f"item_{i:03d}"
+                    dest_dir.mkdir(parents=True, exist_ok=True)
+                    path = s.download(key, str(dest_dir))
                     new_paths.append(Path(path))
                     logger.info("Artifact %s (key=%s) downloaded to %s" % (
                         uri, key_name, path))