bohr-agent-sdk 0.1.119__tar.gz → 0.1.121__tar.gz

{bohr_agent_sdk-0.1.119 → bohr_agent_sdk-0.1.121}/PKG-INFO

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

{bohr_agent_sdk-0.1.119 → bohr_agent_sdk-0.1.121}/setup.py

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

{bohr_agent_sdk-0.1.119 → bohr_agent_sdk-0.1.121}/src/bohr_agent_sdk.egg-info/PKG-INFO

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

{bohr_agent_sdk-0.1.119 → bohr_agent_sdk-0.1.121}/src/dp/agent/server/calculation_mcp_server.py

@@ -59,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)
@@ -82,8 +83,7 @@ def query_job_status(job_id: str, executor: Optional[dict] = None
         status (str): One of "Running", "Succeeded" or "Failed"
     """
     trace_id, exec_id = job_id.split("/")
-    workdir = os.path.join(CALCULATION_MCP_WORKDIR, trace_id)
-    with set_directory(workdir):
+    with set_directory(trace_id):
         executor = load_job_info()["executor"] or executor
         _, executor = init_executor(executor)
         status = executor.query_status(exec_id)
@@ -98,8 +98,7 @@ def terminate_job(job_id: str, executor: Optional[dict] = None):
         job_id (str): The ID of the calculation job
     """
     trace_id, exec_id = job_id.split("/")
-    workdir = os.path.join(CALCULATION_MCP_WORKDIR, trace_id)
-    with set_directory(workdir):
+    with set_directory(trace_id):
         executor = load_job_info()["executor"] or executor
         _, executor = init_executor(executor)
         executor.terminate(exec_id)
@@ -244,8 +243,7 @@ def get_job_results(job_id: str, executor: Optional[dict] = None,
         results (Any): results of the calculation job
     """
     trace_id, exec_id = job_id.split("/")
-    workdir = os.path.join(CALCULATION_MCP_WORKDIR, trace_id)
-    with set_directory(workdir):
+    with set_directory(trace_id):
         job_info = load_job_info()
         executor = job_info["executor"] or executor
         storage = job_info["storage"] or storage
@@ -394,7 +392,7 @@ class CalculationMCPServer:
                 if create_workdir is False:
                     workdir = "."
                 else:
-                    workdir = os.path.join(CALCULATION_MCP_WORKDIR, trace_id)
+                    workdir = trace_id
                 with set_directory(workdir):
                     if preprocess_func is not None:
                         executor, storage, kwargs = preprocess_func(
@@ -439,7 +437,7 @@ class CalculationMCPServer:
                         and executor_type == "local"):
                     workdir = "."
                 else:
-                    workdir = os.path.join(CALCULATION_MCP_WORKDIR, trace_id)
+                    workdir = trace_id
                 with set_directory(workdir):
                     kwargs, input_artifacts = handle_input_artifacts(
                         fn, kwargs, storage)
@@ -472,7 +470,12 @@ class CalculationMCPServer:
             return fn
         return decorator
 
-    def run(self, **kwargs):
+    @property
+    def app(self):
+        self.mcp.settings.stateless_http = True
+        return self.mcp.streamable_http_app()
+
+    def run(self, host=None, port=None, **kwargs):
         if os.environ.get("DP_AGENT_RUNNING_MODE") in ["1", "true"]:
             return
 
@@ -488,4 +491,8 @@ class CalculationMCPServer:
                 include_in_schema=True,
             )
         )
+        if host is not None:
+            self.mcp.settings.host = host
+        if port is not None:
+            self.mcp.settings.port = port
         self.mcp.run(**kwargs)