Prepare the local backend convenience release
This commit is contained in:
@@ -140,11 +140,18 @@ func TestUnknownProfileBackendHasProfileLoadIdentity(t *testing.T) {
|
||||
}
|
||||
|
||||
func TestLocalBackendConstructsAndRegistersConventionalBackend(t *testing.T) {
|
||||
const limit = 2
|
||||
const (
|
||||
localBackendID = "local"
|
||||
limit = 2
|
||||
)
|
||||
if promptkit.BackendLocal != localBackendID {
|
||||
t.Fatalf("BackendLocal=%q, want %q", promptkit.BackendLocal, localBackendID)
|
||||
}
|
||||
|
||||
endpoint := "http://local.example/v1"
|
||||
backend := promptkit.LocalBackend(endpoint, limit)
|
||||
want := promptkit.Backend{
|
||||
ID: promptkit.BackendLocal,
|
||||
ID: localBackendID,
|
||||
Endpoint: endpoint,
|
||||
ConcurrencyLimit: limit,
|
||||
}
|
||||
@@ -157,7 +164,7 @@ func TestLocalBackendConstructsAndRegistersConventionalBackend(t *testing.T) {
|
||||
promptkit.WithBackend(backend),
|
||||
promptkit.WithProfiles(promptkit.Profile{
|
||||
ID: "local-profile",
|
||||
BackendID: promptkit.BackendLocal,
|
||||
BackendID: localBackendID,
|
||||
Model: "model",
|
||||
}),
|
||||
)
|
||||
@@ -169,7 +176,7 @@ func TestLocalBackendConstructsAndRegistersConventionalBackend(t *testing.T) {
|
||||
if err != nil {
|
||||
t.Fatalf("prepare with local backend: %v", err)
|
||||
}
|
||||
if prepared.SelectedBackendID != promptkit.BackendLocal ||
|
||||
if prepared.SelectedBackendID != localBackendID ||
|
||||
prepared.EffectiveModelParams.Endpoint != endpoint {
|
||||
t.Fatalf("unexpected local backend preparation: %+v", prepared)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user