Skip to content

TASK.md

Latest commit

 

History

History
74 lines (58 loc) · 2.85 KB

TASK.md

File metadata and controls

74 lines (58 loc) · 2.85 KB

Tensors Refactoring Task

Goal

Transform tensors from a "sd-server wrapper" to a pure client (CLI + UI) that talks to sd-server API directly.

  • Default API URL: https://sd-api.saiden.dev
  • API key auth via config: sd_server_api_key
  • No wrapping/proxying/restarting of sd-server

Current Architecture

CLI Commands

Command What it does Talks to
tsr info <file> Read safetensor metadata, fetch CivitAI info Local file + CivitAI API
tsr search Search CivitAI models CivitAI API
tsr get <id> Fetch model info from CivitAI CivitAI API
tsr dl <id> Download model from CivitAI CivitAI API
tsr config Manage config (API keys, remotes) Local config
tsr generate Generate images sd-server API (direct or via wrapper)
tsr status Show wrapper status tensors wrapper API
tsr reload Reload sd-server with new model tensors wrapper API
tsr serve Start wrapper API (proxies to sd-server) Starts FastAPI server
tsr db Manage local models DB Local SQLite
tsr images Manage remote gallery tensors wrapper API
tsr models List models on remote tensors wrapper API
tsr remote Manage remote server config Local config

Server (tsr serve)

Starts a FastAPI server that:

  1. Proxies all requests to sd-server (catch-all route)
  2. Serves Vue UI at /
  3. Adds features: gallery, CivitAI search, model listing, downloads
  4. Has a /reload endpoint that proxies to sd-server

What to Remove

Remove Reason
tsr reload command sd-server manages its own models
/reload route in server Same
switch_model in client.py Same
Proxy wrapper concept tensors should call API directly, not proxy

What to Keep/Refactor

Keep Change
tsr serve Just serve Vue UI, no proxying
tsr generate Call sd-server API directly with API key
Vue UI Call sd-server API directly (already does via /api/*)
tsr models List models from sd-server API directly

Already Done

  • Added get_sd_server_api_key() to config.py
  • Added sd_server_api_key to app state in server/init.py
  • Created sd_client.py with get_sd_headers() helper
  • Updated generate_routes.py to use API key headers
  • Updated routes.py to use API key headers
  • Updated models_routes.py to use API key headers
  • Created local config at ~/.xdg/tensors/config.toml with sd-api.saiden.dev

Still TODO

  • Remove tsr reload command from cli.py
  • Remove /reload route if it exists
  • Remove switch_model from client.py
  • Decide: keep tsr serve as UI server or remove entirely?
  • Update Vue UI to call sd-server API directly (not via /api/* proxy)
  • Clean up unused wrapper/proxy code