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

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

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

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

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

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

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

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

@@ -383,17 +383,14 @@ 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)
-                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"):
+                if create_workdir is False:
                     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,
@@ -403,6 +400,7 @@ 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)