28 lines
927 B
Go
28 lines
927 B
Go
package node_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"iop/apps/edge/internal/node"
|
|
"iop/packages/go/config"
|
|
)
|
|
|
|
func TestBuildConfigPayloadProviderOnly(t *testing.T) {
|
|
record := &node.NodeRecord{
|
|
ID: "node", Alias: "node", Token: "token",
|
|
Providers: []config.NodeProviderConf{
|
|
{ID: "ollama-local", Type: "ollama", Category: config.CategoryLocalInference, BaseURL: "http://127.0.0.1:11434", Models: []string{"model"}, Capacity: 1},
|
|
{ID: "api", Type: "openai_api", Category: config.CategoryAPI, Endpoint: "http://127.0.0.1:8000/v1", Models: []string{"model"}, Capacity: 2},
|
|
},
|
|
}
|
|
payload, err := node.BuildConfigPayload(record)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(payload.GetAdapters()) != 2 {
|
|
t.Fatalf("adapters = %d", len(payload.GetAdapters()))
|
|
}
|
|
if payload.GetAdapters()[0].GetType() != "ollama" || payload.GetAdapters()[1].GetType() != "openai_compat" {
|
|
t.Fatalf("adapters = %#v", payload.GetAdapters())
|
|
}
|
|
}
|