bohr-agent-sdk 0.1.22__tar.gz → 0.1.24__tar.gz

{bohr_agent_sdk-0.1.22 → bohr_agent_sdk-0.1.24}/PKG-INFO

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

{bohr_agent_sdk-0.1.22 → bohr_agent_sdk-0.1.24}/setup.py

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

{bohr_agent_sdk-0.1.22 → bohr_agent_sdk-0.1.24}/src/bohr_agent_sdk.egg-info/PKG-INFO

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

{bohr_agent_sdk-0.1.22 → bohr_agent_sdk-0.1.24}/src/dp/agent/server/executor/local_executor.py

@@ -34,6 +34,8 @@ class Tee(io.TextIOBase):
 
 def wrapped_fn(fn, kwargs, redirect_file=None):
     pid = os.getpid()
+    # explicitly reload dflow config
+    reload_dflow_config()
     if redirect_file:
         stdout = sys.stdout
         flog = open(redirect_file, "w")
@@ -61,7 +63,8 @@ def reload_dflow_config():
         s3_config = sys.modules["dflow"].s3_config
         s3_config["storage_client"] = None
         importlib.reload(sys.modules["dflow.config"])
-        if "dflow.plugins.bohrium" in sys.modules:
+        reload_bohrium = "dflow.plugins.bohrium" in sys.modules
+        if reload_bohrium:
             bohrium_config = sys.modules["dflow.plugins.bohrium"].config
             importlib.reload(sys.modules["dflow.plugins.bohrium"])
         importlib.reload(sys.modules["dflow"])
@@ -69,7 +72,7 @@ def reload_dflow_config():
         sys.modules["dflow"].config = config
         s3_config.update(sys.modules["dflow"].s3_config)
         sys.modules["dflow"].s3_config = s3_config
-        if "dflow.plugins.bohrium" in sys.modules:
+        if reload_bohrium:
             bohrium_config.update(sys.modules["dflow.plugins.bohrium"].config)
             sys.modules["dflow.plugins.bohrium"].config = bohrium_config
 
@@ -127,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)
@@ -167,7 +175,7 @@ class LocalExecutor(BaseExecutor):
         os.environ["DP_AGENT_RUNNING_MODE"] = "1"
         old_env = self.set_env()
         try:
-            # explicitly reload dflow config in sync mode
+            # explicitly reload dflow config
             reload_dflow_config()
             if inspect.iscoroutinefunction(fn):
                 result = await fn(**kwargs)