bohr-agent-sdk 0.1.116__tar.gz → 0.1.117__tar.gz

{bohr_agent_sdk-0.1.116 → bohr_agent_sdk-0.1.117}/PKG-INFO

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

{bohr_agent_sdk-0.1.116 → bohr_agent_sdk-0.1.117}/setup.py

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

{bohr_agent_sdk-0.1.116 → bohr_agent_sdk-0.1.117}/src/bohr_agent_sdk.egg-info/PKG-INFO

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

{bohr_agent_sdk-0.1.116 → bohr_agent_sdk-0.1.117}/src/dp/agent/server/calculation_mcp_server.py

@@ -371,7 +371,9 @@ class CalculationMCPServer:
         self.mcp._tool_manager._tools[tool.name] = tool
         return tool
 
-    def tool(self, preprocess_func=None, create_workdir=True):
+    def tool(self, preprocess_func=None, create_workdir=None):
+        # When create_workdir is None, do not create workdir when fn is async
+        # and running locally to avoid chdir conflicts, create otherwise
         if preprocess_func is None:
             preprocess_func = self.preprocess_func
 
@@ -381,11 +383,17 @@ class CalculationMCPServer:
                            **kwargs) -> SubmitResult:
                 trace_id = datetime.today().strftime('%Y-%m-%d-%H:%M:%S.%f')
                 logger.info("Job processing (Trace ID: %s)" % trace_id)
-                workdir = trace_id if create_workdir else "."
+                if preprocess_func is not None:
+                    executor, storage, kwargs = preprocess_func(
+                        executor, storage, kwargs)
+                executor_type, executor = init_executor(executor)
+                if create_workdir is False or (
+                    create_workdir is None and inspect.iscoroutinefunction(fn)
+                        and executor_type == "local"):
+                    workdir = "."
+                else:
+                    workdir = trace_id
                 with set_directory(workdir):
-                    if preprocess_func is not None:
-                        executor, storage, kwargs = preprocess_func(
-                            executor, storage, kwargs)
                     job = {
                         "tool_name": fn.__name__,
                         "executor": executor,
@@ -395,7 +403,6 @@ class CalculationMCPServer:
                         json.dump(job, f, indent=4)
                     kwargs, input_artifacts = handle_input_artifacts(
                         fn, kwargs, storage)
-                    executor_type, executor = init_executor(executor)
                     res = executor.submit(fn, kwargs)
                     exec_id = res["job_id"]
                     job_id = "%s/%s" % (workdir, exec_id)
@@ -417,14 +424,19 @@ class CalculationMCPServer:
                 context = self.mcp.get_context()
                 trace_id = datetime.today().strftime('%Y-%m-%d-%H:%M:%S.%f')
                 logger.info("Job processing (Trace ID: %s)" % trace_id)
-                workdir = trace_id if create_workdir else "."
+                if preprocess_func is not None:
+                    executor, storage, kwargs = preprocess_func(
+                        executor, storage, kwargs)
+                executor_type, executor = init_executor(executor)
+                if create_workdir is False or (
+                    create_workdir is None and inspect.iscoroutinefunction(fn)
+                        and executor_type == "local"):
+                    workdir = "."
+                else:
+                    workdir = trace_id
                 with set_directory(workdir):
-                    if preprocess_func is not None:
-                        executor, storage, kwargs = preprocess_func(
-                            executor, storage, kwargs)
                     kwargs, input_artifacts = handle_input_artifacts(
                         fn, kwargs, storage)
-                    executor_type, executor = init_executor(executor)
                     res = await executor.async_run(
                         fn, kwargs, context, workdir)
                     exec_id = res["job_id"]