From 9cd1f3d8813f0142758be03031592d79aaea3aa1 Mon Sep 17 00:00:00 2001
From: Andrey Nering <andreynering@users.noreply.github.com>
Date: Mon, 30 Mar 2026 10:40:11 -0300
Subject: [PATCH] refactor(venice): remove unused fields + cleanup

---
 cmd/venice/main.go                     | 25 ++++++++-----------------
 internal/providers/configs/venice.json |  2 +-
 2 files changed, 9 insertions(+), 18 deletions(-)

diff --git a/cmd/venice/main.go b/cmd/venice/main.go
index 8260749d9c08652d94f16a293f7c2bb42f9cb29d..c49b63f69443dd1a8cefcec24d686a0f6f62a626 100644
--- a/cmd/venice/main.go
+++ b/cmd/venice/main.go
@@ -23,11 +23,8 @@ type ModelsResponse struct {
 }
 
 type VeniceModel struct {
-	Created   int64           `json:"created"`
 	ID        string          `json:"id"`
 	ModelSpec VeniceModelSpec `json:"model_spec"`
-	Object    string          `json:"object"`
-	OwnedBy   string          `json:"owned_by"`
 	Type      string          `json:"type"`
 }
 
@@ -37,23 +34,17 @@ type VeniceModelSpec struct {
 	Capabilities           VeniceModelCapabilities `json:"capabilities"`
 	Constraints            VeniceModelConstraints  `json:"constraints"`
 	Name                   string                  `json:"name"`
-	ModelSource            string                  `json:"modelSource"`
 	Offline                bool                    `json:"offline"`
 	Pricing                VeniceModelPricing      `json:"pricing"`
-	Traits                 []string                `json:"traits"`
 	Beta                   bool                    `json:"beta"`
 }
 
 type VeniceModelCapabilities struct {
-	OptimizedForCode        bool   `json:"optimizedForCode"`
-	Quantization            string `json:"quantization"`
-	SupportsFunctionCalling bool   `json:"supportsFunctionCalling"`
-	SupportsReasoning       bool   `json:"supportsReasoning"`
-	SupportsReasoningEffort bool   `json:"supportsReasoningEffort"`
-	SupportsResponseSchema  bool   `json:"supportsResponseSchema"`
-	SupportsVision          bool   `json:"supportsVision"`
-	SupportsWebSearch       bool   `json:"supportsWebSearch"`
-	SupportsLogProbs        bool   `json:"supportsLogProbs"`
+	OptimizedForCode        bool `json:"optimizedForCode"`
+	SupportsFunctionCalling bool `json:"supportsFunctionCalling"`
+	SupportsReasoning       bool `json:"supportsReasoning"`
+	SupportsReasoningEffort bool `json:"supportsReasoningEffort"`
+	SupportsVision          bool `json:"supportsVision"`
 }
 
 type VeniceModelConstraints struct {
@@ -71,8 +62,7 @@ type VeniceModelPricing struct {
 }
 
 type VeniceModelPricingValue struct {
-	USD  float64 `json:"usd"`
-	Diem float64 `json:"diem"`
+	USD float64 `json:"usd"`
 }
 
 func fetchVeniceModels(apiEndpoint string) (*ModelsResponse, error) {
@@ -161,7 +151,7 @@ func main() {
 		Models:      []catwalk.Model{},
 	}
 
-	codeOptimizedModels := []catwalk.Model{}
+	var codeOptimizedModels []catwalk.Model
 
 	modelsResp, err := fetchVeniceModels(veniceProvider.APIEndpoint)
 	if err != nil {
@@ -252,6 +242,7 @@ func main() {
 	if err != nil {
 		log.Fatal("Error marshaling Venice provider:", err)
 	}
+	data = append(data, '\n')
 
 	if err := os.WriteFile("internal/providers/configs/venice.json", data, 0o600); err != nil {
 		log.Fatal("Error writing Venice provider config:", err)
diff --git a/internal/providers/configs/venice.json b/internal/providers/configs/venice.json
index b5872862e9ae677cd23e43e239891bec55135bc9..3f78f2ec1610d234e36125bfcf75f7ecf84d500a 100644
--- a/internal/providers/configs/venice.json
+++ b/internal/providers/configs/venice.json
@@ -696,4 +696,4 @@
       "options": {}
     }
   ]
-}
\ No newline at end of file
+}