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

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

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

{bohr_agent_sdk-0.1.120.dist-info → bohr_agent_sdk-0.1.122.dist-info}/RECORD

@@ -62,12 +62,12 @@ 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=Ax3vZbJbU4o66tRRS4R_IGcSrMmXAj9fskl-V5tgrKA,19464
+dp/agent/server/calculation_mcp_server.py,sha256=oD93SjlcHgSmGtxGJnl4V6cN1icUJSW3KEYt0Lctc6Q,19756
 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
 dp/agent/server/executor/base_executor.py,sha256=nR2jI-wFvKoOk8QaK11pnSAkHj2MsE6uyzPWDx-vgJA,3018
-dp/agent/server/executor/dispatcher_executor.py,sha256=CZRxbVkLaDvStXhNaMKrKcx2Z0tPPVzIxkU1ufqWgYc,12081
+dp/agent/server/executor/dispatcher_executor.py,sha256=lWofctAt2Nemrd3OF-IMBVWgVLKVanHDo3iMTE5lot0,12529
 dp/agent/server/executor/local_executor.py,sha256=qRVnfqhvaCGBXr-NO4uxcUteFja2BE_6NXauVJ8vnoo,6642
 dp/agent/server/storage/__init__.py,sha256=Sgsyp5hb0_hhIGugAPfQFzBHt_854rS_MuMuE3sn8Gs,389
 dp/agent/server/storage/base_storage.py,sha256=728-oNG6N8isV95gZVnyi4vTznJPJhSjxw9Gl5Y_y5o,2356
@@ -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.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,,
+bohr_agent_sdk-0.1.122.dist-info/METADATA,sha256=6PXBgaHU2sAKQJHYdN52tD18u7uIpmFEAgVbMBeSNWA,11070
+bohr_agent_sdk-0.1.122.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bohr_agent_sdk-0.1.122.dist-info/entry_points.txt,sha256=5n5kneF5IbDQtoQ2WfF-QuBjDtsimJte9Rv9baSGgc0,86
+bohr_agent_sdk-0.1.122.dist-info/top_level.txt,sha256=87xLUDhu_1nQHoGLwlhJ6XlO7OsjILh6i1nX6ljFzDo,3
+bohr_agent_sdk-0.1.122.dist-info/RECORD,,

dp/agent/server/calculation_mcp_server.py

@@ -470,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
 
@@ -486,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)

dp/agent/server/executor/dispatcher_executor.py

@@ -3,6 +3,7 @@ import inspect
 import json
 import logging
 import os
+import shutil
 import sys
 import time
 from pathlib import Path
@@ -182,9 +183,17 @@ class DispatcherExecutor(BaseExecutor):
         for package in self.python_packages:
             target = os.path.basename(package)
             if os.path.abspath(package) != os.path.abspath(target):
-                if os.path.islink(target):
-                    os.remove(target)
-                os.symlink(package, target)
+                copy_method = os.getenv("DISPATCHER_EXECUTOR_COPY_METHOD",
+                                        "symlink")
+                if copy_method == "symlink":
+                    if os.path.islink(target):
+                        os.remove(target)
+                    os.symlink(package, target)
+                elif copy_method == "copy":
+                    if os.path.isfile(package):
+                        shutil.copy(package, target)
+                    elif os.path.isdir(package):
+                        shutil.copytree(package, target)
             if target not in forward_files:
                 forward_files.append(target)
         for value in kwargs.values():