test message

Author: Patel230

cmd/iterate/config.go

@@ -27,6 +27,8 @@ type iterConfig struct {
 	MaxTokens     int     `json:"max_tokens,omitempty"     toml:"max_tokens"`
 	ThinkingLevel string  `json:"thinking_level,omitempty" toml:"thinking_level"`
 	CacheEnabled  bool    `json:"cache_enabled,omitempty"  toml:"cache_enabled"`
+	// Request timeout in seconds (0 = default 120s).
+	RequestTimeout int `json:"request_timeout,omitempty" toml:"request_timeout"`
 	// Glob-based allow/deny patterns for bash commands.
 	AllowPatterns []string `json:"allow_patterns,omitempty" toml:"allow_patterns"`
 	DenyPatterns  []string `json:"deny_patterns,omitempty"  toml:"deny_patterns"`

cmd/iterate/features_sessions.go

@@ -60,6 +60,10 @@ func saveSession(name string, messages []iteragent.Message) error {
 	if err != nil {
 		return err
 	}
+	// Write .bak backup of previous version before overwriting.
+	if existing, err := os.ReadFile(path); err == nil {
+		_ = atomicWriteFile(path+".bak", existing, 0o644)
+	}
 	return atomicWriteFile(path, data, 0o644)
 }
 
@@ -71,7 +75,20 @@ func loadSession(name string) ([]iteragent.Message, error) {
 	}
 	var msgs []iteragent.Message
 	if err := json.Unmarshal(data, &msgs); err != nil {
-		return nil, err
+		// Try .bak file if primary is corrupt.
+		if bakData, bakErr := os.ReadFile(path + ".bak"); bakErr == nil {
+			var bakMsgs []iteragent.Message
+			if json.Unmarshal(bakData, &bakMsgs) == nil {
+				return bakMsgs, nil
+			}
+		}
+		return nil, fmt.Errorf("session file corrupt: %w", err)
+	}
+	// Integrity check: each message must have a non-empty role.
+	for i, m := range msgs {
+		if m.Role == "" {
+			return nil, fmt.Errorf("session integrity check failed: message %d has empty role", i)
+		}
 	}
 	return msgs, nil
 }

cmd/iterate/features_tools.go

@@ -24,6 +24,10 @@ import (
 // Tool wrappers wait for it to reach 0 before showing a prompt.
 var spinnerActive atomic.Int32
 
+// streamingTokenCount is incremented for each token received during streaming.
+// The spinner reads this to display tok/s.
+var streamingTokenCount atomic.Int64
+
 // deniedTools is the set of tools blocked in safe mode.
 var deniedTools = map[string]bool{
 	"bash":       true,

cmd/iterate/fileutil_test.go

@@ -0,0 +1,79 @@
+package main
+
+import (
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+func TestAtomicWriteFile_BasicWrite(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "test.json")
+	data := []byte(`{"key":"value"}`)
+
+	if err := atomicWriteFile(path, data, 0o644); err != nil {
+		t.Fatalf("atomicWriteFile failed: %v", err)
+	}
+
+	got, err := os.ReadFile(path)
+	if err != nil {
+		t.Fatalf("ReadFile failed: %v", err)
+	}
+	if string(got) != string(data) {
+		t.Errorf("want %q, got %q", data, got)
+	}
+}
+
+func TestAtomicWriteFile_Overwrite(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "test.json")
+
+	// Write initial content.
+	if err := atomicWriteFile(path, []byte("original"), 0o644); err != nil {
+		t.Fatalf("initial write failed: %v", err)
+	}
+
+	// Overwrite with new content.
+	if err := atomicWriteFile(path, []byte("updated"), 0o644); err != nil {
+		t.Fatalf("overwrite failed: %v", err)
+	}
+
+	got, _ := os.ReadFile(path)
+	if string(got) != "updated" {
+		t.Errorf("want %q, got %q", "updated", got)
+	}
+}
+
+func TestAtomicWriteFile_NoTempFilesLeft(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "test.json")
+
+	if err := atomicWriteFile(path, []byte("data"), 0o644); err != nil {
+		t.Fatalf("atomicWriteFile failed: %v", err)
+	}
+
+	// No .tmp- files should remain in the directory.
+	entries, _ := os.ReadDir(dir)
+	for _, e := range entries {
+		if e.Name() != "test.json" {
+			t.Errorf("unexpected file left behind: %s", e.Name())
+		}
+	}
+}
+
+func TestAtomicWriteFile_PermissionSet(t *testing.T) {
+	dir := t.TempDir()
+	path := filepath.Join(dir, "test.json")
+
+	if err := atomicWriteFile(path, []byte("data"), 0o600); err != nil {
+		t.Fatalf("atomicWriteFile failed: %v", err)
+	}
+
+	info, err := os.Stat(path)
+	if err != nil {
+		t.Fatalf("stat failed: %v", err)
+	}
+	if info.Mode().Perm() != 0o600 {
+		t.Errorf("want perm 0600, got %04o", info.Mode().Perm())
+	}
+}

