v2: src/start.py

src/start.py

@@ -0,0 +1,121 @@
+import os
+from pathlib import Path
+
+CONFIG_DIR = Path("/data/.config")
+CONFIG_FILE = CONFIG_DIR / "hermes-config.yaml"
+ENV_FILE = Path("/data/.env")
+
+
+def load_env():
+    env = {}
+    if not ENV_FILE.exists():
+        return env
+    for line in ENV_FILE.read_text().splitlines():
+        line = line.strip()
+        if not line or line.startswith("#"):
+            continue
+        if "=" in line:
+            key, _, val = line.partition("=")
+            env[key.strip()] = val.strip()
+            os.environ[key.strip()] = val.strip()
+    return env
+
+
+def build_config(env):
+    provider = env.get("HERMES_PROVIDER", "openai")
+    model = env.get("HERMES_MODEL", "gpt-4o")
+
+    providers = {}
+
+    builtins = {
+        "openai": {
+            "base_url": lambda e: e.get("OPENAI_BASE_URL", e.get("OPENAI_API_BASE", "https://api.openai.com/v1")),
+            "key_env": "OPENAI_API_KEY",
+        },
+        "anthropic": {
+            "base_url": lambda e: e.get("ANTHROPIC_BASE_URL", "https://api.anthropic.com"),
+            "key_env": "ANTHROPIC_API_KEY",
+        },
+        "deepseek": {
+            "base_url": lambda e: e.get("DEEPSEEK_BASE_URL", "https://api.deepseek.com/v1"),
+            "key_env": "DEEPSEEK_API_KEY",
+        },
+        "gemini": {
+            "base_url": lambda e: e.get("GEMINI_BASE_URL", "https://generativelanguage.googleapis.com/v1beta/openai"),
+            "key_env": "GEMINI_API_KEY",
+        },
+        "openrouter": {
+            "base_url": lambda e: e.get("OPENROUTER_BASE_URL", "https://openrouter.ai/api/v1"),
+            "key_env": "OPENROUTER_API_KEY",
+        },
+    }
+
+    for name, cfg in builtins.items():
+        key_val = os.getenv(cfg["key_env"], "")
+        if key_val and not key_val.startswith("sk-your-"):
+            providers[name] = {
+                "name": name,
+                "base_url": cfg["base_url"](env),
+                "key_env": cfg["key_env"],
+                "transport": "openai_chat",
+                "default_model": model,
+                "models": [model],
+            }
+
+    i = 1
+    while True:
+        prefix = "CUSTOM_{}".format(i)
+        name = env.get("{}_NAME".format(prefix), "")
+        if not name:
+            break
+        base_url = env.get("{}_BASE_URL".format(prefix), "").rstrip("/")
+        key_env = "{}_API_KEY".format(prefix)
+        default_model = env.get("{}_DEFAULT_MODEL".format(prefix), "")
+        if base_url and os.getenv(key_env) and default_model:
+            providers[name] = {
+                "name": name,
+                "base_url": base_url,
+                "key_env": key_env,
+                "transport": "openai_chat",
+                "default_model": default_model,
+                "models": [default_model],
+            }
+        i += 1
+
+    lines = []
+    lines.append("model:")
+    lines.append("  provider: {}".format(provider))
+    lines.append('  default: "{}"'.format(model))
+    lines.append("")
+
+    if providers:
+        lines.append("providers:")
+        for pname, pcfg in providers.items():
+            lines.append("  {}:".format(pname))
+            for k, v in [("name", pcfg["name"]), ("base_url", pcfg["base_url"]),
+                         ("key_env", pcfg["key_env"]), ("transport", pcfg["transport"])]:
+                lines.append("    {}: {}".format(k, v))
+            lines.append('    default_model: "{}"'.format(pcfg["default_model"]))
+            lines.append("    models:")
+            for m in pcfg["models"]:
+                lines.append('      "{}":'.format(m))
+
+    lines.append("")
+    lines.append("terminal:")
+    lines.append("  backend: local")
+
+    return "\n".join(lines) + "\n"
+
+
+def main():
+    CONFIG_DIR.mkdir(parents=True, exist_ok=True)
+    env = load_env()
+    config = build_config(env)
+    CONFIG_FILE.write_text(config, encoding="utf-8")
+    print("[start.py] Config written to {}".format(CONFIG_FILE))
+    print("[start.py] Provider: {}".format(env.get("HERMES_PROVIDER", "openai")))
+    print("[start.py] Model: {}".format(env.get("HERMES_MODEL", "gpt-4o")))
+
+
+if __name__ == "__main__":
+    main()