diff --git a/cmd/freeride-diag/main.go b/cmd/freeride-diag/main.go index ecd204adb..5bf51e5a0 100644 --- a/cmd/freeride-diag/main.go +++ b/cmd/freeride-diag/main.go @@ -20,9 +20,9 @@ type Model struct { Prompt string `json:"prompt"` Completion string `json:"completion"` } `json:"pricing"` - Created int64 `json:"created"` - Score float64 - LastError string + Created int64 `json:"created"` + Score float64 + LastError string IsReachable bool } @@ -70,7 +70,7 @@ func main() { } m := &freeModels[i] fmt.Printf("[%d/%d] Testing %s... ", i+1, len(freeModels), m.ID) - + err := testModel(apiKey, m.ID) if err == nil { m.IsReachable = true @@ -113,7 +113,7 @@ func main() { func fetchModels(apiKey string) ([]Model, error) { req, _ := http.NewRequest("GET", "https://openrouter.ai/api/v1/models", nil) req.Header.Set("Authorization", "Bearer "+apiKey) - + resp, err := http.DefaultClient.Do(req) if err != nil { return nil, err @@ -138,11 +138,11 @@ func testModel(apiKey, modelID string) error { "max_tokens": 10, } body, _ := json.Marshal(payload) - + req, _ := http.NewRequest("POST", "https://openrouter.ai/api/v1/chat/completions", bytes.NewReader(body)) req.Header.Set("Authorization", "Bearer "+apiKey) req.Header.Set("Content-Type", "application/json") - + client := &http.Client{Timeout: 10 * time.Second} resp, err := client.Do(req) if err != nil { @@ -154,6 +154,6 @@ func testModel(apiKey, modelID string) error { respBody, _ := io.ReadAll(resp.Body) return fmt.Errorf("HTTP %d: %s", resp.StatusCode, string(respBody)) } - + return nil } diff --git a/pkg/agent/loop.go b/pkg/agent/loop.go index dc49f7979..0051c761e 100644 --- a/pkg/agent/loop.go +++ b/pkg/agent/loop.go @@ -153,15 +153,15 @@ func NewAgentLoop( eventBus := NewEventBus() al := &AgentLoop{ - bus: msgBus, - cfg: cfg, - configPath: configPath, - registry: registry, - state: stateManager, - eventBus: eventBus, - fallback: fallbackChain, - cmdRegistry: commands.NewRegistry(commands.BuiltinDefinitions()), - steering: newSteeringQueue(parseSteeringMode(cfg.Agents.Defaults.SteeringMode)), + bus: msgBus, + cfg: cfg, + configPath: configPath, + registry: registry, + state: stateManager, + eventBus: eventBus, + fallback: fallbackChain, + cmdRegistry: commands.NewRegistry(commands.BuiltinDefinitions()), + steering: newSteeringQueue(parseSteeringMode(cfg.Agents.Defaults.SteeringMode)), } al.agentCacheTTL = 24 * time.Hour diff --git a/pkg/tools/freeride.go b/pkg/tools/freeride.go index b7e2d0c54..09a6f3ba2 100644 --- a/pkg/tools/freeride.go +++ b/pkg/tools/freeride.go @@ -240,7 +240,7 @@ func (t *FreeRideTool) handleAuto(ctx context.Context, limit int) *ToolResult { msg := fmt.Sprintf("Success! Added %d free models as fallbacks: %s.\n", len(addedModels), strings.Join(addedModels, ", ")) msg += "Re-loading configuration to apply changes..." - + if t.reloadFunc != nil { if err := t.reloadFunc(); err != nil { return ErrorResult(fmt.Sprintf("%s\nFailed to reload: %v", msg, err)) diff --git a/pkg/tools/freeride_test.go b/pkg/tools/freeride_test.go index a5769db8b..6ff24ad4d 100644 --- a/pkg/tools/freeride_test.go +++ b/pkg/tools/freeride_test.go @@ -86,7 +86,7 @@ func TestFreeRideTool_Auto(t *testing.T) { ModelList: []*config.ModelConfig{}, } initialCfg.Agents.Defaults.ModelName = "existing-model" - + if err := config.SaveConfig(configPath, initialCfg); err != nil { t.Fatalf("failed to save initial config: %v", err) }