Changeset 0.9.0 (#50)

server/providers/venice.go

@@ -1,6 +1,8 @@
 package providers
 
 import (
+	"git.gobha.me/xcaliber/chat-switchboard/capabilities"
+	"git.gobha.me/xcaliber/chat-switchboard/models"
 	"context"
 	"encoding/json"
 	"fmt"
@@ -62,29 +64,30 @@ func (p *VeniceProvider) ListModels(ctx context.Context, cfg ProviderConfig) ([]
 		return nil, fmt.Errorf("venice decode models: %w", err)
 	}
 
-	models := make([]Model, 0, len(result.Data))
+	out := make([]Model, 0, len(result.Data))
 	for _, m := range result.Data {
 		spec := m.ModelSpec
 		vcaps := spec.Capabilities
 
-		caps := ModelCapabilities{
+		caps := models.ModelCapabilities{
 			Streaming:     true,
 			ToolCalling:   vcaps.SupportsFunctionCalling,
 			Vision:        vcaps.SupportsVision,
 			Reasoning:     vcaps.SupportsReasoning,
 			WebSearch:     vcaps.SupportsWebSearch,
+			CodeOptimized: vcaps.OptimizedForCode,
 			MaxContext:     spec.AvailableContextTokens,
 		}
 
 		// Venice doesn't report max output tokens directly.
 		// Try known table, then derive from context.
-		if known, ok := LookupKnownModel(m.ID); ok && known.MaxOutputTokens > 0 {
+		if known, ok := capabilities.LookupKnownModel(m.ID); ok && known.MaxOutputTokens > 0 {
 			caps.MaxOutputTokens = known.MaxOutputTokens
 		}
 
-		var pricing *ModelPricing
+		var pricing *models.ModelPricing
 		if spec.Pricing.Input.USD > 0 {
-			pricing = &ModelPricing{
+			pricing = &models.ModelPricing{
 				InputPerM:  spec.Pricing.Input.USD,
 				OutputPerM: spec.Pricing.Output.USD,
 			}
@@ -95,7 +98,7 @@ func (p *VeniceProvider) ListModels(ctx context.Context, cfg ProviderConfig) ([]
 			name = m.ID
 		}
 
-		models = append(models, Model{
+		out = append(out, Model{
 			ID:           m.ID,
 			Name:         name,
 			OwnedBy:      "venice",
@@ -103,7 +106,7 @@ func (p *VeniceProvider) ListModels(ctx context.Context, cfg ProviderConfig) ([]
 			Pricing:      pricing,
 		})
 	}
-	return models, nil
+	return out, nil
 }
 
 // ── Venice Wire Types ───────────────────────
@@ -137,6 +140,7 @@ type veniceCapabilities struct {
 	SupportsResponseSchema  bool `json:"supportsResponseSchema"`
 	SupportsAudioInput      bool `json:"supportsAudioInput"`
 	SupportsLogProbs        bool `json:"supportsLogProbs"`
+	OptimizedForCode        bool `json:"optimizedForCode"`
 }
 
 type venicePricing struct {