fix(adapters): emit correct provider/model schema per harness

2026-10-12 00:45:22 +00:00 · 2026-05-23 12:48:10 +01:00
parent 941b1d86c3
commit da49624b99
15 changed files with 556 additions and 46 deletions
@@ -1,7 +1,14 @@
 // Package common provides shared helpers for harness adapters.
 package common

-import "github.com/lukaszraczylo/harness-sync/internal/canonical"
+import (
+	"encoding/json"
+	"strings"
+
+	"github.com/lukaszraczylo/harness-sync/internal/canonical"
+)
+
+const gatewayProviderID = "harness-sync-gateway"

 // ProviderEntry is a generic provider record. Adapters may massage further.
 type ProviderEntry = map[string]any
@@ -13,7 +20,7 @@ func BuildProviders(p *canonical.Profile) []ProviderEntry {
 	out := make([]ProviderEntry, 0, len(p.Upstreams)+1)
 	if p.Gateway.URL != "" {
 		gw := ProviderEntry{
-			"id":       "harness-sync-gateway",
+			"id":       gatewayProviderID,
 			"name":     "harness-sync gateway",
 			"base_url": p.Gateway.URL,
 			"api_key":  p.Gateway.Token,
@@ -48,30 +55,81 @@ func buildModels(models []canonical.Model) []map[string]any {
 	return out
 }

+// crushModel builds a 10-field model entry required by the crush schema.
+func crushModel(id string) map[string]any {
+	return map[string]any{
+		"id":                     id,
+		"name":                   id,
+		"cost_per_1m_in":         0,
+		"cost_per_1m_out":        0,
+		"cost_per_1m_in_cached":  0,
+		"cost_per_1m_out_cached": 0,
+		"context_window":         200000,
+		"default_max_tokens":     8192,
+		"can_reason":             false,
+		"supports_attachments":   true,
+	}
+}
+
+// ProvidersAsCrushMap returns a map[providerID]providerObject for the crush
+// harness. Each provider has type, base_url, api_key, and models[].
+// models[] entries carry all 10 required crush schema fields.
+func ProvidersAsCrushMap(p *canonical.Profile) map[string]any {
+	out := map[string]any{}
+	if p.Gateway.URL == "" {
+		return out
+	}
+	models := make([]map[string]any, 0)
+	for _, m := range p.Models {
+		models = append(models, crushModel(m.ID))
+	}
+	if len(models) == 0 && p.Gateway.DefaultModel != "" {
+		models = append(models, crushModel(p.Gateway.DefaultModel))
+	}
+	out[gatewayProviderID] = map[string]any{
+		"type":     "openai-compat",
+		"base_url": p.Gateway.URL,
+		"api_key":  p.Gateway.Token,
+		"models":   models,
+	}
+	return out
+}
+
+// CrushRoleModels returns the top-level models map for crush with large, small,
+// and title roles all pointing at the gateway provider + default model.
+func CrushRoleModels(p *canonical.Profile) map[string]any {
+	if p.Gateway.DefaultModel == "" {
+		return map[string]any{}
+	}
+	sel := map[string]any{
+		"model":    p.Gateway.DefaultModel,
+		"provider": gatewayProviderID,
+	}
+	return map[string]any{
+		"large": sel,
+		"small": sel,
+		"title": sel,
+	}
+}
+
 // ProvidersAsMap returns a name-keyed map of provider objects for harnesses
-// like opencode that use `provider` (singular, map-shaped) instead of a slice.
-// The gateway entry is keyed "harness-sync-gateway"; upstreams use their name.
+// like opencode/kilo that use `provider` (singular, map-shaped) instead of a
+// slice. Each entry has npm + options.{baseURL,apiKey} per the opencode schema.
 func ProvidersAsMap(p *canonical.Profile) map[string]any {
 	out := map[string]any{}
 	if p.Gateway.URL != "" {
-		entry := map[string]any{
-			"name": "harness-sync gateway",
+		out[gatewayProviderID] = map[string]any{
+			"npm": "@ai-sdk/openai-compatible",
 			"options": map[string]any{
 				"baseURL": p.Gateway.URL,
 				"apiKey":  p.Gateway.Token,
 			},
 		}
-		if models := buildModels(p.Models); len(models) > 0 {
-			modelsMap := map[string]any{}
-			for _, m := range p.Models {
-				modelsMap[m.ID] = map[string]any{"name": m.ID}
-			}
-			entry["models"] = modelsMap
-		}
-		out["harness-sync-gateway"] = entry
 	}
 	for _, up := range p.Upstreams {
-		entry := map[string]any{"name": up.Name}
+		entry := map[string]any{
+			"npm": "@ai-sdk/openai-compatible",
+		}
 		opts := map[string]any{}
 		if up.BaseURL != "" {
 			opts["baseURL"] = up.BaseURL
@@ -87,6 +145,121 @@ func ProvidersAsMap(p *canonical.Profile) map[string]any {
 	return out
 }

+// KiloModelString returns the "providerID/modelID" string used by kilo/opencode.
+func KiloModelString(p *canonical.Profile) string {
+	if p.Gateway.DefaultModel == "" {
+		return ""
+	}
+	return gatewayProviderID + "/" + p.Gateway.DefaultModel
+}
+
+// ProvidersAsCagentMap returns the providers block for cagent:
+// map[providerID]{base_url, token_key, provider}.
+func ProvidersAsCagentMap(p *canonical.Profile) map[string]any {
+	out := map[string]any{}
+	if p.Gateway.URL == "" {
+		return out
+	}
+	out[gatewayProviderID] = map[string]any{
+		"base_url":  p.Gateway.URL,
+		"token_key": "HARNESS_SYNC_GATEWAY_TOKEN",
+		"provider":  "openai",
+	}
+	return out
+}
+
+// ZedLanguageModels returns the language_models.openai block for Zed when a
+// gateway URL is configured.
+func ZedLanguageModels(p *canonical.Profile) map[string]any {
+	if p.Gateway.URL == "" {
+		return map[string]any{}
+	}
+	models := make([]map[string]any, 0)
+	for _, m := range p.Models {
+		models = append(models, map[string]any{
+			"name": m.ID,
+			"display_name": func() string {
+				if m.Alias != "" {
+					return m.Alias
+				}
+				return m.ID
+			}(),
+			"max_tokens": 8192,
+		})
+	}
+	if len(models) == 0 && p.Gateway.DefaultModel != "" {
+		models = append(models, map[string]any{
+			"name":         p.Gateway.DefaultModel,
+			"display_name": p.Gateway.DefaultModel,
+			"max_tokens":   8192,
+		})
+	}
+	return map[string]any{
+		"openai": map[string]any{
+			"api_url":          p.Gateway.URL,
+			"available_models": models,
+		},
+	}
+}
+
+// GooseCustomProviderFile returns the JSON body and provider ID for the goose
+// custom_providers/<id>.json file. Returns nil body when Gateway.URL is empty.
+func GooseCustomProviderFile(p *canonical.Profile) ([]byte, string) {
+	if p.Gateway.URL == "" {
+		return nil, ""
+	}
+	providerID := gatewayProviderID
+	providerName := "custom_" + providerID
+
+	models := make([]map[string]any, 0)
+	for _, m := range p.Models {
+		models = append(models, map[string]any{
+			"name":          m.ID,
+			"context_limit": 200000,
+		})
+	}
+	if len(models) == 0 && p.Gateway.DefaultModel != "" {
+		// strip optional "provider/" prefix from DefaultModel
+		modelID := p.Gateway.DefaultModel
+		if idx := strings.IndexByte(modelID, '/'); idx >= 0 {
+			modelID = modelID[idx+1:]
+		}
+		models = append(models, map[string]any{
+			"name":          modelID,
+			"context_limit": 200000,
+		})
+	}
+
+	// display_name: hostname of the gateway URL, or just providerID.
+	displayName := providerID
+	if idx := strings.Index(p.Gateway.URL, "://"); idx >= 0 {
+		rest := p.Gateway.URL[idx+3:]
+		if slash := strings.IndexByte(rest, '/'); slash >= 0 {
+			rest = rest[:slash]
+		}
+		if rest != "" {
+			displayName = rest
+		}
+	}
+
+	chatURL := strings.TrimRight(p.Gateway.URL, "/") + "/v1/chat/completions"
+
+	entry := map[string]any{
+		"name":          providerName,
+		"engine":        "openai",
+		"display_name":  displayName,
+		"base_url":      chatURL,
+		"models":        models,
+		"requires_auth": true,
+		"api_key_env":   "CUSTOM_HARNESS_SYNC_GATEWAY_API_KEY",
+	}
+	body, err := json.MarshalIndent(entry, "", "  ")
+	if err != nil {
+		return nil, ""
+	}
+	return body, providerName
+}
+
 // BuildMCPMap returns canonical MCP servers as a name -> object map.
 // Empty map (not nil) is returned when there are no servers so the caller
 // can decide whether to include the key.