mirror of
https://github.com/mudler/LocalAI.git
synced 2026-06-22 07:39:02 -04:00
9565db5f94
* feat(config): add model alias field and self-validation Add ModelConfig.Alias (yaml: alias), IsAlias(), and an alias short-circuit at the top of Validate() that rejects self-reference and forbids setting backend/parameters.model on a pure-redirect alias. Assisted-by: Claude:claude-opus-4-8 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * feat(config): resolve and validate model alias targets in the loader Assisted-by: Claude:opus-4-8 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * feat(middleware): resolve model aliases and stamp requested/served identity Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * feat(modeladmin): reject alias configs with invalid targets on create/edit Validate alias targets at create/swap entry points (ImportModelEndpoint, EditYAML, PatchConfig) so a dangling, chained, or disabled alias target is rejected at save time rather than surfacing as a runtime error. Assisted-by: Claude:opus-4-8 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * feat(api): add GET /api/aliases to list model aliases Adds an admin-gated read-only endpoint that lists every model alias config as {name, target} pairs, backed by the loader's existing GetAllModelsConfigs(). Assisted-by: Claude:opus-4.8 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * feat(mcp): add set_alias and list_aliases tools Expose model-alias management over the LocalAI Assistant MCP surface: list_aliases (read-only, GET /api/aliases) and set_alias (mutating). SetAlias is swap-first: PATCH /api/models/config-json/:name swaps an existing alias's target (validated, non-destructive) and a 404 falls back to POST /models/import to create a fresh {name, alias} config. The inproc client mirrors this via ConfigService.PatchConfig + a create path modeled on ImportModelEndpoint. Deletion reuses delete_model. Assisted-by: Claude:claude-opus-4 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * style(mcp): replace em dashes in alias tool comments Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * feat(config-meta): expose alias as a model-select field Add an 'alias' section to DefaultSections() and an 'alias' field override in DefaultRegistry() so the schema-driven React editor renders the new top-level ModelConfig.Alias field as a model picker in its own section. Assisted-by: Claude:opus-4.8 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * feat(ui): add alias template card and Manage alias badge Add an 'Alias / Routing' template to the create-flow gallery that seeds a minimal name + alias config, and a read-only 'alias -> target' badge on the Manage Models tab. The capabilities row payload does not carry the alias field, so the badge resolves targets from GET /api/aliases looked up by name. Assisted-by: Claude:claude-opus-4 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * docs: document model aliases Assisted-by: Claude:claude-opus-4-8 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * docs(swagger): regenerate for GET /api/aliases Adds the /api/aliases path and AliasInfo schema generated from the ListAliasesEndpoint annotation. Assisted-by: Claude:claude-opus-4-8 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * test(localai): check os.RemoveAll error in aliases_test Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * fix: correct alias conversion docs and advertise /api/aliases in instructions Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * fix(mcp): write alias config 0600 to satisfy gosec G306 The inproc createAlias path wrote the alias YAML with 0644, which gosec flags as a new G306 finding on the PR. The LocalAI process is the sole reader/writer of model configs, so 0600 is correct and keeps the scan clean. Assisted-by: Claude:claude-opus-4-8 [Claude Code] Signed-off-by: Ettore Di Giacinto <mudler@localai.io> --------- Signed-off-by: Ettore Di Giacinto <mudler@localai.io> Co-authored-by: Ettore Di Giacinto <mudler@localai.io>
58 lines
1.6 KiB
Go
58 lines
1.6 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)
|
|
registerAliasTools(srv, client, opts)
|
|
registerBackendTools(srv, client, opts)
|
|
registerConfigTools(srv, client, opts)
|
|
registerSystemTools(srv, client, opts)
|
|
registerStateTools(srv, client, opts)
|
|
registerBrandingTools(srv, client, opts)
|
|
registerUsageTools(srv, client, opts)
|
|
registerPIITools(srv, client, opts)
|
|
registerMiddlewareTools(srv, client, opts)
|
|
|
|
return srv
|
|
}
|