bohr-agent-sdk 0.1.119__py3-none-any.whl → 0.1.120__py3-none-any.whl

{bohr_agent_sdk-0.1.119.dist-info → bohr_agent_sdk-0.1.120.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bohr-agent-sdk
-Version: 0.1.119
+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.119.dist-info → bohr_agent_sdk-0.1.120.dist-info}/RECORD

@@ -62,7 +62,7 @@ dp/agent/device/device/__init__.py,sha256=w7_1S16S1vWUq0RGl0GFgjq2vFkc5oNvy8cQTn
 dp/agent/device/device/device.py,sha256=9ZRIJth-4qMO-i-u_b_cO3d6a4eTbTQjPaxFsV_zEkc,9643
 dp/agent/device/device/types.py,sha256=JuxB-hjf1CjjvfBxCLwRAXVFlYS-nPEdiJpBWLFVCzo,1924
 dp/agent/server/__init__.py,sha256=rckaYd8pbYyB4ENEhgjXKeGMXjdnrgcJpdM1gu5u1Wc,508
-dp/agent/server/calculation_mcp_server.py,sha256=Nq1lv-H1E8J1yNz85TCbl1GtSQXQGZfAIB1qKajZcNY,19664
+dp/agent/server/calculation_mcp_server.py,sha256=Ax3vZbJbU4o66tRRS4R_IGcSrMmXAj9fskl-V5tgrKA,19464
 dp/agent/server/preprocessor.py,sha256=XUWu7QOwo_sIDMYS2b1OTrM33EXEVH_73vk-ju1Ok8A,1264
 dp/agent/server/utils.py,sha256=cIKaAg8UaP5yMwvIVTgUVBjy-B3S16bEdnucUf4UDIM,2055
 dp/agent/server/executor/__init__.py,sha256=s95M5qKQk39Yi9qaVJZhk_nfj54quSf7EDghR3OCFUA,248
@@ -75,8 +75,8 @@ dp/agent/server/storage/bohrium_storage.py,sha256=EsKX4dWWvZTn2TEhZv4zsvihfDK0mm
 dp/agent/server/storage/http_storage.py,sha256=KiySq7g9-iJr12XQCKKyJLn8wJoDnSRpQAR5_qPJ1ZU,1471
 dp/agent/server/storage/local_storage.py,sha256=t1wfjByjXew9ws3PuUxWxmZQ0-Wt1a6t4wmj3fW62GI,1352
 dp/agent/server/storage/oss_storage.py,sha256=pgjmi7Gir3Y5wkMDCvU4fvSls15fXT7Ax-h9MYHFPK0,3359
-bohr_agent_sdk-0.1.119.dist-info/METADATA,sha256=oG4bst9eor1ytVutI5UGppqY8aJ-C9c12jKOGBoHh2g,11070
-bohr_agent_sdk-0.1.119.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-bohr_agent_sdk-0.1.119.dist-info/entry_points.txt,sha256=5n5kneF5IbDQtoQ2WfF-QuBjDtsimJte9Rv9baSGgc0,86
-bohr_agent_sdk-0.1.119.dist-info/top_level.txt,sha256=87xLUDhu_1nQHoGLwlhJ6XlO7OsjILh6i1nX6ljFzDo,3
-bohr_agent_sdk-0.1.119.dist-info/RECORD,,
+bohr_agent_sdk-0.1.120.dist-info/METADATA,sha256=GBYxEWv13ktFJwnVq8yXqNFTVeHRhlxZdC2nBlecdsw,11070
+bohr_agent_sdk-0.1.120.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bohr_agent_sdk-0.1.120.dist-info/entry_points.txt,sha256=5n5kneF5IbDQtoQ2WfF-QuBjDtsimJte9Rv9baSGgc0,86
+bohr_agent_sdk-0.1.120.dist-info/top_level.txt,sha256=87xLUDhu_1nQHoGLwlhJ6XlO7OsjILh6i1nX6ljFzDo,3
+bohr_agent_sdk-0.1.120.dist-info/RECORD,,

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)