Compare commits

...

6 Commits

Author SHA1 Message Date
imfozilbek
9c94335729 feat(ipuaro): add commands configuration
- Add CommandsConfigSchema with timeout option
- Integrate timeout configuration in RunCommandTool
- Add 22 new unit tests (19 schema + 3 integration)
- Complete v0.22.0 Extended Configuration milestone
2025-12-02 03:03:57 +05:00
imfozilbek
c34d57c231 chore(ipuaro): release v0.22.4 2025-12-02 02:29:56 +05:00
imfozilbek
60052c0db9 feat(ipuaro): add autocomplete configuration
- Add AutocompleteConfigSchema with enabled, source, maxSuggestions
- Update useAutocomplete hook to read from config
- Add 27 unit tests for autocomplete config
- Fix unused variable in Chat component
- Update ROADMAP and CHANGELOG
2025-12-02 02:26:36 +05:00
imfozilbek
fa647c41aa feat(ipuaro): add context configuration
- Add ContextConfigSchema with systemPromptTokens, maxContextUsage, autoCompressAt, compressionMethod
- Update ContextManager to read compression threshold from config
- Update HandleMessage and useSession to pass context config
- Add 40 unit tests (32 schema + 8 integration)
- Coverage: 97.63% lines, 91.34% branches
2025-12-02 02:02:34 +05:00
imfozilbek
98b365bd94 chore(ipuaro): release v0.22.2 2025-12-02 01:39:37 +05:00
imfozilbek
a7669f8947 feat(ipuaro): add session configuration
- Add SessionConfigSchema with persistIndefinitely, maxHistoryMessages, saveInputHistory
- Implement Session.truncateHistory() method for limiting message history
- Update HandleMessage to support history truncation and input history toggle
- Add config flow through useSession and App components
- Add 19 unit tests for SessionConfigSchema
- Update CHANGELOG.md and ROADMAP.md for v0.22.2
2025-12-02 01:34:04 +05:00
30 changed files with 1217 additions and 103 deletions

View File

@@ -5,6 +5,213 @@ All notable changes to this project will be documented in this file.
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
## [0.22.5] - 2025-12-02 - Commands Configuration
### Added
- **CommandsConfigSchema (0.22.5)**
- New configuration schema for command settings in `src/shared/constants/config.ts`
- `timeout: number | null` (default: null) - global timeout for shell commands in milliseconds
- Integrated into main ConfigSchema with `.default({})`
- Exported `CommandsConfig` type from config module
### Changed
- **RunCommandTool**
- Added optional `config?: CommandsConfig` parameter to constructor
- Timeout priority: `params.timeout``config.timeout``DEFAULT_TIMEOUT (30000)`
- Updated parameter description to reflect configuration support
- Config-based timeout enables global command timeout without per-call specification
### Technical Details
- Total tests: 1679 passed (was 1657, +22 new tests)
- New test file: `commands-config.test.ts` with 19 tests
- Default values validation (timeout: null)
- `timeout` nullable positive integer validation (including edge cases: zero, negative, float rejection)
- Partial and full config merging tests
- Updated RunCommandTool tests: 3 new tests for configuration integration
- Config timeout behavior
- Null config timeout fallback to default
- Param timeout priority over config timeout
- Coverage: 97.64% lines, 91.36% branches, 98.77% functions, 97.64% statements
- 0 ESLint errors, 5 warnings (acceptable TUI component warnings)
- Build successful with no TypeScript errors
### Notes
This release completes the v0.22.0 Extended Configuration milestone. All items for v0.22.0 are now complete:
- ✅ 0.22.1 - Display Configuration
- ✅ 0.22.2 - Session Configuration
- ✅ 0.22.3 - Context Configuration
- ✅ 0.22.4 - Autocomplete Configuration
- ✅ 0.22.5 - Commands Configuration
---
## [0.22.4] - 2025-12-02 - Autocomplete Configuration
### Added
- **AutocompleteConfigSchema (0.22.4)**
- New configuration schema for autocomplete settings in `src/shared/constants/config.ts`
- `enabled: boolean` (default: true) - toggle autocomplete feature
- `source: "redis-index" | "filesystem" | "both"` (default: "redis-index") - autocomplete source
- `maxSuggestions: number` (default: 10) - maximum number of suggestions to display
- Integrated into main ConfigSchema with `.default({})`
- Exported `AutocompleteConfig` type from config module
### Changed
- **useAutocomplete Hook**
- Added optional `config?: AutocompleteConfig` parameter to `UseAutocompleteOptions`
- Config priority: `config``props``defaults`
- Reads `enabled` and `maxSuggestions` from config if provided
- Falls back to prop values, then to defaults
- Internal variables renamed: `enabled``isEnabled`, `maxSuggestions``maxSuggestionsCount`
- **Chat Component**
- Fixed ESLint error: removed unused `roleColor` variable in `ToolMessage` component
- Removed unused `theme` parameter from `ToolMessage` function signature
### Technical Details
- Total tests: 1657 passed (was 1630, +27 new tests)
- New test file: `autocomplete-config.test.ts` with 27 tests
- Default values validation (enabled, source, maxSuggestions)
- `enabled` boolean validation
- `source` enum validation ("redis-index", "filesystem", "both")
- `maxSuggestions` positive integer validation (including edge cases: zero, negative, float rejection)
- Partial and full config merging tests
- Coverage: 97.59% lines, 91.23% branches, 98.77% functions, 97.59% statements
- 0 ESLint errors, 5 warnings (acceptable TUI component warnings)
- Build successful with no TypeScript errors
### Notes
This release completes the fourth item (0.22.4) of the v0.22.0 Extended Configuration milestone. Remaining item for v0.22.0:
- 0.22.5 - Commands Configuration
---
## [0.22.3] - 2025-12-02 - Context Configuration
### Added
- **ContextConfigSchema (0.22.3)**
- New configuration schema for context management in `src/shared/constants/config.ts`
- `systemPromptTokens: number` (default: 2000) - token budget for system prompt
- `maxContextUsage: number` (default: 0.8) - maximum context window usage ratio (0-1)
- `autoCompressAt: number` (default: 0.8) - threshold for automatic context compression (0-1)
- `compressionMethod: "llm-summary" | "truncate"` (default: "llm-summary") - compression strategy
- Integrated into main ConfigSchema with `.default({})`
- Exported `ContextConfig` type from config module
### Changed
- **ContextManager**
- Added optional `config?: ContextConfig` parameter to constructor
- Added private `compressionThreshold: number` field (read from config or default)
- Added private `compressionMethod: "llm-summary" | "truncate"` field (read from config or default)
- Updated `needsCompression()` to use configurable `compressionThreshold` instead of hardcoded constant
- Enables dynamic compression threshold configuration per session/deployment
- **HandleMessage Use Case**
- Added optional `contextConfig?: ContextConfig` parameter to constructor
- Added `contextConfig?: ContextConfig` to `HandleMessageOptions`
- Passes context config to ContextManager during initialization
- Context management behavior now fully configurable
- **useSession Hook**
- Passes `deps.config?.context` to HandleMessage constructor
- Passes `contextConfig: deps.config?.context` to HandleMessage options
- Context configuration flows from config through to ContextManager
### Technical Details
- Total tests: 1630 passed (was 1590, +40 new tests)
- New test file: `context-config.test.ts` with 32 tests
- Default values validation (systemPromptTokens, maxContextUsage, autoCompressAt, compressionMethod)
- `systemPromptTokens` positive integer validation (including edge cases: zero, negative, float rejection)
- `maxContextUsage` ratio validation (0-1 range, rejects out-of-bounds)
- `autoCompressAt` ratio validation (0-1 range, rejects out-of-bounds)
- `compressionMethod` enum validation (llm-summary, truncate)
- Partial and full config merging tests
- Updated ContextManager tests: 8 new tests for configuration integration
- Custom compression threshold behavior
- Edge cases: autoCompressAt = 0 and autoCompressAt = 1
- Full context config acceptance
- Coverage: 97.63% lines, 91.34% branches, 98.77% functions, 97.63% statements
- 0 ESLint errors, 0 warnings
- Build successful with no TypeScript errors
### Notes
This release completes the third item (0.22.3) of the v0.22.0 Extended Configuration milestone. Remaining items for v0.22.0:
- 0.22.4 - Autocomplete Configuration
- 0.22.5 - Commands Configuration
---
## [0.22.2] - 2025-12-02 - Session Configuration
### Added
- **SessionConfigSchema (0.22.2)**
- New configuration schema for session settings in `src/shared/constants/config.ts`
- `persistIndefinitely: boolean` (default: true) - toggle indefinite session persistence
- `maxHistoryMessages: number` (default: 100) - maximum number of messages to keep in session history
- `saveInputHistory: boolean` (default: true) - toggle saving user input to history
- Integrated into main ConfigSchema with `.default({})`
- Exported `SessionConfig` type from config module
- **Session.truncateHistory() Method**
- New method in `src/domain/entities/Session.ts`
- Truncates message history to specified maximum length
- Keeps most recent messages when truncating
### Changed
- **HandleMessage Use Case**
- Added `maxHistoryMessages?: number` option to `HandleMessageOptions`
- Added `saveInputHistory?: boolean` option to `HandleMessageOptions`
- Added `truncateHistoryIfNeeded()` private method for automatic history truncation
- Calls `truncateHistoryIfNeeded()` after every message addition (6 locations)
- Checks `saveInputHistory` before saving input to history
- Ensures history stays within configured limits automatically
- **useSession Hook**
- Added `config?: Config` to `UseSessionDependencies`
- Passes `maxHistoryMessages` and `saveInputHistory` from config to HandleMessage options
- Session configuration now flows from config through to message handling
- **App Component**
- Added `config?: Config` to `AppDependencies`
- Passes config to useSession hook
- Enables configuration-driven session management
### Technical Details
- Total tests: 1590 passed (was 1571, +19 new tests)
- New test file: `session-config.test.ts` with 19 tests
- Default values validation
- `persistIndefinitely` boolean validation
- `maxHistoryMessages` positive integer validation (including edge cases: zero, negative, float rejection)
- `saveInputHistory` boolean validation
- Partial and full config merging tests
- Coverage: 97.62% lines, 91.32% branches, 98.77% functions, 97.62% statements
- 0 ESLint errors, 0 warnings
- Build successful with no TypeScript errors
### Notes
This release completes the second item (0.22.2) of the v0.22.0 Extended Configuration milestone. Remaining items for v0.22.0:
- 0.22.3 - Context Configuration
- 0.22.4 - Autocomplete Configuration
- 0.22.5 - Commands Configuration
---
## [0.22.1] - 2025-12-02 - Display Configuration
### Added

