v0.0.1: agentu mvp

Add OpenAI-compatible providers, TUI, slash commands, and model switching. Add local file tools plus optional yolo shell execution. Document config, usage, and development workflow.
2026-06-22 17:28:22 +08:00
commit a431d1ec95
23 changed files with 3526 additions and 0 deletions
@@ -0,0 +1,129 @@
+package main
+
+import (
+	"bufio"
+	"context"
+	"errors"
+	"flag"
+	"fmt"
+	"net/http"
+	"os"
+	"os/signal"
+	"strings"
+
+	"agentu/internal/agent"
+	"agentu/internal/config"
+	"agentu/internal/llm"
+	"agentu/internal/session"
+	"agentu/internal/tools"
+	"agentu/internal/tui"
+)
+
+func main() {
+	if err := run(); err != nil {
+		fmt.Fprintln(os.Stderr, "agentu:", err)
+		os.Exit(1)
+	}
+}
+
+func run() error {
+	configPath := flag.String("config", config.DefaultPath, "path to YAML config")
+	yolo := flag.Bool("yolo", false, "also enable automatic file writes and shell execution")
+	plain := flag.Bool("plain", false, "use the simple line-based REPL instead of the TUI")
+	themeName := flag.String("theme", "light", "TUI theme: light or dark")
+	flag.Parse()
+
+	themeMode, err := tui.ParseThemeMode(*themeName)
+	if err != nil {
+		return err
+	}
+
+	cfg, err := config.Load(*configPath)
+	if err != nil {
+		if errors.Is(err, os.ErrNotExist) {
+			resolvedPath, resolveErr := config.ResolvePath(*configPath)
+			if resolveErr != nil {
+				return resolveErr
+			}
+			return fmt.Errorf("config not found at %s; create ~/.agentu/config.yaml from agentu.example.yaml and set AGENTU_API_KEY", resolvedPath)
+		}
+		return err
+	}
+
+	providerName := cfg.DefaultProviderName()
+	providerConfig := cfg.Providers[providerName]
+	provider := llm.NewOpenAICompatibleClient(providerConfig.BaseURL, providerConfig.APIKey, http.DefaultClient)
+
+	var registry *tools.Registry
+	if *yolo {
+		registry = tools.Builtins(cfg.Agent.WorkingDir)
+	} else {
+		registry = tools.ReadOnlyBuiltins(cfg.Agent.WorkingDir)
+	}
+	systemPrompt := strings.TrimSpace(cfg.Agent.SystemPrompt + "\n\n" + tools.AgentInstructions(cfg.Agent.WorkingDir, *yolo))
+
+	assistant := agent.New(agent.Options{
+		Provider:        provider,
+		ProviderName:    providerName,
+		Model:           providerConfig.Model,
+		Thinking:        providerConfig.Thinking,
+		ThinkingParam:   providerConfig.ThinkingParam,
+		ThinkingEnabled: providerConfig.ThinkingConfigured,
+		SystemPrompt:    systemPrompt,
+		ToolRegistry:    registry,
+		ToolTimeout:     cfg.Agent.ToolTimeout,
+	})
+	modelManager, err := session.NewManager(cfg, assistant, http.DefaultClient)
+	if err != nil {
+		return err
+	}
+
+	ctx, stop := signal.NotifyContext(context.Background(), os.Interrupt)
+	defer stop()
+
+	if !*plain {
+		return tui.Run(ctx, assistant, tui.Options{
+			ModelName:    providerConfig.Model,
+			Yolo:         *yolo,
+			ThemeMode:    themeMode,
+			ModelManager: modelManager,
+		})
+	}
+
+	return repl(ctx, assistant)
+}
+
+func repl(ctx context.Context, assistant *agent.Agent) error {
+	scanner := bufio.NewScanner(os.Stdin)
+	scanner.Buffer(make([]byte, 0, 64*1024), 1024*1024)
+
+	fmt.Println("agentu")
+	fmt.Println("Type /exit to quit, /clear to reset context.")
+	for {
+		fmt.Print("> ")
+		if !scanner.Scan() {
+			if err := scanner.Err(); err != nil {
+				return err
+			}
+			fmt.Println()
+			return nil
+		}
+
+		input := strings.TrimSpace(scanner.Text())
+		switch input {
+		case "":
+			continue
+		case "/exit", "/quit":
+			return nil
+		case "/clear":
+			assistant.Clear()
+			fmt.Println("context cleared")
+			continue
+		}
+
+		if err := assistant.RunTurn(ctx, input, os.Stdout, os.Stderr); err != nil {
+			fmt.Fprintln(os.Stderr, "\nerror:", err)
+		}
+		fmt.Println()
+	}
+}