cmd/iterate/provider.go

@@ -5,6 +5,7 @@ import (
 	"os"
 
 	iteragent "github.com/GrayCodeAI/iteragent"
+	"github.com/GrayCodeAI/iterate/internal/ui/selector"
 )
 
 // resolveProviderConfig merges flag values with persisted config.
@@ -43,11 +44,13 @@ func resolveThinkingLevel(flagThinking string, cfg iterConfig) string {
 }
 
 // initProvider creates an LLM provider from the given name and API key.
+// It also wires the provider's context window into the selector for display.
 func initProvider(providerName, apiKey string, logger *slog.Logger) (iteragent.Provider, error) {
 	p, err := iteragent.NewProvider(providerName, apiKey)
 	if err != nil {
 		return nil, err
 	}
 	logger.Info("using provider", "name", p.Name())
+	selector.ContextWindow = iteragent.ProviderContextWindow(p)
 	return p, nil
 }

cmd/iterate/repl.go

@@ -60,13 +60,19 @@ func makeAgent(p iteragent.Provider, repoPath string, thinking iteragent.Thinkin
 	tools := wrapToolsWithPermissions(base)
 	skills, _ := iteragent.LoadSkills([]string{filepath.Join(repoPath, "skills")})
 	defaultTemp := float32(0.9)
+	ctxCfg := iteragent.DefaultContextConfig()
+	if cw := iteragent.ProviderContextWindow(p); cw > 0 {
+		// Use 80% of the provider's actual context window as the compaction threshold.
+		ctxCfg.MaxTokens = cw * 8 / 10
+	}
 	ag := iteragent.New(p, tools, logger).
 		WithSystemPrompt(replSystemPrompt(repoPath)).
 		WithSkillSet(skills).
 		WithThinkingLevel(thinking).
 		WithToolExecutionStrategy(iteragent.NewParallelStrategy()).
 		WithHooks(replHooks()).
-		WithTemperature(defaultTemp)
+		WithTemperature(defaultTemp).
+		WithContextConfig(ctxCfg)
 	if rtConfig.Temperature != nil {
 		ag = ag.WithTemperature(*rtConfig.Temperature)
 	}
@@ -140,6 +146,7 @@ func setupSigintHandler() {
 func applyLoadedConfig(loadedCfg iterConfig, thinking iteragent.ThinkingLevel) iteragent.ThinkingLevel {
 	cfg.SafeMode = loadedCfg.SafeMode
 	cfg.NotifyEnabled = loadedCfg.Notify
+	cfg.RequestTimeout = loadedCfg.RequestTimeout
 	if loadedCfg.Theme != "" {
 		if t, ok := themes[loadedCfg.Theme]; ok {
 			applyTheme(t)
@@ -276,6 +283,7 @@ func handleModelProviderSwitch(line string, p *iteragent.Provider, thinking *ite
 				closeProvider(*p)
 				*p = newP
 				os.Setenv("ITERATE_PROVIDER", providerName)
+				selector.ContextWindow = iteragent.ProviderContextWindow(newP)
 				_ = (*a).Close() // best-effort cleanup
 				*a = makeAgent(*p, repoPath, *thinking, logger)
 				fmt.Printf("%s✓ switched to %s%s\n\n", colorLime, (*p).Name(), colorReset)
@@ -293,13 +301,20 @@ func printSessionSummary(a *iteragent.Agent, repoPath string) {
 		_ = saveSession("autosave", a.Messages) // best-effort cleanup
 	}
 	fmt.Println()
-	fmt.Printf("%s  session:%s %s%s%s %s·%s %s%d messages%s %s·%s %s%d tokens%s\n",
+	costStr := ""
+	if sess.CostUSD > 0 {
+		costStr = fmt.Sprintf(" %s·%s %s%s%s",
+			colorDim, colorReset,
+			colorLime, formatSessionCost(sess.CostUSD), colorReset)
+	}
+	fmt.Printf("%s  session:%s %s%s%s %s·%s %s%d messages%s %s·%s %s%d tokens%s%s\n",
 		colorDim, colorReset,
 		colorCyan, elapsed, colorReset,
 		colorDim, colorReset,
 		colorDim, sess.Messages, colorReset,
 		colorDim, colorReset,
-		colorPurple, sess.InputTokens+sess.OutputTokens, colorReset)
+		colorPurple, sess.InputTokens+sess.OutputTokens, colorReset,
+		costStr)
 	if len(a.Messages) > 0 {
 		fmt.Printf("%s  autosaved · /load autosave to restore%s\n", colorDim, colorReset)
 	}

cmd/iterate/repl_helpers.go

@@ -140,8 +140,69 @@ func replSystemPrompt(repoPath string) string {
 		base += "\n## Project Context (ITERATE.md)\n" + string(iterateMD)
 	}
 
+	// Inject AGENTS.md if present (OpenAI Codex / Claude convention)
+	if agentsMD, err := os.ReadFile(filepath.Join(repoPath, "AGENTS.md")); err == nil {
+		base += "\n## Agent Instructions (AGENTS.md)\n" + string(agentsMD)
+	}
+
+	// Inject detected project language/framework info
+	if lang := detectProjectStack(repoPath); lang != "" {
+		base += "\n## Project Stack\n" + lang + "\n"
+	}
+
 	if index := buildRepoIndex(repoPath); index != "" {
 		base += "\n## Repo structure\n```\n" + index + "\n```\n"
 	}
 	return base
 }
+
+// detectProjectStack detects the primary language and frameworks used in the repo.
+func detectProjectStack(repoPath string) string {
+	var tags []string
+
+	check := func(path string) bool {
+		_, err := os.Stat(filepath.Join(repoPath, path))
+		return err == nil
+	}
+
+	// Go
+	if check("go.mod") {
+		tags = append(tags, "Go")
+	}
+	// Node / JS
+	if check("package.json") {
+		if check("tsconfig.json") {
+			tags = append(tags, "TypeScript")
+		} else {
+			tags = append(tags, "JavaScript/Node.js")
+		}
+		if check("next.config.js") || check("next.config.ts") || check("next.config.mjs") {
+			tags = append(tags, "Next.js")
+		}
+	}
+	// Python
+	if check("pyproject.toml") || check("requirements.txt") || check("setup.py") {
+		tags = append(tags, "Python")
+	}
+	// Rust
+	if check("Cargo.toml") {
+		tags = append(tags, "Rust")
+	}
+	// Ruby
+	if check("Gemfile") {
+		tags = append(tags, "Ruby")
+	}
+	// Java / Kotlin
+	if check("pom.xml") || check("build.gradle") || check("build.gradle.kts") {
+		tags = append(tags, "JVM (Java/Kotlin)")
+	}
+	// Docker
+	if check("Dockerfile") || check("docker-compose.yml") || check("docker-compose.yaml") {
+		tags = append(tags, "Docker")
+	}
+
+	if len(tags) == 0 {
+		return ""
+	}
+	return strings.Join(tags, ", ")
+}

cmd/iterate/repl_models.go

@@ -45,6 +45,7 @@ func selectModel(currentThinking iteragent.ThinkingLevel) (iteragent.Provider, i
 		return nil, currentThinking
 	}
 	os.Setenv("ITERATE_PROVIDER", providerName)
+	selector.ContextWindow = iteragent.ProviderContextWindow(newP)
 	return newP, currentThinking
 }
 
@@ -74,6 +75,7 @@ func selectOllamaModel(currentThinking iteragent.ThinkingLevel) (iteragent.Provi
 		fmt.Printf("%serror: %s%s\n\n", colorRed, err, colorReset)
 		return nil, currentThinking
 	}
+	selector.ContextWindow = iteragent.ProviderContextWindow(p)
 	return p, currentThinking
 }