View File

@@ -1648,7 +1648,7 @@ interface DiffViewProps {
## Version 0.22.0 - Extended Configuration ⚙️
**Priority:** MEDIUM
**Status:** In Progress (1/5 complete)
**Status:** Complete (5/5 complete)
### 0.22.1 - Display Configuration ✅
@@ -1670,7 +1670,7 @@ export const DisplayConfigSchema = z.object({
- [x] Configurable stats display
- [x] Unit tests (46 new tests: 20 schema, 24 theme, 2 bell)
### 0.22.2 - Session Configuration
### 0.22.2 - Session Configuration
```typescript
// src/shared/constants/config.ts additions
@@ -1682,12 +1682,12 @@ export const SessionConfigSchema = z.object({
```
**Deliverables:**
- [ ] SessionConfigSchema in config.ts
- [ ] History truncation based on maxHistoryMessages
- [ ] Input history persistence toggle
- [ ] Unit tests
- [x] SessionConfigSchema in config.ts
- [x] History truncation based on maxHistoryMessages
- [x] Input history persistence toggle
- [x] Unit tests (19 new tests)
### 0.22.3 - Context Configuration
### 0.22.3 - Context Configuration
```typescript
// src/shared/constants/config.ts additions
@@ -1700,12 +1700,12 @@ export const ContextConfigSchema = z.object({
```
**Deliverables:**
- [ ] ContextConfigSchema in config.ts
- [ ] ContextManager reads from config
- [ ] Configurable compression threshold
- [ ] Unit tests
- [x] ContextConfigSchema in config.ts
- [x] ContextManager reads from config
- [x] Configurable compression threshold
- [x] Unit tests (40 new tests: 32 schema, 8 ContextManager integration)
### 0.22.4 - Autocomplete Configuration
### 0.22.4 - Autocomplete Configuration
```typescript
// src/shared/constants/config.ts additions
@@ -1717,11 +1717,11 @@ export const AutocompleteConfigSchema = z.object({
```
**Deliverables:**
- [ ] AutocompleteConfigSchema in config.ts
- [ ] useAutocomplete reads from config
- [ ] Unit tests
- [x] AutocompleteConfigSchema in config.ts
- [x] useAutocomplete reads from config
- [x] Unit tests (27 tests)
### 0.22.5 - Commands Configuration
### 0.22.5 - Commands Configuration
```typescript
// src/shared/constants/config.ts additions
@@ -1731,13 +1731,13 @@ export const CommandsConfigSchema = z.object({
```
**Deliverables:**
- [ ] CommandsConfigSchema in config.ts
- [ ] Timeout support for run_command tool
- [ ] Unit tests
- [x] CommandsConfigSchema in config.ts
- [x] Timeout support for run_command tool
- [x] Unit tests (19 schema tests + 3 RunCommandTool integration tests)
**Tests:**
- [ ] Unit tests for all new config schemas
- [ ] Integration tests for config loading
- [x] Unit tests for CommandsConfigSchema (19 tests)
- [x] Integration tests for RunCommandTool with config (3 tests)
---

View File

@@ -79,7 +79,7 @@ export class AuthService {
return {
token,
expiresAt,
userId: user.id
userId: user.id,
}
}
}

View File

@@ -21,7 +21,7 @@ async function main(): Promise<void> {
email: "demo@example.com",
name: "Demo User",
password: "password123",
role: "admin"
role: "admin",
})
logger.info("Demo user created", { userId: user.id })

View File

@@ -25,9 +25,7 @@ export class UserService {
}
// Check if user already exists
const existingUser = Array.from(this.users.values()).find(
(u) => u.email === dto.email
)
const existingUser = Array.from(this.users.values()).find((u) => u.email === dto.email)
if (existingUser) {
throw new Error("User with this email already exists")
@@ -40,7 +38,7 @@ export class UserService {
name: dto.name,
role: dto.role || "user",
createdAt: new Date(),
updatedAt: new Date()
updatedAt: new Date(),
}
this.users.set(user.id, user)
@@ -71,7 +69,7 @@ export class UserService {
...user,
...(dto.name && { name: dto.name }),
...(dto.role && { role: dto.role }),
updatedAt: new Date()
updatedAt: new Date(),
}
this.users.set(id, updated)

View File

@@ -30,7 +30,7 @@ export class Logger {
level,
context: this.context,
message,
...(meta && { meta })
...(meta && { meta }),
}
console.log(JSON.stringify(logEntry))
}

View File

@@ -20,7 +20,7 @@ export function sanitizeInput(input: string): string {
export class ValidationError extends Error {
constructor(
message: string,
public field: string
public field: string,
) {
super(message)
this.name = "ValidationError"

View File

@@ -18,7 +18,7 @@ describe("UserService", () => {
const user = await userService.createUser({
email: "test@example.com",
name: "Test User",
password: "password123"
password: "password123",
})
expect(user).toBeDefined()
@@ -32,8 +32,8 @@ describe("UserService", () => {
userService.createUser({
email: "invalid-email",
name: "Test User",
password: "password123"
})
password: "password123",
}),
).rejects.toThrow(ValidationError)
})
@@ -42,8 +42,8 @@ describe("UserService", () => {
userService.createUser({
email: "test@example.com",
name: "Test User",
password: "weak"
})
password: "weak",
}),
).rejects.toThrow(ValidationError)
})
@@ -51,15 +51,15 @@ describe("UserService", () => {
await userService.createUser({
email: "test@example.com",
name: "Test User",
password: "password123"
password: "password123",
})
await expect(
userService.createUser({
email: "test@example.com",
name: "Another User",
password: "password123"
})
password: "password123",
}),
).rejects.toThrow("already exists")
})
})
@@ -69,7 +69,7 @@ describe("UserService", () => {
const created = await userService.createUser({
email: "test@example.com",
name: "Test User",
password: "password123"
password: "password123",
})
const found = await userService.getUserById(created.id)
@@ -87,11 +87,11 @@ describe("UserService", () => {
const user = await userService.createUser({
email: "test@example.com",
name: "Test User",
password: "password123"
password: "password123",
})
const updated = await userService.updateUser(user.id, {
name: "Updated Name"
name: "Updated Name",
})
expect(updated.name).toBe("Updated Name")
@@ -99,9 +99,9 @@ describe("UserService", () => {
})
it("should throw error for non-existent user", async () => {
await expect(
userService.updateUser("non-existent", { name: "Test" })
).rejects.toThrow("not found")
await expect(userService.updateUser("non-existent", { name: "Test" })).rejects.toThrow(
"not found",
)
})
})
@@ -110,7 +110,7 @@ describe("UserService", () => {
const user = await userService.createUser({
email: "test@example.com",
name: "Test User",
password: "password123"
password: "password123",
})
await userService.deleteUser(user.id)
@@ -125,13 +125,13 @@ describe("UserService", () => {
await userService.createUser({
email: "user1@example.com",
name: "User 1",
password: "password123"
password: "password123",
})
await userService.createUser({
email: "user2@example.com",
name: "User 2",
password: "password123"
password: "password123",
})
const users = await userService.listUsers()

View File

@@ -3,6 +3,6 @@ import { defineConfig } from "vitest/config"
export default defineConfig({
test: {
globals: true,
environment: "node"
}
environment: "node",
},
})

View File

@@ -1,6 +1,6 @@
{
"name": "@samiyev/ipuaro",
"version": "0.22.1",
"version": "0.22.4",
"description": "Local AI agent for codebase operations with infinite context feeling",
"author": "Fozilbek Samiyev <fozilbek.samiyev@gmail.com>",
"license": "MIT",

View File

@@ -2,6 +2,7 @@ import type { ContextState, Session } from "../../domain/entities/Session.js"
import type { ILLMClient } from "../../domain/services/ILLMClient.js"
import { type ChatMessage, createSystemMessage } from "../../domain/value-objects/ChatMessage.js"
import { CONTEXT_COMPRESSION_THRESHOLD, CONTEXT_WINDOW_SIZE } from "../../domain/constants/index.js"
import type { ContextConfig } from "../../shared/constants/config.js"
/**
* File in context with token count.
@@ -39,9 +40,13 @@ export class ContextManager {
private readonly filesInContext = new Map<string, FileContext>()
private currentTokens = 0
private readonly contextWindowSize: number
private readonly compressionThreshold: number
private readonly compressionMethod: "llm-summary" | "truncate"
constructor(contextWindowSize: number = CONTEXT_WINDOW_SIZE) {
constructor(contextWindowSize: number = CONTEXT_WINDOW_SIZE, config?: ContextConfig) {
this.contextWindowSize = contextWindowSize
this.compressionThreshold = config?.autoCompressAt ?? CONTEXT_COMPRESSION_THRESHOLD
this.compressionMethod = config?.compressionMethod ?? "llm-summary"
}
/**
@@ -97,7 +102,7 @@ export class ContextManager {
* Check if compression is needed.
*/
needsCompression(): boolean {
return this.getUsage() > CONTEXT_COMPRESSION_THRESHOLD
return this.getUsage() > this.compressionThreshold
}
/**

View File

@@ -68,6 +68,9 @@ export interface HandleMessageEvents {
export interface HandleMessageOptions {
autoApply?: boolean
maxToolCalls?: number
maxHistoryMessages?: number
saveInputHistory?: boolean
contextConfig?: import("../../shared/constants/config.js").ContextConfig
}
const DEFAULT_MAX_TOOL_CALLS = 20
@@ -96,13 +99,14 @@ export class HandleMessage {
llm: ILLMClient,
tools: IToolRegistry,
projectRoot: string,
contextConfig?: import("../../shared/constants/config.js").ContextConfig,
) {
this.storage = storage
this.sessionStorage = sessionStorage
this.llm = llm
this.tools = tools
this.projectRoot = projectRoot
this.contextManager = new ContextManager(llm.getContextWindowSize())
this.contextManager = new ContextManager(llm.getContextWindowSize(), contextConfig)
this.executeTool = new ExecuteTool(storage, sessionStorage, tools, projectRoot)
}
@@ -135,6 +139,15 @@ export class HandleMessage {
this.llm.abort()
}
/**
* Truncate session history if maxHistoryMessages is set.
*/
private truncateHistoryIfNeeded(session: Session): void {
if (this.options.maxHistoryMessages !== undefined) {
session.truncateHistory(this.options.maxHistoryMessages)
}
}
/**
* Execute the message handling flow.
*/
@@ -145,7 +158,12 @@ export class HandleMessage {
if (message.trim()) {
const userMessage = createUserMessage(message)
session.addMessage(userMessage)
session.addInputToHistory(message)
this.truncateHistoryIfNeeded(session)
if (this.options.saveInputHistory !== false) {
session.addInputToHistory(message)
}
this.emitMessage(userMessage)
}
@@ -183,6 +201,7 @@ export class HandleMessage {
toolCalls: 0,
})
session.addMessage(assistantMessage)
this.truncateHistoryIfNeeded(session)
this.emitMessage(assistantMessage)
this.contextManager.addTokens(response.tokens)
this.contextManager.updateSession(session)
@@ -197,6 +216,7 @@ export class HandleMessage {
toolCalls: parsed.toolCalls.length,
})
session.addMessage(assistantMessage)
this.truncateHistoryIfNeeded(session)
this.emitMessage(assistantMessage)
toolCallCount += parsed.toolCalls.length
@@ -204,6 +224,7 @@ export class HandleMessage {
const errorMsg = `Maximum tool calls (${String(maxToolCalls)}) exceeded`
const errorMessage = createSystemMessage(errorMsg)
session.addMessage(errorMessage)
this.truncateHistoryIfNeeded(session)
this.emitMessage(errorMessage)
this.emitStatus("ready")
return
@@ -227,6 +248,7 @@ export class HandleMessage {
const toolMessage = createToolMessage(results)
session.addMessage(toolMessage)
this.truncateHistoryIfNeeded(session)
this.contextManager.addTokens(response.tokens)
@@ -306,6 +328,7 @@ export class HandleMessage {
const errorMessage = createSystemMessage(`Error: ${ipuaroError.message}`)
session.addMessage(errorMessage)
this.truncateHistoryIfNeeded(session)
this.emitMessage(errorMessage)
this.emitStatus("ready")

View File

@@ -94,6 +94,12 @@ export class Session {
}
}
truncateHistory(maxMessages: number): void {
if (this.history.length > maxMessages) {
this.history = this.history.slice(-maxMessages)
}
}
clearHistory(): void {
this.history = []
this.context = {

View File

@@ -6,6 +6,7 @@ import {
createSuccessResult,
type ToolResult,
} from "../../../domain/value-objects/ToolResult.js"
import type { CommandsConfig } from "../../../shared/constants/config.js"
import { CommandSecurity } from "./CommandSecurity.js"
const execAsync = promisify(exec)
@@ -60,7 +61,7 @@ export class RunCommandTool implements ITool {
{
name: "timeout",
type: "number",
description: "Timeout in milliseconds (default: 30000)",
description: "Timeout in milliseconds (default: from config or 30000, max: 600000)",
required: false,
},
]
@@ -69,10 +70,12 @@ export class RunCommandTool implements ITool {
private readonly security: CommandSecurity
private readonly execFn: typeof execAsync
private readonly configTimeout: number | null
constructor(security?: CommandSecurity, execFn?: typeof execAsync) {
constructor(security?: CommandSecurity, execFn?: typeof execAsync, config?: CommandsConfig) {
this.security = security ?? new CommandSecurity()
this.execFn = execFn ?? execAsync
this.configTimeout = config?.timeout ?? null
}
validateParams(params: Record<string, unknown>): string | null {
@@ -104,7 +107,7 @@ export class RunCommandTool implements ITool {
const callId = `${this.name}-${String(startTime)}`
const command = params.command as string
const timeout = (params.timeout as number) ?? DEFAULT_TIMEOUT
const timeout = (params.timeout as number) ?? this.configTimeout ?? DEFAULT_TIMEOUT
const securityCheck = this.security.check(command)

View File

@@ -97,6 +97,41 @@ export const DisplayConfigSchema = z.object({
progressBar: z.boolean().default(true),
})
/**
* Session configuration schema.
*/
export const SessionConfigSchema = z.object({
persistIndefinitely: z.boolean().default(true),
maxHistoryMessages: z.number().int().positive().default(100),
saveInputHistory: z.boolean().default(true),
})
/**
* Context configuration schema.
*/
export const ContextConfigSchema = z.object({
systemPromptTokens: z.number().int().positive().default(2000),
maxContextUsage: z.number().min(0).max(1).default(0.8),
autoCompressAt: z.number().min(0).max(1).default(0.8),
compressionMethod: z.enum(["llm-summary", "truncate"]).default("llm-summary"),
})
/**
* Autocomplete configuration schema.
*/
export const AutocompleteConfigSchema = z.object({
enabled: z.boolean().default(true),
source: z.enum(["redis-index", "filesystem", "both"]).default("redis-index"),
maxSuggestions: z.number().int().positive().default(10),
})
/**
* Commands configuration schema.
*/
export const CommandsConfigSchema = z.object({
timeout: z.number().int().positive().nullable().default(null),
})
/**
* Full configuration schema.
*/
@@ -109,6 +144,10 @@ export const ConfigSchema = z.object({
edit: EditConfigSchema.default({}),
input: InputConfigSchema.default({}),
display: DisplayConfigSchema.default({}),
session: SessionConfigSchema.default({}),
context: ContextConfigSchema.default({}),
autocomplete: AutocompleteConfigSchema.default({}),
commands: CommandsConfigSchema.default({}),
})
/**
@@ -123,6 +162,10 @@ export type UndoConfig = z.infer<typeof UndoConfigSchema>
export type EditConfig = z.infer<typeof EditConfigSchema>
export type InputConfig = z.infer<typeof InputConfigSchema>
export type DisplayConfig = z.infer<typeof DisplayConfigSchema>
export type SessionConfig = z.infer<typeof SessionConfigSchema>
export type ContextConfig = z.infer<typeof ContextConfigSchema>
export type AutocompleteConfig = z.infer<typeof AutocompleteConfigSchema>
export type CommandsConfig = z.infer<typeof CommandsConfigSchema>
/**
* Default configuration.

View File

@@ -10,6 +10,7 @@ import type { ISessionStorage } from "../domain/services/ISessionStorage.js"
import type { IStorage } from "../domain/services/IStorage.js"
import type { DiffInfo } from "../domain/services/ITool.js"
import type { ErrorOption } from "../shared/errors/IpuaroError.js"
import type { Config } from "../shared/constants/config.js"
import type { IToolRegistry } from "../application/interfaces/IToolRegistry.js"
import type { ConfirmationResult } from "../application/use-cases/ExecuteTool.js"
import type { ProjectStructure } from "../infrastructure/llm/prompts.js"
@@ -25,6 +26,7 @@ export interface AppDependencies {
llm: ILLMClient
tools: IToolRegistry
projectStructure?: ProjectStructure
config?: Config
}
export interface ExtendedAppProps extends AppProps {
@@ -129,6 +131,7 @@ export function App({
projectRoot: projectPath,
projectName,
projectStructure: deps.projectStructure,
config: deps.config,
},
{
autoApply,

View File

@@ -120,9 +120,7 @@ function AssistantMessage({
)
}
function ToolMessage({ message, theme }: MessageComponentProps): React.JSX.Element {
const roleColor = getRoleColor("tool", theme)
function ToolMessage({ message }: MessageComponentProps): React.JSX.Element {
return (
<Box flexDirection="column" marginBottom={1} marginLeft={2}>
{message.toolResults?.map((result) => (

View File

@@ -5,6 +5,7 @@
import { useCallback, useEffect, useState } from "react"
import type { IStorage } from "../../domain/services/IStorage.js"
import type { AutocompleteConfig } from "../../shared/constants/config.js"
import path from "node:path"
export interface UseAutocompleteOptions {
@@ -12,6 +13,7 @@ export interface UseAutocompleteOptions {
projectRoot: string
enabled?: boolean
maxSuggestions?: number
config?: AutocompleteConfig
}
export interface UseAutocompleteReturn {
@@ -107,13 +109,18 @@ function getCommonPrefix(suggestions: string[]): string {
}
export function useAutocomplete(options: UseAutocompleteOptions): UseAutocompleteReturn {
const { storage, projectRoot, enabled = true, maxSuggestions = 10 } = options
const { storage, projectRoot, enabled, maxSuggestions, config } = options
// Read from config if provided, otherwise use options, otherwise use defaults
const isEnabled = config?.enabled ?? enabled ?? true
const maxSuggestionsCount = config?.maxSuggestions ?? maxSuggestions ?? 10
const [filePaths, setFilePaths] = useState<string[]>([])
const [suggestions, setSuggestions] = useState<string[]>([])
// Load file paths from storage
useEffect(() => {
if (!enabled) {
if (!isEnabled) {
return
}
@@ -135,11 +142,11 @@ export function useAutocomplete(options: UseAutocompleteOptions): UseAutocomplet
loadPaths().catch(() => {
// Ignore errors
})
}, [storage, projectRoot, enabled])
}, [storage, projectRoot, isEnabled])
const complete = useCallback(
(partial: string): string[] => {
if (!enabled || !partial.trim()) {
if (!isEnabled || !partial.trim()) {
setSuggestions([])
return []
}
@@ -154,13 +161,13 @@ export function useAutocomplete(options: UseAutocompleteOptions): UseAutocomplet
}))
.filter((item) => item.score > 0)
.sort((a, b) => b.score - a.score)
.slice(0, maxSuggestions)
.slice(0, maxSuggestionsCount)
.map((item) => item.path)
setSuggestions(scored)
return scored
},
[enabled, filePaths, maxSuggestions],
[isEnabled, filePaths, maxSuggestionsCount],
)
const accept = useCallback(

View File

@@ -11,6 +11,7 @@ import type { IStorage } from "../../domain/services/IStorage.js"
import type { DiffInfo } from "../../domain/services/ITool.js"
import type { ChatMessage } from "../../domain/value-objects/ChatMessage.js"
import type { ErrorOption } from "../../shared/errors/IpuaroError.js"
import type { Config } from "../../shared/constants/config.js"
import type { IToolRegistry } from "../../application/interfaces/IToolRegistry.js"
import {
HandleMessage,
@@ -30,6 +31,7 @@ export interface UseSessionDependencies {
projectRoot: string
projectName: string
projectStructure?: ProjectStructure
config?: Config
}
export interface UseSessionOptions {
@@ -107,11 +109,17 @@ async function initializeSession(
deps.llm,
deps.tools,
deps.projectRoot,
deps.config?.context,
)
if (deps.projectStructure) {
handleMessage.setProjectStructure(deps.projectStructure)
}
handleMessage.setOptions({ autoApply: options.autoApply })
handleMessage.setOptions({
autoApply: options.autoApply,
maxHistoryMessages: deps.config?.session.maxHistoryMessages,
saveInputHistory: deps.config?.session.saveInputHistory,
contextConfig: deps.config?.context,
})
handleMessage.setEvents(createEventHandlers(setters, options))
refs.current.handleMessage = handleMessage
refs.current.undoChange = new UndoChange(deps.sessionStorage, deps.storage)

View File

@@ -245,4 +245,65 @@ describe("ContextManager", () => {
expect(state.needsCompression).toBe(false)
})
})
describe("configuration", () => {
it("should use default compression threshold when no config provided", () => {
const manager = new ContextManager(CONTEXT_SIZE)
manager.addToContext("test.ts", CONTEXT_SIZE * 0.85)
expect(manager.needsCompression()).toBe(true)
})
it("should use custom compression threshold from config", () => {
const manager = new ContextManager(CONTEXT_SIZE, { autoCompressAt: 0.9 })
manager.addToContext("test.ts", CONTEXT_SIZE * 0.85)
expect(manager.needsCompression()).toBe(false)
})
it("should trigger compression at custom threshold", () => {
const manager = new ContextManager(CONTEXT_SIZE, { autoCompressAt: 0.9 })
manager.addToContext("test.ts", CONTEXT_SIZE * 0.95)
expect(manager.needsCompression()).toBe(true)
})
it("should accept compression method in config", () => {
const manager = new ContextManager(CONTEXT_SIZE, { compressionMethod: "truncate" })
expect(manager).toBeDefined()
})
it("should use default compression method when not specified", () => {
const manager = new ContextManager(CONTEXT_SIZE, {})
expect(manager).toBeDefined()
})
it("should accept full context config", () => {
const manager = new ContextManager(CONTEXT_SIZE, {
systemPromptTokens: 3000,
maxContextUsage: 0.9,
autoCompressAt: 0.85,
compressionMethod: "llm-summary",
})
manager.addToContext("test.ts", CONTEXT_SIZE * 0.87)
expect(manager.needsCompression()).toBe(true)
})
it("should handle edge case: autoCompressAt = 0", () => {
const manager = new ContextManager(CONTEXT_SIZE, { autoCompressAt: 0 })
manager.addToContext("test.ts", 1)
expect(manager.needsCompression()).toBe(true)
})
it("should handle edge case: autoCompressAt = 1", () => {
const manager = new ContextManager(CONTEXT_SIZE, { autoCompressAt: 1 })
manager.addToContext("test.ts", CONTEXT_SIZE * 0.99)
expect(manager.needsCompression()).toBe(false)
})
})
})

View File

@@ -19,7 +19,7 @@ vi.mock("../../../../src/infrastructure/indexer/FileScanner.js", () => ({
return 'export function main() { return "hello" }'
}
if (path.includes("utils.ts")) {
return 'export const add = (a: number, b: number) => a + b'
return "export const add = (a: number, b: number) => a + b"
}
return null
}
@@ -31,7 +31,16 @@ vi.mock("../../../../src/infrastructure/indexer/ASTParser.js", () => ({
parse() {
return {
...createEmptyFileAST(),
functions: [{ name: "test", lineStart: 1, lineEnd: 5, params: [], isAsync: false, isExported: true }],
functions: [
{
name: "test",
lineStart: 1,
lineEnd: 5,
params: [],
isAsync: false,
isExported: true,
},
],
}
}
},
@@ -116,7 +125,7 @@ describe("IndexProject", () => {
expect.objectContaining({
hash: expect.any(String),
lines: expect.any(Array),
})
}),
)
})
@@ -128,7 +137,7 @@ describe("IndexProject", () => {
"src/index.ts",
expect.objectContaining({
functions: expect.any(Array),
})
}),
)
})
@@ -136,19 +145,14 @@ describe("IndexProject", () => {
await useCase.execute("/test/project")
expect(mockStorage.setMeta).toHaveBeenCalledTimes(2)
expect(mockStorage.setMeta).toHaveBeenCalledWith(
"src/index.ts",
expect.any(Object)
)
expect(mockStorage.setMeta).toHaveBeenCalledWith("src/index.ts", expect.any(Object))
})
it("should build and store symbol index", async () => {
await useCase.execute("/test/project")
expect(mockStorage.setSymbolIndex).toHaveBeenCalledTimes(1)
expect(mockStorage.setSymbolIndex).toHaveBeenCalledWith(
expect.any(Map)
)
expect(mockStorage.setSymbolIndex).toHaveBeenCalledWith(expect.any(Map))
})
it("should build and store dependency graph", async () => {
@@ -159,7 +163,7 @@ describe("IndexProject", () => {
expect.objectContaining({
imports: expect.any(Map),
importedBy: expect.any(Map),
})
}),
)
})
@@ -168,7 +172,7 @@ describe("IndexProject", () => {
expect(mockStorage.setProjectConfig).toHaveBeenCalledWith(
"last_indexed",
expect.any(Number)
expect.any(Number),
)
})
@@ -186,7 +190,7 @@ describe("IndexProject", () => {
total: expect.any(Number),
currentFile: expect.any(String),
phase: expect.stringMatching(/scanning|parsing|analyzing|indexing/),
})
}),
)
})
@@ -198,7 +202,7 @@ describe("IndexProject", () => {
})
const scanningCalls = progressCallback.mock.calls.filter(
(call) => call[0].phase === "scanning"
(call) => call[0].phase === "scanning",
)
expect(scanningCalls.length).toBeGreaterThan(0)
})
@@ -211,7 +215,7 @@ describe("IndexProject", () => {
})
const parsingCalls = progressCallback.mock.calls.filter(
(call) => call[0].phase === "parsing"
(call) => call[0].phase === "parsing",
)
expect(parsingCalls.length).toBeGreaterThan(0)
})
@@ -224,7 +228,7 @@ describe("IndexProject", () => {
})
const analyzingCalls = progressCallback.mock.calls.filter(
(call) => call[0].phase === "analyzing"
(call) => call[0].phase === "analyzing",
)
expect(analyzingCalls.length).toBeGreaterThan(0)
})
@@ -237,7 +241,7 @@ describe("IndexProject", () => {
})
const indexingCalls = progressCallback.mock.calls.filter(
(call) => call[0].phase === "indexing"
(call) => call[0].phase === "indexing",
)
expect(indexingCalls.length).toBeGreaterThan(0)
})
@@ -245,10 +249,7 @@ describe("IndexProject", () => {
it("should detect TypeScript files", async () => {
await useCase.execute("/test/project")
expect(mockStorage.setAST).toHaveBeenCalledWith(
"src/index.ts",
expect.any(Object)
)
expect(mockStorage.setAST).toHaveBeenCalledWith("src/index.ts", expect.any(Object))
})
it("should handle files without parseable language", async () => {
@@ -276,7 +277,7 @@ describe("IndexProject", () => {
expect(mockStorage.setAST).toHaveBeenCalledWith(
expect.stringContaining(".ts"),
expect.any(Object)
expect.any(Object),
)
})
})
@@ -294,7 +295,7 @@ describe("IndexProject", () => {
})
const callsWithFiles = progressCallback.mock.calls.filter(
(call) => call[0].currentFile && call[0].currentFile.length > 0
(call) => call[0].currentFile && call[0].currentFile.length > 0,
)
expect(callsWithFiles.length).toBeGreaterThan(0)
})
@@ -307,7 +308,7 @@ describe("IndexProject", () => {
})
const parsingCalls = progressCallback.mock.calls.filter(
(call) => call[0].phase === "parsing"
(call) => call[0].phase === "parsing",
)
if (parsingCalls.length > 0) {
expect(parsingCalls[0][0].total).toBe(2)

View File

@@ -123,8 +123,7 @@ describe("OllamaClient", () => {
mockOllamaInstance.chat.mockResolvedValue({
message: {
role: "assistant",
content:
'<tool_call name="get_lines"><path>src/index.ts</path></tool_call>',
content: '<tool_call name="get_lines"><path>src/index.ts</path></tool_call>',
tool_calls: undefined,
},
eval_count: 30,
@@ -408,7 +407,6 @@ describe("OllamaClient", () => {
})
})
describe("error handling", () => {
it("should handle ECONNREFUSED errors", async () => {
mockOllamaInstance.chat.mockRejectedValue(new Error("ECONNREFUSED"))
@@ -435,7 +433,9 @@ describe("OllamaClient", () => {
const client = new OllamaClient(defaultConfig)
await expect(client.chat([createUserMessage("Hello")])).rejects.toThrow(/Request was aborted/)
await expect(client.chat([createUserMessage("Hello")])).rejects.toThrow(
/Request was aborted/,
)
})
it("should handle model not found errors", async () => {
@@ -443,7 +443,9 @@ describe("OllamaClient", () => {
const client = new OllamaClient(defaultConfig)
await expect(client.chat([createUserMessage("Hello")])).rejects.toThrow(/Model.*not found/)
await expect(client.chat([createUserMessage("Hello")])).rejects.toThrow(
/Model.*not found/,
)
})
})
})

View File

@@ -303,7 +303,9 @@ describe("GetFunctionTool", () => {
})
it("should handle error when reading lines fails", async () => {
const ast = createMockAST([createMockFunction({ name: "test", lineStart: 1, lineEnd: 1 })])
const ast = createMockAST([
createMockFunction({ name: "test", lineStart: 1, lineEnd: 1 }),
])
const storage: IStorage = {
getFile: vi.fn().mockResolvedValue(null),
getAST: vi.fn().mockResolvedValue(ast),

View File

@@ -354,6 +354,36 @@ describe("RunCommandTool", () => {
expect(execFn).toHaveBeenCalledWith("ls", expect.objectContaining({ timeout: 5000 }))
})
it("should use config timeout", async () => {
const execFn = createMockExec({})
const toolWithMock = new RunCommandTool(undefined, execFn, { timeout: 45000 })
const ctx = createMockContext()
await toolWithMock.execute({ command: "ls" }, ctx)
expect(execFn).toHaveBeenCalledWith("ls", expect.objectContaining({ timeout: 45000 }))
})
it("should use null config timeout as default", async () => {
const execFn = createMockExec({})
const toolWithMock = new RunCommandTool(undefined, execFn, { timeout: null })
const ctx = createMockContext()
await toolWithMock.execute({ command: "ls" }, ctx)
expect(execFn).toHaveBeenCalledWith("ls", expect.objectContaining({ timeout: 30000 }))
})
it("should prefer param timeout over config timeout", async () => {
const execFn = createMockExec({})
const toolWithMock = new RunCommandTool(undefined, execFn, { timeout: 45000 })
const ctx = createMockContext()
await toolWithMock.execute({ command: "ls", timeout: 5000 }, ctx)
expect(execFn).toHaveBeenCalledWith("ls", expect.objectContaining({ timeout: 5000 }))
})
it("should execute in project root", async () => {
const execFn = createMockExec({})
const toolWithMock = new RunCommandTool(undefined, execFn)

View File

@@ -0,0 +1,204 @@
/**
* Tests for AutocompleteConfigSchema.
*/
import { describe, expect, it } from "vitest"
import { AutocompleteConfigSchema } from "../../../src/shared/constants/config.js"
describe("AutocompleteConfigSchema", () => {
describe("default values", () => {
it("should use defaults when empty object provided", () => {
const result = AutocompleteConfigSchema.parse({})
expect(result).toEqual({
enabled: true,
source: "redis-index",
maxSuggestions: 10,
})
})
it("should use defaults via .default({})", () => {
const result = AutocompleteConfigSchema.default({}).parse({})
expect(result).toEqual({
enabled: true,
source: "redis-index",
maxSuggestions: 10,
})
})
})
describe("enabled", () => {
it("should accept true", () => {
const result = AutocompleteConfigSchema.parse({ enabled: true })
expect(result.enabled).toBe(true)
})
it("should accept false", () => {
const result = AutocompleteConfigSchema.parse({ enabled: false })
expect(result.enabled).toBe(false)
})
it("should reject non-boolean", () => {
expect(() => AutocompleteConfigSchema.parse({ enabled: "true" })).toThrow()
})
it("should reject number", () => {
expect(() => AutocompleteConfigSchema.parse({ enabled: 1 })).toThrow()
})
})
describe("source", () => {
it("should accept redis-index", () => {
const result = AutocompleteConfigSchema.parse({ source: "redis-index" })
expect(result.source).toBe("redis-index")
})
it("should accept filesystem", () => {
const result = AutocompleteConfigSchema.parse({ source: "filesystem" })
expect(result.source).toBe("filesystem")
})
it("should accept both", () => {
const result = AutocompleteConfigSchema.parse({ source: "both" })
expect(result.source).toBe("both")
})
it("should use default redis-index", () => {
const result = AutocompleteConfigSchema.parse({})
expect(result.source).toBe("redis-index")
})
it("should reject invalid source", () => {
expect(() => AutocompleteConfigSchema.parse({ source: "invalid" })).toThrow()
})
it("should reject non-string", () => {
expect(() => AutocompleteConfigSchema.parse({ source: 123 })).toThrow()
})
})
describe("maxSuggestions", () => {
it("should accept valid positive integer", () => {
const result = AutocompleteConfigSchema.parse({ maxSuggestions: 5 })
expect(result.maxSuggestions).toBe(5)
})
it("should accept default value", () => {
const result = AutocompleteConfigSchema.parse({ maxSuggestions: 10 })
expect(result.maxSuggestions).toBe(10)
})
it("should accept large value", () => {
const result = AutocompleteConfigSchema.parse({ maxSuggestions: 100 })
expect(result.maxSuggestions).toBe(100)
})
it("should accept 1", () => {
const result = AutocompleteConfigSchema.parse({ maxSuggestions: 1 })
expect(result.maxSuggestions).toBe(1)
})
it("should reject zero", () => {
expect(() => AutocompleteConfigSchema.parse({ maxSuggestions: 0 })).toThrow()
})
it("should reject negative number", () => {
expect(() => AutocompleteConfigSchema.parse({ maxSuggestions: -5 })).toThrow()
})
it("should reject float", () => {
expect(() => AutocompleteConfigSchema.parse({ maxSuggestions: 10.5 })).toThrow()
})
it("should reject non-number", () => {
expect(() => AutocompleteConfigSchema.parse({ maxSuggestions: "10" })).toThrow()
})
})
describe("partial config", () => {
it("should merge partial config with defaults (enabled only)", () => {
const result = AutocompleteConfigSchema.parse({
enabled: false,
})
expect(result).toEqual({
enabled: false,
source: "redis-index",
maxSuggestions: 10,
})
})
it("should merge partial config with defaults (source only)", () => {
const result = AutocompleteConfigSchema.parse({
source: "filesystem",
})
expect(result).toEqual({
enabled: true,
source: "filesystem",
maxSuggestions: 10,
})
})
it("should merge partial config with defaults (maxSuggestions only)", () => {
const result = AutocompleteConfigSchema.parse({
maxSuggestions: 20,
})
expect(result).toEqual({
enabled: true,
source: "redis-index",
maxSuggestions: 20,
})
})
it("should merge multiple partial fields", () => {
const result = AutocompleteConfigSchema.parse({
enabled: false,
maxSuggestions: 5,
})
expect(result).toEqual({
enabled: false,
source: "redis-index",
maxSuggestions: 5,
})
})
})
describe("full config", () => {
it("should accept valid full config", () => {
const config = {
enabled: false,
source: "both" as const,
maxSuggestions: 15,
}
const result = AutocompleteConfigSchema.parse(config)
expect(result).toEqual(config)
})
it("should accept all defaults explicitly", () => {
const config = {
enabled: true,
source: "redis-index" as const,
maxSuggestions: 10,
}
const result = AutocompleteConfigSchema.parse(config)
expect(result).toEqual(config)
})
it("should accept filesystem as source", () => {
const config = {
enabled: true,
source: "filesystem" as const,
maxSuggestions: 20,
}
const result = AutocompleteConfigSchema.parse(config)
expect(result).toEqual(config)
})
})
})

View File

@@ -0,0 +1,137 @@
/**
* Tests for CommandsConfigSchema.
*/
import { describe, expect, it } from "vitest"
import { CommandsConfigSchema } from "../../../src/shared/constants/config.js"
describe("CommandsConfigSchema", () => {
describe("default values", () => {
it("should use defaults when empty object provided", () => {
const result = CommandsConfigSchema.parse({})
expect(result).toEqual({
timeout: null,
})
})
it("should use defaults via .default({})", () => {
const result = CommandsConfigSchema.default({}).parse({})
expect(result).toEqual({
timeout: null,
})
})
})
describe("timeout", () => {
it("should accept null (default)", () => {
const result = CommandsConfigSchema.parse({ timeout: null })
expect(result.timeout).toBe(null)
})
it("should accept positive integer", () => {
const result = CommandsConfigSchema.parse({ timeout: 5000 })
expect(result.timeout).toBe(5000)
})
it("should accept large timeout", () => {
const result = CommandsConfigSchema.parse({ timeout: 600000 })
expect(result.timeout).toBe(600000)
})
it("should accept 1", () => {
const result = CommandsConfigSchema.parse({ timeout: 1 })
expect(result.timeout).toBe(1)
})
it("should accept small timeout", () => {
const result = CommandsConfigSchema.parse({ timeout: 100 })
expect(result.timeout).toBe(100)
})
it("should reject zero", () => {
expect(() => CommandsConfigSchema.parse({ timeout: 0 })).toThrow()
})
it("should reject negative number", () => {
expect(() => CommandsConfigSchema.parse({ timeout: -5000 })).toThrow()
})
it("should reject float", () => {
expect(() => CommandsConfigSchema.parse({ timeout: 5000.5 })).toThrow()
})
it("should reject string", () => {
expect(() => CommandsConfigSchema.parse({ timeout: "5000" })).toThrow()
})
it("should reject boolean", () => {
expect(() => CommandsConfigSchema.parse({ timeout: true })).toThrow()
})
it("should reject undefined (use null instead)", () => {
const result = CommandsConfigSchema.parse({ timeout: undefined })
expect(result.timeout).toBe(null)
})
})
describe("partial config", () => {
it("should use default null when timeout not provided", () => {
const result = CommandsConfigSchema.parse({})
expect(result).toEqual({
timeout: null,
})
})
it("should accept explicit null", () => {
const result = CommandsConfigSchema.parse({
timeout: null,
})
expect(result).toEqual({
timeout: null,
})
})
it("should accept explicit timeout value", () => {
const result = CommandsConfigSchema.parse({
timeout: 10000,
})
expect(result).toEqual({
timeout: 10000,
})
})
})
describe("full config", () => {
it("should accept valid config with null", () => {
const config = {
timeout: null,
}
const result = CommandsConfigSchema.parse(config)
expect(result).toEqual(config)
})
it("should accept valid config with timeout", () => {
const config = {
timeout: 30000,
}
const result = CommandsConfigSchema.parse(config)
expect(result).toEqual(config)
})
it("should accept default explicitly", () => {
const config = {
timeout: null,
}
const result = CommandsConfigSchema.parse(config)
expect(result).toEqual(config)
})
})
})

View File

@@ -0,0 +1,221 @@
/**
* Tests for ContextConfigSchema.
*/
import { describe, expect, it } from "vitest"
import { ContextConfigSchema } from "../../../src/shared/constants/config.js"
describe("ContextConfigSchema", () => {
describe("default values", () => {
it("should use defaults when empty object provided", () => {
const result = ContextConfigSchema.parse({})
expect(result).toEqual({
systemPromptTokens: 2000,
maxContextUsage: 0.8,
autoCompressAt: 0.8,
compressionMethod: "llm-summary",
})
})
it("should use defaults via .default({})", () => {
const result = ContextConfigSchema.default({}).parse({})
expect(result).toEqual({
systemPromptTokens: 2000,
maxContextUsage: 0.8,
autoCompressAt: 0.8,
compressionMethod: "llm-summary",
})
})
})
describe("systemPromptTokens", () => {
it("should accept valid positive integer", () => {
const result = ContextConfigSchema.parse({ systemPromptTokens: 1500 })
expect(result.systemPromptTokens).toBe(1500)
})
it("should accept default value", () => {
const result = ContextConfigSchema.parse({ systemPromptTokens: 2000 })
expect(result.systemPromptTokens).toBe(2000)
})
it("should accept large value", () => {
const result = ContextConfigSchema.parse({ systemPromptTokens: 5000 })
expect(result.systemPromptTokens).toBe(5000)
})
it("should reject zero", () => {
expect(() => ContextConfigSchema.parse({ systemPromptTokens: 0 })).toThrow()
})
it("should reject negative number", () => {
expect(() => ContextConfigSchema.parse({ systemPromptTokens: -100 })).toThrow()
})
it("should reject float", () => {
expect(() => ContextConfigSchema.parse({ systemPromptTokens: 1500.5 })).toThrow()
})
it("should reject non-number", () => {
expect(() => ContextConfigSchema.parse({ systemPromptTokens: "2000" })).toThrow()
})
})
describe("maxContextUsage", () => {
it("should accept valid ratio", () => {
const result = ContextConfigSchema.parse({ maxContextUsage: 0.7 })
expect(result.maxContextUsage).toBe(0.7)
})
it("should accept default value", () => {
const result = ContextConfigSchema.parse({ maxContextUsage: 0.8 })
expect(result.maxContextUsage).toBe(0.8)
})
it("should accept minimum value (0)", () => {
const result = ContextConfigSchema.parse({ maxContextUsage: 0 })
expect(result.maxContextUsage).toBe(0)
})
it("should accept maximum value (1)", () => {
const result = ContextConfigSchema.parse({ maxContextUsage: 1 })
expect(result.maxContextUsage).toBe(1)
})
it("should reject value above 1", () => {
expect(() => ContextConfigSchema.parse({ maxContextUsage: 1.1 })).toThrow()
})
it("should reject negative value", () => {
expect(() => ContextConfigSchema.parse({ maxContextUsage: -0.1 })).toThrow()
})
it("should reject non-number", () => {
expect(() => ContextConfigSchema.parse({ maxContextUsage: "0.8" })).toThrow()
})
})
describe("autoCompressAt", () => {
it("should accept valid ratio", () => {
const result = ContextConfigSchema.parse({ autoCompressAt: 0.75 })
expect(result.autoCompressAt).toBe(0.75)
})
it("should accept default value", () => {
const result = ContextConfigSchema.parse({ autoCompressAt: 0.8 })
expect(result.autoCompressAt).toBe(0.8)
})
it("should accept minimum value (0)", () => {
const result = ContextConfigSchema.parse({ autoCompressAt: 0 })
expect(result.autoCompressAt).toBe(0)
})
it("should accept maximum value (1)", () => {
const result = ContextConfigSchema.parse({ autoCompressAt: 1 })
expect(result.autoCompressAt).toBe(1)
})
it("should reject value above 1", () => {
expect(() => ContextConfigSchema.parse({ autoCompressAt: 1.5 })).toThrow()
})
it("should reject negative value", () => {
expect(() => ContextConfigSchema.parse({ autoCompressAt: -0.5 })).toThrow()
})
it("should reject non-number", () => {
expect(() => ContextConfigSchema.parse({ autoCompressAt: "0.8" })).toThrow()
})
})
describe("compressionMethod", () => {
it("should accept llm-summary", () => {
const result = ContextConfigSchema.parse({ compressionMethod: "llm-summary" })
expect(result.compressionMethod).toBe("llm-summary")
})
it("should accept truncate", () => {
const result = ContextConfigSchema.parse({ compressionMethod: "truncate" })
expect(result.compressionMethod).toBe("truncate")
})
it("should reject invalid method", () => {
expect(() => ContextConfigSchema.parse({ compressionMethod: "invalid" })).toThrow()
})
it("should reject non-string", () => {
expect(() => ContextConfigSchema.parse({ compressionMethod: 123 })).toThrow()
})
})
describe("partial config", () => {
it("should merge partial config with defaults (systemPromptTokens)", () => {
const result = ContextConfigSchema.parse({
systemPromptTokens: 3000,
})
expect(result).toEqual({
systemPromptTokens: 3000,
maxContextUsage: 0.8,
autoCompressAt: 0.8,
compressionMethod: "llm-summary",
})
})
it("should merge partial config with defaults (autoCompressAt)", () => {
const result = ContextConfigSchema.parse({
autoCompressAt: 0.9,
})
expect(result).toEqual({
systemPromptTokens: 2000,
maxContextUsage: 0.8,
autoCompressAt: 0.9,
compressionMethod: "llm-summary",
})
})
it("should merge multiple partial fields", () => {
const result = ContextConfigSchema.parse({
maxContextUsage: 0.7,
compressionMethod: "truncate",
})
expect(result).toEqual({
systemPromptTokens: 2000,
maxContextUsage: 0.7,
autoCompressAt: 0.8,
compressionMethod: "truncate",
})
})
})
describe("full config", () => {
it("should accept valid full config", () => {
const config = {
systemPromptTokens: 3000,
maxContextUsage: 0.9,
autoCompressAt: 0.85,
compressionMethod: "truncate" as const,
}
const result = ContextConfigSchema.parse(config)
expect(result).toEqual(config)
})
it("should accept all defaults explicitly", () => {
const config = {
systemPromptTokens: 2000,
maxContextUsage: 0.8,
autoCompressAt: 0.8,
compressionMethod: "llm-summary" as const,
}
const result = ContextConfigSchema.parse(config)
expect(result).toEqual(config)
})
})
})

View File

@@ -0,0 +1,146 @@
/**
* Tests for SessionConfigSchema.
*/
import { describe, expect, it } from "vitest"
import { SessionConfigSchema } from "../../../src/shared/constants/config.js"
describe("SessionConfigSchema", () => {
describe("default values", () => {
it("should use defaults when empty object provided", () => {
const result = SessionConfigSchema.parse({})
expect(result).toEqual({
persistIndefinitely: true,
maxHistoryMessages: 100,
saveInputHistory: true,
})
})
it("should use defaults via .default({})", () => {
const result = SessionConfigSchema.default({}).parse({})
expect(result).toEqual({
persistIndefinitely: true,
maxHistoryMessages: 100,
saveInputHistory: true,
})
})
})
describe("persistIndefinitely", () => {
it("should accept true", () => {
const result = SessionConfigSchema.parse({ persistIndefinitely: true })
expect(result.persistIndefinitely).toBe(true)
})
it("should accept false", () => {
const result = SessionConfigSchema.parse({ persistIndefinitely: false })
expect(result.persistIndefinitely).toBe(false)
})
it("should reject non-boolean", () => {
expect(() => SessionConfigSchema.parse({ persistIndefinitely: "yes" })).toThrow()
})
})
describe("maxHistoryMessages", () => {
it("should accept valid positive integer", () => {
const result = SessionConfigSchema.parse({ maxHistoryMessages: 50 })
expect(result.maxHistoryMessages).toBe(50)
})
it("should accept default value", () => {
const result = SessionConfigSchema.parse({ maxHistoryMessages: 100 })
expect(result.maxHistoryMessages).toBe(100)
})
it("should accept large value", () => {
const result = SessionConfigSchema.parse({ maxHistoryMessages: 1000 })
expect(result.maxHistoryMessages).toBe(1000)
})
it("should reject zero", () => {
expect(() => SessionConfigSchema.parse({ maxHistoryMessages: 0 })).toThrow()
})
it("should reject negative number", () => {
expect(() => SessionConfigSchema.parse({ maxHistoryMessages: -10 })).toThrow()
})
it("should reject float", () => {
expect(() => SessionConfigSchema.parse({ maxHistoryMessages: 10.5 })).toThrow()
})
it("should reject non-number", () => {
expect(() => SessionConfigSchema.parse({ maxHistoryMessages: "100" })).toThrow()
})
})
describe("saveInputHistory", () => {
it("should accept true", () => {
const result = SessionConfigSchema.parse({ saveInputHistory: true })
expect(result.saveInputHistory).toBe(true)
})
it("should accept false", () => {
const result = SessionConfigSchema.parse({ saveInputHistory: false })
expect(result.saveInputHistory).toBe(false)
})
it("should reject non-boolean", () => {
expect(() => SessionConfigSchema.parse({ saveInputHistory: "yes" })).toThrow()
})
})
describe("partial config", () => {
it("should merge partial config with defaults", () => {
const result = SessionConfigSchema.parse({
maxHistoryMessages: 50,
})
expect(result).toEqual({
persistIndefinitely: true,
maxHistoryMessages: 50,
saveInputHistory: true,
})
})
it("should merge multiple partial fields", () => {
const result = SessionConfigSchema.parse({
persistIndefinitely: false,
saveInputHistory: false,
})
expect(result).toEqual({
persistIndefinitely: false,
maxHistoryMessages: 100,
saveInputHistory: false,
})
})
})
describe("full config", () => {
it("should accept valid full config", () => {
const config = {
persistIndefinitely: false,
maxHistoryMessages: 200,
saveInputHistory: false,
}
const result = SessionConfigSchema.parse(config)
expect(result).toEqual(config)
})
it("should accept all defaults explicitly", () => {
const config = {
persistIndefinitely: true,
maxHistoryMessages: 100,
saveInputHistory: true,
}
const result = SessionConfigSchema.parse(config)
expect(result).toEqual(config)
})
})
})

View File

@@ -218,28 +218,32 @@ describe("Input", () => {
it("should be active when multiline is true", () => {
const multiline = true
const lines = ["single line"]
const isMultilineActive = multiline === true || (multiline === "auto" && lines.length > 1)
const isMultilineActive =
multiline === true || (multiline === "auto" && lines.length > 1)
expect(isMultilineActive).toBe(true)
})
it("should not be active when multiline is false", () => {
const multiline = false
const lines = ["line1", "line2"]
const isMultilineActive = multiline === true || (multiline === "auto" && lines.length > 1)
const isMultilineActive =
multiline === true || (multiline === "auto" && lines.length > 1)
expect(isMultilineActive).toBe(false)
})
it("should be active in auto mode with multiple lines", () => {
const multiline = "auto"
const lines = ["line1", "line2"]
const isMultilineActive = multiline === true || (multiline === "auto" && lines.length > 1)
const isMultilineActive =
multiline === true || (multiline === "auto" && lines.length > 1)
expect(isMultilineActive).toBe(true)
})
it("should not be active in auto mode with single line", () => {
const multiline = "auto"
const lines = ["single line"]
const isMultilineActive = multiline === true || (multiline === "auto" && lines.length > 1)
const isMultilineActive =
multiline === true || (multiline === "auto" && lines.length > 1)
expect(isMultilineActive).toBe(false)
})
})

View File

@@ -3,7 +3,12 @@
*/
import { describe, expect, it } from "vitest"
import { getColorScheme, getContextColor, getRoleColor, getStatusColor } from "../../../../src/tui/utils/theme.js"
import {
getColorScheme,
getContextColor,
getRoleColor,
getStatusColor,
} from "../../../../src/tui/utils/theme.js"
describe("theme utilities", () => {
describe("getColorScheme", () => {