mirror of
https://github.com/mudler/LocalAI.git
synced 2026-05-16 20:52:08 -04:00
b1a99436c7
Adds a whitelabeling feature so an operator can replace the LocalAI
instance name, tagline, square logo, horizontal logo, and favicon from
the admin Settings page. Defaults fall back to the bundled assets so
existing installs are unaffected.
The public GET /api/branding endpoint is reachable pre-auth so the
login screen can render the configured branding before sign-in.
Mutating routes (POST/DELETE /api/branding/asset/:kind) remain
admin-only. Text fields (instance_name, instance_tagline) ride the
existing /api/settings flow; binary assets get a dedicated multipart
upload route that persists files under DynamicConfigsDir/branding/.
To prevent the Settings page's stale local state from clobbering an
upload on save, UpdateSettingsEndpoint preserves whatever the on-disk
asset filename fields are regardless of the body — /api/branding/asset/*
are the sole writers for those fields.
The MCP catalog gains get_branding and set_branding tools (text fields
only; file upload stays UI-only) plus a configure_branding skill prompt.
While wiring this up, the same restart-loss class of bug surfaced for
several existing fields whose RuntimeSettings entries were never read
by the startup loader. Fix loadRuntimeSettingsFromFile() to load:
- branding (instance_name, instance_tagline, *_file basenames)
- auto_upgrade_backends, prefer_development_backends
- localai_assistant_enabled
- open_responses_store_ttl
- the 7 existing AgentPool fields (enabled, default/embedding model,
chunking sizes, enable_logs, collection_db_path)
Also exposes 3 new AgentPool runtime settings (vector_engine,
database_url, agent_hub_url) via /api/settings + the Settings UI, with
the same load-on-startup wiring. The file watcher's manual-edit path
is intentionally not changed — the in-process API endpoints already
update appConfig directly, so the watcher is redundant for supported
flows and a separate refactor for everything else.
15 TDD specs cover the loader behaviour (1 branding + 11 adjacent + 3
new agent-pool); 2 specs cover the persistence helpers and the
clobber-prevention contract.
Assisted-by: claude-code:claude-opus-4-7
Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
54 lines
1.5 KiB
Go
54 lines
1.5 KiB
Go
package localaitools
|
|
|
|
import (
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
|
|
|
"github.com/mudler/LocalAI/internal"
|
|
)
|
|
|
|
// Options control which tools the server registers and how the embedded
|
|
// skill prompts are surfaced.
|
|
type Options struct {
|
|
// DisableMutating omits all tools that change server state. Used by the
|
|
// "--read-only" flavour of the standalone stdio CLI.
|
|
DisableMutating bool
|
|
|
|
// ServerName overrides the MCP server's advertised Implementation.Name.
|
|
// Defaults to "localai-admin".
|
|
ServerName string
|
|
|
|
// ServerVersion overrides the advertised version. Defaults to the linked
|
|
// internal.PrintableVersion().
|
|
ServerVersion string
|
|
}
|
|
|
|
// NewServer builds an MCP server that exposes LocalAI's admin surface as
|
|
// tools, backed by the supplied LocalAIClient. The same server type is used
|
|
// in-process (paired in-memory transport) and out-of-process (stdio).
|
|
func NewServer(client LocalAIClient, opts Options) *mcp.Server {
|
|
name := opts.ServerName
|
|
if name == "" {
|
|
name = DefaultServerName
|
|
}
|
|
version := opts.ServerVersion
|
|
if version == "" {
|
|
version = internal.PrintableVersion()
|
|
}
|
|
|
|
srv := mcp.NewServer(&mcp.Implementation{
|
|
Name: name,
|
|
Version: version,
|
|
}, &mcp.ServerOptions{
|
|
Instructions: SystemPrompt(opts),
|
|
})
|
|
|
|
registerModelTools(srv, client, opts)
|
|
registerBackendTools(srv, client, opts)
|
|
registerConfigTools(srv, client, opts)
|
|
registerSystemTools(srv, client, opts)
|
|
registerStateTools(srv, client, opts)
|
|
registerBrandingTools(srv, client, opts)
|
|
|
|
return srv
|
|
}
|