bohr-agent-sdk 0.1.101__py3-none-any.whl → 0.1.102__py3-none-any.whl

{bohr_agent_sdk-0.1.101.dist-info → bohr_agent_sdk-0.1.102.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bohr-agent-sdk
-Version: 0.1.101
+Version: 0.1.102
 Summary: SDK for scientific agents
 Home-page: https://github.com/dptech-corp/bohr-agent-sdk/
 Author: DP Technology
@@ -30,6 +30,9 @@ Requires-Dist: paho-mqtt>=2.1.0
 Requires-Dist: redis>=6.2.0
 Requires-Dist: twine>=6.1.0
 Requires-Dist: build>=1.2.2.post1
+Requires-Dist: watchdog>=6.0.0
+Requires-Dist: fastapi>=0.116.0
+Requires-Dist: bohrium-open-sdk==0.1.5
 Provides-Extra: device
 Requires-Dist: pywinauto-recorder>=0.1.0; extra == "device"
 Provides-Extra: cloud
@@ -44,6 +47,7 @@ Requires-Dist: black>=23.11.0; extra == "dev"
 Requires-Dist: isort>=5.12.0; extra == "dev"
 Requires-Dist: mypy>=1.7.0; extra == "dev"
 Requires-Dist: pylint>=3.0.0; extra == "dev"
+Requires-Dist: google-adk; extra == "dev"
 Provides-Extra: docs
 Requires-Dist: sphinx>=7.2.0; extra == "docs"
 Requires-Dist: sphinx-rtd-theme>=1.3.0; extra == "docs"
@@ -92,7 +96,7 @@ dp-agent run tool cloud
 dp-agent run tool calculation
 
 # 运行代理
-dp-agent run agent
+dp-agent run agent --config config.json
 
 # 调试模式
 dp-agent run debug

bohr_agent_sdk-0.1.102.dist-info/RECORD

@@ -0,0 +1,80 @@
+dp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dp/agent/__init__.py,sha256=4kwF1khocxEkqJjnVhBV5cCsWkR_MBGrUthuHk44pT8,153
+dp/agent/adapter/adk/__init__.py,sha256=mxX2SxGgDjwQ7WjfKMw9mQkq4ddl8XlCA6gnUinCW0w,493
+dp/agent/adapter/adk/storage_artifact_service.py,sha256=Hd1CmEB8HgEz3x_riHVSSmYdw6W2bDYYa3c5ySCqmCE,5414
+dp/agent/adapter/adk/utils.py,sha256=Hf6zZOmNi0YSI9t5acmR5kzowWaYr-UfyPiN8fefPkI,7359
+dp/agent/adapter/adk/client/__init__.py,sha256=F1xfFNa4ZG8jV9adeGI2D3YBiSX-5RkvqEdTqNdJce4,209
+dp/agent/adapter/adk/client/calculation_mcp_tool.py,sha256=8_uBP12SLaYUiQL6c0vysVsePEZS_r0YMogvW93ZFHo,12188
+dp/agent/adapter/camel/__init__.py,sha256=RN1NhdmsJyN43fTxTXFld4UKZksjpSV0b2QvFn5gK7o,77
+dp/agent/adapter/camel/client/__init__.py,sha256=ld-r0_WsZLFv6yyrmxjWmR8JgnrQzOw4fX0hwVHzciY,93
+dp/agent/adapter/camel/client/calculation_mcp_client.py,sha256=JZZUYYfMgXvHzK4f6IJp-ia33wn3aYZqDSDVa5yKtdc,1860
+dp/agent/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dp/agent/cli/cli.py,sha256=3CPdZW3nn_n5-1Ohak_LOOHqNQJbnYZHoTXIVjI19VU,12367
+dp/agent/cli/templates/__init__.py,sha256=h5__iNn8QzUyYpCUORJO9GTd42tga_BFQ3ZjA3QtSCI,15
+dp/agent/cli/templates/main.py.template,sha256=gEv_naKkBKUmVY1aGM_RZPWqXhkYDIrk4eu7uQFpWrA,1986
+dp/agent/cli/templates/calculation/simple.py.template,sha256=AkOrMWZf9YKQuLnl5yRgl2V-fepFuZJ1qDcTc9S5Gj0,395
+dp/agent/cli/templates/device/tescan_device.py.template,sha256=c7RIwWhfwp5WiPy1S6puqnEJm7ClaZtic6ndv2WE7o8,4663
+dp/agent/cli/templates/ui/__init__.py,sha256=QgYZneX7gKyajZIEvbHIEMD3av3bKj64E_eeDPBeuvY,14
+dp/agent/cli/templates/ui/test_download.py,sha256=RzZ59pVd-4fYIbUbW6ZO1b1Twbycu5G6yb5yLFuT0oI,3002
+dp/agent/cli/templates/ui/ui_utils.py,sha256=OMgQxZ-UvIJB_a58YBMypQKTI7N9B3AQBOR7kOTudRE,9774
+dp/agent/cli/templates/ui/websocket-server.py,sha256=YLVnXyBelrZLo9EcU-17_qts4-UVscMM8-cX08X6XYY,2809
+dp/agent/cli/templates/ui/api/__init__.py,sha256=eq1djuBJYvWlGTgxXvT9A386Eu21FNCcB7gV2lU46dU,8
+dp/agent/cli/templates/ui/api/config.py,sha256=ZhHApcXDrmO1JjI-Ok09ETYB-BVis390ZrMRt140DhQ,1018
+dp/agent/cli/templates/ui/api/constants.py,sha256=mHY81zf4ksfKEOsSiVsWD_A_KcQ6wveLvAImCay_kVM,1391
+dp/agent/cli/templates/ui/api/debug.py,sha256=URC1k429s84Ue7ElcK7jFh73QeK04IlRyXEjVgLXnbY,8490
+dp/agent/cli/templates/ui/api/files.py,sha256=DDqU8-u5yQmHz2sd9vuEURpdgTkQVlBjAyv7dhWz-mA,16469
+dp/agent/cli/templates/ui/api/files_upload.py,sha256=SOPy7GdQQG4grj5DvuJt3p1_HHJYdk6EuehqtKMAcqs,4411
+dp/agent/cli/templates/ui/api/files_user.py,sha256=mEFCpZChHzxNUxYUM7A3VqyMbgM3cR0M9FdT93wXJR8,1718
+dp/agent/cli/templates/ui/api/messages.py,sha256=umIkjfVA5rlXIonWQP7JRLqpfjDdvOc046gBFidMk5A,4992
+dp/agent/cli/templates/ui/api/projects.py,sha256=KOMRAPrfsbBKNBPotKHEFIKyBypQpox4lSO3ocnjA_M,5008
+dp/agent/cli/templates/ui/api/sessions.py,sha256=Mi_Sq4mfW163oZ2K0lRFiSgE7MpvVArmmSItDL2fBFo,2994
+dp/agent/cli/templates/ui/api/utils.py,sha256=6oTf1P105GEGcwDct-TB3v2pt2iQezTmywfniwrlOqM,4728
+dp/agent/cli/templates/ui/api/websocket.py,sha256=kPc9ZArxHSpvEZHl8So5vAH7rWaKNcfTk2O5_61wYUg,9013
+dp/agent/cli/templates/ui/config/__init__.py,sha256=F7JR_fnXNnZDHeqRCdEiWtVMy03_qFr8cOilIQ6XLUk,14
+dp/agent/cli/templates/ui/config/agent_config.py,sha256=WFpukdBuZ0t-QQKnhE7NQ6tYPl98FP6fmweRpNbOZQM,11255
+dp/agent/cli/templates/ui/frontend/index.html,sha256=QKDypXDFg72KI3dPq8uJ9dA0WG0XmkiZNSKCSu7_bTc,362
+dp/agent/cli/templates/ui/frontend/package.json,sha256=JGICM80tJ1Jvroi-D5RcBjBbCoqqQF_ARFl5qyci-wc,1270
+dp/agent/cli/templates/ui/frontend/tsconfig.json,sha256=1WbW4SvrlJDyfu1pkDY5-53AwTqqs1c8y7FTPEC3Rlw,664
+dp/agent/cli/templates/ui/frontend/tsconfig.node.json,sha256=aEcXXaWrgHnbZYO5LCCfeaiuOdNfOwh23fsHF7dS1Mg,212
+dp/agent/cli/templates/ui/frontend/vite.config.ts,sha256=YFMZLo35W8b0SgLQ0xPxMWk8byraKEpd4gCDdag34v8,930
+dp/agent/cli/templates/ui/frontend/ui-static/index.html,sha256=lUvP2GnHOhP3hDzXh-lUYjE8P6sFJBPdZ9pXpYu26M4,460
+dp/agent/cli/templates/ui/frontend/ui-static/assets/index-DdAmKhul.js,sha256=v6CrzD6cvbf6QiUwdpngBlkhOwGBAlKbEdm11uB7KIs,1174176
+dp/agent/cli/templates/ui/frontend/ui-static/assets/index-DfN2raU9.css,sha256=Ks6tewLxZv00LYmBf_U8CNUr-pPDSqGXFBiIJ5HaES0,68144
+dp/agent/cli/templates/ui/scripts/build_ui.py,sha256=zoK6agl1dt-InaYQyXSABxjtJj24gtCbm0KKVv0RTSA,1661
+dp/agent/cli/templates/ui/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+dp/agent/cli/templates/ui/server/app.py,sha256=ZX7ZA_qJyA-Ur7BODICzSn_S4FwGnRWcQP_qq2MZJhw,3803
+dp/agent/cli/templates/ui/server/connection.py,sha256=huet_WWxGjo_J9ozj-G8tEYSiEYTsgxOTgjEEJ6CubU,6457
+dp/agent/cli/templates/ui/server/file_watcher.py,sha256=t0e7IMgf4E1y0CxeboCqsVeZPBo1CGSw_qoO8wG6Wy4,2923
+dp/agent/cli/templates/ui/server/middleware.py,sha256=bNR5aFqKjxv2einQ0e26aq6tiUW4SMses3IHkrmHDCk,1474
+dp/agent/cli/templates/ui/server/models.py,sha256=obfMkFFB9qZZXDdwoC25u3tBlUC6Rz7sjeyuh7a_UWk,1680
+dp/agent/cli/templates/ui/server/session_manager.py,sha256=ZbNHGCFvswa-LKWn6c6RMHWDdfeOvY5L6A8CfrsF0OE,46536
+dp/agent/cli/templates/ui/server/user_files.py,sha256=khkiyY2UOOysHqO6JgCPUDqtrInp83G1M62i3Lj-0aY,2995
+dp/agent/cli/templates/ui/server/utils.py,sha256=f4NfwFBq_RdZyFn_KCW6ZThYW8TvQyVruK7PJZ-DA80,1530
+dp/agent/cloud/__init__.py,sha256=e16ymCZX2f-S8DyGB5jSK8gnQqVObRIsvtLXLALIKxQ,441
+dp/agent/cloud/main.py,sha256=5QIEjpZ1RxWnR8wyLf-vlgz1bn9oOnxCYn158LBaLN4,727
+dp/agent/cloud/mcp.py,sha256=tsAwC3doVMLYr6Oh8PxVqF-qCygYkDZJTIhoF_h8eGQ,4537
+dp/agent/cloud/mqtt.py,sha256=e8Dz1Wffzjj8V0Xrr6LukwfdiPrj9VsVXatD3u4NpkQ,23036
+dp/agent/device/__init__.py,sha256=G3i0zG7orCgMcXBEEhm7dufDeKY0QgFdQinOjDh8zSg,400
+dp/agent/device/mqtt_device_twin.py,sha256=xHQFi2WQxrL5KqSEU2DeRSBiaPT5FLXvgszinyanQ_Q,8671
+dp/agent/device/device/__init__.py,sha256=w7_1S16S1vWUq0RGl0GFgjq2vFkc5oNvy8cQTnvPm24,319
+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=eClRP7A-t5hMGyTh81KC3GAKjSPNJIylOrOKyzqwo8o,11459
+dp/agent/server/preprocessor.py,sha256=XUWu7QOwo_sIDMYS2b1OTrM33EXEVH_73vk-ju1Ok8A,1264
+dp/agent/server/utils.py,sha256=8jgYZEW4XBp86AF2Km6QkwHltBmrnS-soTpHov7ZEJw,4501
+dp/agent/server/executor/__init__.py,sha256=s95M5qKQk39Yi9qaVJZhk_nfj54quSf7EDghR3OCFUA,248
+dp/agent/server/executor/base_executor.py,sha256=EFJBsYVYAvuRbiLAbLOwLTw3h7ScjN025xnSP4uJHrQ,2052
+dp/agent/server/executor/dispatcher_executor.py,sha256=urpzmKH_tBOgblBdJEa3y8eEhXqUDrdcdWCnUdJpfZk,9420
+dp/agent/server/executor/local_executor.py,sha256=wYCclNZFkLb3v7KpW1nCnupO8piBES-esYlDAuz86zk,6120
+dp/agent/server/storage/__init__.py,sha256=Sgsyp5hb0_hhIGugAPfQFzBHt_854rS_MuMuE3sn8Gs,389
+dp/agent/server/storage/base_storage.py,sha256=728-oNG6N8isV95gZVnyi4vTznJPJhSjxw9Gl5Y_y5o,2356
+dp/agent/server/storage/bohrium_storage.py,sha256=EsKX4dWWvZTn2TEhZv4zsvihfDK0mmPFecrln-Ytk40,10488
+dp/agent/server/storage/http_storage.py,sha256=w0lY95wQqKmjXTGFRhEG2hLu8GBFwgqG8ocm5lJ_fYc,1470
+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.102.dist-info/METADATA,sha256=7FQMW--VfxyMkPlKfbj9_j5oMWFExoM_0cpxTG1a4-Y,6495
+bohr_agent_sdk-0.1.102.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+bohr_agent_sdk-0.1.102.dist-info/entry_points.txt,sha256=5n5kneF5IbDQtoQ2WfF-QuBjDtsimJte9Rv9baSGgc0,86
+bohr_agent_sdk-0.1.102.dist-info/top_level.txt,sha256=87xLUDhu_1nQHoGLwlhJ6XlO7OsjILh6i1nX6ljFzDo,3
+bohr_agent_sdk-0.1.102.dist-info/RECORD,,

dp/agent/cli/cli.py

@@ -8,7 +8,18 @@ import signal
 import uuid
 import requests
 
+try:
+    from dotenv import load_dotenv
+    env_file = Path('.env')
+    if not env_file.exists():
+        env_file = Path(__file__).parent.parent.parent.parent.parent / '.env'
+    if env_file.exists():
+        load_dotenv(env_file)
+except ImportError:
+    pass
+
 from ..server.storage import storage_dict
+from .templates.ui.ui_utils import UIConfigManager, UIProcessManager
 
 @click.group()
 def cli():
@@ -29,37 +40,34 @@ def scaffolding(type):
     """Fetch scaffolding for the science agent."""
     click.echo(f"Generating {type} project scaffold...")
     
-    # 获取模板目录路径
     templates_dir = Path(__file__).parent / 'templates'
-    
-    # 获取用户当前工作目录
     current_dir = Path.cwd()
     
     
-    # 创建必要的目录结构
+    # Create necessary directory structure
     if type == 'device':
-        project_dirs = project_dirs = ['cloud', 'device']
+        project_dirs = ['cloud', 'device']
     elif type == 'calculation':
-        project_dirs = project_dirs = ['calculation']
+        project_dirs = ['calculation']
         
     for dir_name in project_dirs:
         dst_dir = current_dir / dir_name
         
         if dst_dir.exists():
-            click.echo(f"Warning: {dir_name} already exists，skipping...")
-            click.echo(f"If you want to create a new scaffold, please delete the existing project folder first.")
+            click.echo(f"Warning: {dir_name} already exists, skipping...")
+            click.echo(f"To create a new scaffold, please delete the existing folder first.")
             continue
             
-        # 只创建目录，不复制SDK文件
+        # Create directory only
         dst_dir.mkdir(parents=True, exist_ok=True)
     
-    # 创建__init__.py文件以使目录成为Python包
+    # Create __init__.py files to make directories Python packages
     for dir_name in project_dirs:
         init_file = current_dir / dir_name / '__init__.py'
         if not init_file.exists():
             init_file.write_text('')
     
-    # 从模板创建main.py文件
+    # Create main.py from template
     main_template = templates_dir / 'main.py.template'
     main_file = current_dir / 'main.py'
     if not main_file.exists():
@@ -78,11 +86,11 @@ def scaffolding(type):
         calculation_file = current_dir / 'calculation' / 'simple.py'
         if not calculation_file.exists():
             shutil.copy2(calculation_template, calculation_file)
-            click.echo("\nCreated calculation example implementation in calculation/calculation.py")
-            click.echo("Please modify this file according to your actual calculation control requirements.")
+            click.echo("\nCreated calculation example implementation in calculation/simple.py")
+            click.echo("Please modify this file according to your actual calculation requirements.")
     
-    click.echo("\nSucceed for fetching scaffold!")
-    click.echo("Now you can use dp-agent run-cloud or dp-agent run-device to run this project!")
+    click.echo("\nSuccessfully created scaffold!")
+    click.echo("Now you can use dp-agent run tool cloud/device/calculation to run this project!")
 
 @fetch.command()
 def config():
@@ -104,7 +112,6 @@ def config():
         response = requests.get(remote_env_url)
         with open(env_file, 'w') as f:
             f.write(response.text)
-        #shutil.copy2(env_template, env_file)
         click.echo("Configuration file .env has been created.")
         click.echo("\nIMPORTANT: Please update the following configurations in your .env file:")
         click.echo("1. MQTT_INSTANCE_ID - Your Aliyun MQTT instance ID")
@@ -172,16 +179,109 @@ def calculation():
         sys.exit(1)
 
 @run.command()
-def agent():
-    """Run the science agent."""
-    click.echo("Starting agent...")
-    click.echo("Agent started.")
+@click.option('--ui/--no-ui', default=True, help='Enable/disable Web UI interface')
+@click.option('--config', help='Configuration file path (default: agent-config.json)')
+@click.option('--port', type=int, help='Server port (default from config)')
+@click.option('--module', help='Agent module path (default: agent)')
+@click.option('--agent-name', help='Agent variable name (default: root_agent)')
+@click.option('--dev/--prod', default=False, help='Development mode (default: production)')
+def agent(ui, config, port, module, agent_name, dev):
+    """Run the science agent with optional UI interface."""
+    if not ui:
+            click.echo("Starting agent in console mode...")
+        click.echo("Console mode not yet implemented.")
+        return
+    
+    current_dir = Path.cwd()
+    config_path = Path(config) if config else current_dir / "agent-config.json"
+    config_manager = UIConfigManager(config_path if config_path.exists() else None)
+    
+    if module:
+        config_manager.config['agent']['module'] = module
+    
+    agent_module = config_manager.config['agent']['module']
+    
+    if '/' in agent_module or '\\' in agent_module or agent_module.endswith('.py'):
+        file_path = Path(agent_module)
+
+        if not file_path.is_absolute():
+            file_path = current_dir / file_path
+
+        if not file_path.exists():
+            click.echo(f"Error: Agent file not found: {file_path}")
+            click.echo("\nPlease ensure:")
+            click.echo("1. File path is correct")
+            click.echo("2. File exists and is accessible")
+            sys.exit(1)
+    else:
+        module_parts = agent_module.split('.')
+        module_dir = current_dir / Path(*module_parts[:-1])
+        module_file = current_dir / Path(*module_parts[:-1]) / f"{module_parts[-1]}.py"
+        module_init = module_dir / "__init__.py"
+        
+        if not (module_dir.exists() or module_file.exists()):
+            click.echo(f"Error: Module {agent_module} not found.")
+            click.echo(f"Tried to find: {module_dir} or {module_file}")
+            click.echo("\nPlease ensure:")
+            click.echo("1. You have created the agent module")
+            click.echo("2. The 'agent.module' path is correctly configured in config.json")
+            click.echo("3. Or use --module parameter to specify the correct module path")
+            sys.exit(1)
+    
+    if os.environ.get('UI_TEMPLATE_DIR'):
+        ui_dir = Path(os.environ.get('UI_TEMPLATE_DIR'))
+    else:
+        try:
+            current_file = Path(__file__).resolve()
+
+            if 'site-packages' in str(current_file):
+                ui_dir = Path(__file__).parent / "templates" / "ui"
+            else:
+                ui_dir = current_file.parent / "templates" / "ui"
+                click.echo(f"🔧 Development mode detected, using source path: {ui_dir}")
+
+        except Exception:
+            ui_dir = Path(__file__).parent / "templates" / "ui"
+    
+    if not ui_dir.exists():
+        click.echo(f"Error: UI template directory not found: {ui_dir}")
+        click.echo("Tip: You can specify template path via UI_TEMPLATE_DIR environment variable")
+        sys.exit(1)
+    
+    if agent_name:
+        config_manager.config['agent']['rootAgent'] = agent_name
+    if port:
+        config_manager.config['server']['port'] = port
+
+    temp_config = ui_dir / "config" / "agent-config.temp.json"
+    config_manager.save_config(temp_config)
+
+    os.environ['AGENT_CONFIG_PATH'] = str(temp_config)
+    os.environ['PYTHONPATH'] = f"{current_dir}:{os.environ.get('PYTHONPATH', '')}"
+    
+    try:
+        process_manager = UIProcessManager(ui_dir, config_manager.config)
+
+        process_manager.start_websocket_server()
+        process_manager.start_frontend_server(dev_mode=dev)
+
+        click.echo("\nPress Ctrl+C to stop the service...")
+        process_manager.wait_for_processes()
+
+    except KeyboardInterrupt:
+        click.echo("\nShutting down services...")
+        if 'process_manager' in locals():
+            process_manager.cleanup()
+        sys.exit(0)
+    except Exception as e:
+        click.echo(f"Error: {e}")
+        if 'process_manager' in locals():
+            process_manager.cleanup()
+        sys.exit(1)
+    finally:
+        if 'temp_config' in locals() and temp_config.exists():
+            temp_config.unlink()
 
-@run.command()
-def debug():
-    """Debug the science agent in cloud environment."""
-    click.echo("Starting cloud environment in debug mode...")
-    click.echo("Cloud environment debug mode started.")
 
 @cli.group()
 def artifact():
@@ -229,6 +329,7 @@ def download(**kwargs):
     path = storage.download(key, path)
     click.echo("%s has been downloaded to %s" % (uri, path))
 
+
 def main():
     cli()
 

dp/agent/cli/templates/__init__.py

@@ -0,0 +1 @@
+# CLI 模板包

dp/agent/cli/templates/calculation/simple.py.template

@@ -0,0 +1,15 @@
+from dp.agent.server import CalculationMCPServer
+
+# 创建 MCP Server 实例
+mcp = CalculationMCPServer("Demo", host="0.0.0.0", port=5001)
+
+# 用装饰器注册远程调用函数
+@mcp.tool()
+def add_numbers(a: int, b: int) -> int:
+    """Add two numbers."""
+    return a + b
+
+def run():
+    print("Starting MCP server on http://127.0.0.1:5001 ...")
+    mcp.run(transport="streamable-http")
+    

dp/agent/cli/templates/device/tescan_device.py.template

@@ -0,0 +1,158 @@
+"""
+Tescan device implementation.
+
+This module contains the implementation of a Tescan device with
+specific parameter and result types.
+"""
+import time
+from typing import Dict, TypedDict
+from dp.agent.device.device.types import SuccessResult, BaseParams, ErrorResult
+from dp.agent.device.device.device import Device, action
+
+class TakePictureParams(BaseParams):
+    """Parameters for take_picture action."""
+    horizontal_width: str
+
+class GetStagePositionParams(BaseParams):
+    """Parameters for get_stage_position action."""
+    pass
+
+class MoveStageParams(BaseParams):
+    """Parameters for move_stage action."""
+    x: float
+    y: float
+    z: float
+
+class StagePosition(TypedDict):
+    """Stage position data."""
+    x: float
+    y: float
+    z: float
+
+class PictureData(TypedDict):
+    """Picture data."""
+    image_id: str
+
+
+class SleepParams(BaseParams):
+    """Parameters for sleep action."""
+    seconds: int
+
+class StagePositionResult(SuccessResult):
+    """Result for get_stage_position action."""
+    data: StagePosition
+
+class PictureResult(SuccessResult):
+    """Result for take_picture action."""
+    data: PictureData
+
+class MoveStageResult(SuccessResult):
+    """Result for move_stage action."""
+    data: Dict[str, float]
+
+class TescanDevice(Device):
+    """Tescan device implementation."""
+    device_name = "tescan"
+
+    @action("start_electron_beam")
+    def start_electron_beam(self, params: BaseParams):
+        print("start electron beam")
+    
+    @action("start_ion_beam")
+    def start_ion_beam(self, params: BaseParams):
+        print("start ion beam")
+    
+    @action("toggle_electron_scanning")
+    def toggle_electron_scanning(self, params: BaseParams):
+        """start scanning, this will cost 5 seconds to finish. call sleep(5) after toggle this.
+        """
+        print("toggle electron scanning")
+    
+    @action("toggle_ion_scanning")
+    def toggle_ion_scanning(self, params: BaseParams):
+        print("toggle ion scanning")
+
+    @action("sleep")
+    def sleep(self, params: SleepParams):
+        time.sleep(params.get("seconds", 0))
+
+    @action("take_picture")
+    def take_picture(self, params: TakePictureParams) -> PictureResult:
+        """Take a picture with the microscope.
+        
+        Args:
+            params: Parameters for the action
+                - horizontal_width: Horizontal width of the image
+                
+        Returns:
+            Result of the action
+        """
+        time.sleep(1)
+        
+        hw = params.get("horizontal_width", "default")
+        
+        return PictureResult(
+            message=f"Picture taken with {self.device_name} (width: {hw})",
+            data={"image_id": "mock_image_123"}
+        )
+    
+    @action("get_stage_position")
+    def get_stage_position(self, params: GetStagePositionParams) -> StagePositionResult:
+        """Get the stage position.
+        
+        Args:
+            params: Parameters for the action (none required)
+                
+        Returns:
+            Result of the action with x, y, z coordinates
+        """
+        time.sleep(0.5)
+        
+        return StagePositionResult(
+            message=f"Stage position retrieved for {self.device_name}",
+            data=StagePosition(
+                x=10.5,
+                y=20.3,
+                z=5.1
+            )
+        )
+    
+    @action("move_stage")
+    def move_stage(self, params: MoveStageParams) -> MoveStageResult:
+        """Move the stage to a new position.
+        
+        Args:
+            params: Parameters for the action
+                - x: X coordinate
+                - y: Y coordinate
+                - z: Z coordinate
+                
+        Returns:
+            Result of the action
+        """
+        time.sleep(2)
+        
+        if not all(k in params for k in ["x", "y", "z"]):
+            # Use a proper MoveStageResult with error status instead of ErrorResult
+            return MoveStageResult(
+                message=f"Missing required parameters: x, y, z",
+                data={}
+            )
+        
+        try:
+            new_position = {
+                "x": float(params["x"]),
+                "y": float(params["y"]),
+                "z": float(params["z"])
+            }
+            
+            return MoveStageResult(
+                message=f"Stage moved for {self.device_name}",
+                data=new_position
+            )
+        except (KeyError, TypeError, ValueError):
+            # Use a proper MoveStageResult with error status instead of ErrorResult
+            return MoveStageResult(
+                message=f"Invalid parameters: x, y, z must be numeric values",
+                data={}
+            )

dp/agent/cli/templates/main.py.template

@@ -0,0 +1,67 @@
+import sys
+import signal
+from pathlib import Path
+import os
+
+# 从SDK包中导入所需模块
+from dp.agent.cloud.mcp import mcp
+from dp.agent.cloud.mqtt import get_mqtt_cloud_instance
+from dp.agent.device.mqtt_device_twin import DeviceTwin
+# 从本地导入 TescanDevice
+
+def run_cloud():
+    """Run in cloud environment"""
+    mqtt_instance = get_mqtt_cloud_instance()
+    from dp.agent.device.device.device import register_mcp_tools
+    from device.tescan_device import TescanDevice
+    register_mcp_tools(mcp, TescanDevice())
+    def signal_handler(sig, frame):
+        """Handle SIGINT signal to gracefully shutdown."""
+        print("Shutting down...")
+        mqtt_instance.stop()
+        sys.exit(0)
+    
+    # Register signal handler for graceful shutdown
+    signal.signal(signal.SIGINT, signal_handler)
+    
+    # Start MCP server
+    print("Starting MCP server...")
+    mcp.run(transport="sse")
+def run_device():
+    """Run in lab environment"""
+    # Initialize device and twin
+    from device.tescan_device import TescanDevice
+    tescan_device = TescanDevice()
+    device_twin = DeviceTwin(tescan_device)
+    
+    # Run device twin
+    try:
+        device_twin.run()
+    except KeyboardInterrupt:
+        print("\nShutting down lab environment...")
+        sys.exit(0)
+
+def run_calculation():
+    """Run in calculation environment"""
+    # 在此实现你自己的计算环境逻辑
+    print("Running calculation task...")
+    # 模拟任务处理逻辑
+    from calculation import simple
+    simple.run()
+    
+if __name__ == "__main__":
+    if len(sys.argv) != 2:
+        print("Usage: python main.py [device|cloud|calculation]")
+        sys.exit(1)
+        
+    mode = sys.argv[1]
+    if mode == "device":
+        run_device()
+    elif mode == "cloud":
+        run_cloud()
+    elif mode == "calculation":  
+        run_calculation()
+    else:
+        print(f"Unknown mode: {mode}")
+        print("Usage: python main.py [device|cloud|calculation]")
+        sys.exit(1) 

dp/agent/cli/templates/ui/__init__.py

@@ -0,0 +1 @@
+# UI 模板包

dp/agent/cli/templates/ui/api/__init__.py

@@ -0,0 +1 @@
+# API !W

dp/agent/cli/templates/ui/api/config.py

@@ -0,0 +1,32 @@
+# Configuration API
+from fastapi import Request
+from fastapi.responses import JSONResponse
+
+from server.utils import get_ak_info_from_request
+from config.agent_config import agentconfig
+
+
+async def get_config(request: Request):
+    """Get frontend configuration"""
+    access_key, _ = get_ak_info_from_request(request.headers)
+    return JSONResponse(content={
+        "agent": agentconfig.config.get("agent", {}),
+        "ui": agentconfig.get_ui_config(),
+        "files": agentconfig.get_files_config(),
+        "websocket": agentconfig.get_websocket_config(),
+        "user_type": "registered" if access_key else "temporary"
+    })
+
+
+async def status():
+    """API status"""
+    return {
+        "message": f"{agentconfig.config.get('agent', {}).get('name', 'Agent')} WebSocket 服务器正在运行",
+        "mode": "session",
+        "endpoints": {
+            "websocket": "/ws",
+            "files": "/api/files",
+            "file_tree": "/api/files/tree",
+            "config": "/api/config"
+        }
+    }