Spaces:
Sleeping
Sleeping
File size: 3,944 Bytes
3d94637 e54a7d5 3d94637 e54a7d5 3d94637 e54a7d5 3d94637 18a5cda e54a7d5 3d94637 e54a7d5 6c8facb 3d94637 18a5cda 6c8facb 3d94637 67fda28 3d94637 e54a7d5 6c8facb e54a7d5 67fda28 18a5cda 3d94637 e54a7d5 3d94637 67fda28 3d94637 67fda28 18a5cda 3d94637 e54a7d5 3d94637 18a5cda 67fda28 3d94637 67fda28 3d94637 18a5cda 3d94637 67fda28 e54a7d5 67fda28 18a5cda 3d94637 e54a7d5 3d94637 18a5cda 67fda28 18a5cda 67fda28 18a5cda 3d94637 e54a7d5 3d94637 18a5cda 67fda28 18a5cda 67fda28 e54a7d5 67fda28 18a5cda 3d94637 18a5cda 67fda28 3d94637 67fda28 18a5cda 3d94637 e54a7d5 3d94637 67fda28 e54a7d5 67fda28 6c8facb 18a5cda 3d94637 18a5cda 3d94637 18a5cda 6c8facb 18a5cda 6c8facb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 | import os
import json
import subprocess
from pathlib import Path
import spaces
import gradio as gr
# ============================================================
# ZeroGPU registration
# ============================================================
@spaces.GPU(duration=1)
def _zerogpu_probe():
return None
# A real Gradio event bound to the GPU-decorated function.
# The button is never exposed to the user.
with gr.Blocks() as _zero_gpu_demo:
_probe = gr.Button(visible=False)
_probe.click(
fn=_zerogpu_probe,
inputs=[],
outputs=[],
)
# IMPORTANT:
# ZeroGPU detects GPU handlers when Blocks.launch() is called.
# We launch Gradio on an INTERNAL port so nanobot can own 7860.
_zero_gpu_demo.launch(
server_name="127.0.0.1",
server_port=7861,
prevent_thread_lock=True,
)
# ============================================================
# Paths
# ============================================================
DATA = Path("/data/nanobot")
CONFIG = DATA / "config.json"
WORKSPACE = DATA / "workspace"
DATA.mkdir(parents=True, exist_ok=True)
WORKSPACE.mkdir(parents=True, exist_ok=True)
# ============================================================
# Secrets
# ============================================================
WEB_TOKEN = os.environ.get("NANOBOT_WEB_TOKEN")
GEMINI_KEY = os.environ.get("GEMINI_API_KEY")
if not WEB_TOKEN:
raise RuntimeError("NANOBOT_WEB_TOKEN is not set.")
if not GEMINI_KEY:
raise RuntimeError("GEMINI_API_KEY is not set.")
# ============================================================
# Load existing configuration
# ============================================================
if CONFIG.exists():
config = json.loads(
CONFIG.read_text(encoding="utf-8")
)
else:
config = {}
# ============================================================
# Gemini provider
# ============================================================
config.setdefault("providers", {})
config["providers"]["gemini"] = {
"apiKey": "${GEMINI_API_KEY}"
}
# ============================================================
# Primary model
# ============================================================
config.setdefault("modelPresets", {})
config["modelPresets"]["primary"] = {
"provider": "gemini",
"model": "gemini-2.5-flash",
"maxTokens": 8192,
"contextWindowTokens": 1048576,
"temperature": 0.1,
}
# ============================================================
# Agent defaults
# ============================================================
config.setdefault("agents", {})
config["agents"].setdefault("defaults", {})
config["agents"]["defaults"]["modelPreset"] = "primary"
config["agents"]["defaults"]["workspace"] = str(WORKSPACE)
# ============================================================
# Nanobot WebUI
# ============================================================
config.setdefault("channels", {})
config["channels"]["websocket"] = {
"enabled": True,
"host": "0.0.0.0",
"port": 7860,
"tokenIssueSecret": "${NANOBOT_WEB_TOKEN}",
"websocketRequiresToken": True,
}
# ============================================================
# Tool security
# ============================================================
config.setdefault("tools", {})
config["tools"]["restrictToWorkspace"] = True
config["tools"]["webuiAllowRemotePackageInstall"] = False
# ============================================================
# Save configuration
# ============================================================
CONFIG.write_text(
json.dumps(config, indent=2),
encoding="utf-8",
)
# ============================================================
# Start nanobot
# ============================================================
subprocess.run(
[
"nanobot",
"gateway",
"--config",
str(CONFIG),
"--workspace",
str(WORKSPACE),
],
check=True,
) |