bohr-agent-sdk 0.1.23__tar.gz → 0.1.100__tar.gz

{bohr_agent_sdk-0.1.23 → bohr_agent_sdk-0.1.100}/PKG-INFO

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

{bohr_agent_sdk-0.1.23 → bohr_agent_sdk-0.1.100}/setup.py

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

{bohr_agent_sdk-0.1.23 → bohr_agent_sdk-0.1.100}/src/bohr_agent_sdk.egg-info/PKG-INFO

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

{bohr_agent_sdk-0.1.23 → bohr_agent_sdk-0.1.100}/src/dp/agent/server/calculation_mcp_server.py

@@ -11,6 +11,8 @@ from typing import Literal, Optional, get_origin
 from mcp.server.fastmcp import FastMCP
 from mcp.server.fastmcp.server import Context
 from mcp.server.fastmcp.utilities.func_metadata import _get_typed_signature
+from starlette.responses import JSONResponse
+from starlette.routing import Route
 
 from .executor import executor_dict
 from .storage import storage_dict
@@ -282,4 +284,16 @@ class CalculationMCPServer:
     def run(self, **kwargs):
         if os.environ.get("DP_AGENT_RUNNING_MODE") in ["1", "true"]:
             return
+        async def health_check(request) :
+            return JSONResponse({"status": "ok"})
+        
+        self.mcp._custom_starlette_routes.append(
+            Route(
+                "/health",
+                endpoint=health_check,
+                methods=["GET"],
+                name="health_check",
+                include_in_schema=True,
+            )
+        )
         self.mcp.run(**kwargs)

{bohr_agent_sdk-0.1.23 → bohr_agent_sdk-0.1.100}/src/dp/agent/server/executor/local_executor.py

@@ -130,7 +130,12 @@ class LocalExecutor(BaseExecutor):
                         extra_info["workflow_link"] = wf_link
                         break
                 if not alive:
-                    break
+                    if os.path.isfile("err"):
+                        with open("err", "r") as f:
+                            err_msg = f.read()
+                    else:
+                        err_msg = "No workflow submitted"
+                    raise RuntimeError(err_msg)
                 logger.info("Waiting workflow to be submitted")
                 time.sleep(1)
         self.recover_env(old_env)