mirror of
https://github.com/mudler/LocalAI.git
synced 2026-11-06 12:33:16 -05:00
Ettore Di Giacinto
1834d4fd72
No release ever shipped add-ons, so there is no installed base on api 1.
The manifest must now declare api {"min": 2, "max": 2}; a manifest with
api 1, no api or a wider range is refused with a message that names the
range.
Every field that was gated on api 2 is now part of the contract, so the
Min*API constants and their gates are gone, and so are HostAPIMin and
the apiMin host field. Three older-manifest shims go with them: the
resource listing the host implied for one add-on, the legacyArgs hint
for arguments an older version of a tool took, and the tests that
checked the gates. All shipped manifests and the test fixtures declare
api 2.
Assisted-by: Claude:claude-sonnet-5-5 go test
67 lines
3.2 KiB
Go
67 lines
3.2 KiB
Go
// SPDX-License-Identifier: MIT
|
|
package protocol
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
|
|
"github.com/mudler/LocalAI/pkg/addons/contract"
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("typed declaration echo", func() {
|
|
It("echoes model actions and typed declarations together separately and absent over authenticated health", func() {
|
|
for mask := 0; mask < 16; mask++ {
|
|
caps := contract.Capabilities{Service: true}
|
|
if mask&1 != 0 {
|
|
caps.UI = &contract.UI{Entry: "ui/main.js", SDK: contract.VersionRange{Min: 2, Max: 2}, Pages: []string{"home"}, MainPage: "home", HomeCommands: []contract.HomeCommand{{ID: "open"}}}
|
|
}
|
|
if mask&8 != 0 {
|
|
if caps.UI == nil {
|
|
caps.UI = &contract.UI{Entry: "ui.js", SDK: contract.VersionRange{Min: 2, Max: 2}, Pages: []string{"home"}}
|
|
}
|
|
caps.UI.ModelActions = []contract.ModelAction{{Capability: "FLAG_IMAGE", Page: "home", Path: "images"}}
|
|
}
|
|
if mask&2 != 0 {
|
|
caps.Actions = true
|
|
caps.Tools = []string{"run", "health"}
|
|
caps.ToolDefinitions = []contract.ToolDefinition{{Name: "run", Description: "Run", Kind: "action", Shadows: "bash", InputSchema: json.RawMessage(`{"type":"object","properties":{"command":{"type":"string"}},"required":["command"],"additionalProperties":false}`)}, {Name: "health", Description: "Status", Kind: "read", UIOnly: true, InputSchema: json.RawMessage(`{"type":"object","additionalProperties":false}`)}}
|
|
caps.ExecutionProvider = &contract.ExecutionProvider{Tools: []string{"run"}, StatusTool: "health"}
|
|
}
|
|
if mask&4 != 0 {
|
|
caps.HostOperations = &contract.HostOperations{Required: []string{"workspace.materialize"}, Optional: []string{"egress.check"}}
|
|
}
|
|
v := contract.VersionRange{Min: 1, Max: 1}
|
|
manifest := contract.Manifest{SchemaVersion: 1, ID: "sample", Version: "1.0.0", Executable: "bin/sample", OS: "linux", Arch: "amd64", Host: v, API: contract.VersionRange{Min: 2, Max: 2}, Protocol: v, DataFormat: v, Capabilities: caps}
|
|
b, e := json.Marshal(manifest)
|
|
Expect(e).NotTo(HaveOccurred())
|
|
parsed, e := contract.ParseManifest(b)
|
|
Expect(e).NotTo(HaveOccurred())
|
|
Expect(parsed.Capabilities).To(Equal(caps))
|
|
original := Startup{Registration: Registration{AddonID: "sample", Generation: 7, Incarnation: "inc", Protocol: 1, Capabilities: parsed.Capabilities}, Credential: "secret", Limits: contract.DefaultLimits()}
|
|
b, e = json.Marshal(original)
|
|
Expect(e).NotTo(HaveOccurred())
|
|
var startup Startup
|
|
Expect(json.Unmarshal(b, &startup)).To(Succeed())
|
|
req := httptest.NewRequest(http.MethodPost, "/health", bytes.NewBufferString(`{}`))
|
|
req.Header.Set("Authorization", "Bearer secret")
|
|
req.Header.Set("X-LocalAI-Incarnation", "inc")
|
|
req.Header.Set("Content-Type", "application/json")
|
|
rec := httptest.NewRecorder()
|
|
Handler(startup, nil).ServeHTTP(rec, req)
|
|
Expect(rec.Code).To(Equal(http.StatusOK))
|
|
var echo Registration
|
|
Expect(json.Unmarshal(rec.Body.Bytes(), &echo)).To(Succeed())
|
|
Expect(echo).To(Equal(original.Registration))
|
|
if mask == 0 {
|
|
for _, key := range []string{"homeCommands", "executionProvider", "hostOperations", "modelActions"} {
|
|
Expect(rec.Body.String()).NotTo(ContainSubstring(key))
|
|
}
|
|
}
|
|
}
|
|
})
|
|
})
|