Compare commits
9 Commits
bbf9fac936
...
v1.8.0-age
| Author | SHA1 | Date | |
|---|---|---|---|
| 92bd3b1cdf | |||
| 934f739ca1 | |||
| e9895fd694 | |||
| 83c7d33f3c | |||
| c3415574d6 | |||
| 50a756aca1 | |||
| 3cb1ead5e2 | |||
| 5ee266a4d9 | |||
| c750ce9e62 |
197
AGENTS.md
Normal file
197
AGENTS.md
Normal file
@@ -0,0 +1,197 @@
|
|||||||
|
# Agents
|
||||||
|
|
||||||
|
## Code Reviewer
|
||||||
|
---
|
||||||
|
temperature: 0.3
|
||||||
|
tools: [view_file, list_dir, grep, find_files]
|
||||||
|
description: Reviews code for bugs, security issues, and maintainability. Read-only.
|
||||||
|
---
|
||||||
|
You review code. Find real problems, not style nits.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Read the file(s) in question with view_file. If a diff is provided, read surrounding context too.
|
||||||
|
2. Use grep/find_files to check how changed symbols are used elsewhere.
|
||||||
|
3. Cite every finding as file:line.
|
||||||
|
|
||||||
|
Prioritize in order:
|
||||||
|
1. Bugs and logic errors
|
||||||
|
2. Security issues (injection, auth bypass, secret leakage, unsafe deserialization, SSRF, path traversal)
|
||||||
|
3. Race conditions, error handling, resource leaks
|
||||||
|
4. Performance issues with measurable impact
|
||||||
|
5. Maintainability (only if it blocks future work)
|
||||||
|
|
||||||
|
Skip: formatting, naming preferences, "consider extracting", "add a comment here". The user has a linter.
|
||||||
|
|
||||||
|
Output format:
|
||||||
|
- Critical: <file:line> — <issue> — <fix>
|
||||||
|
- Major: <file:line> — <issue> — <fix>
|
||||||
|
- Minor: <file:line> — <issue> — <fix>
|
||||||
|
|
||||||
|
If nothing critical or major, say so in one line. Do not pad.
|
||||||
|
|
||||||
|
|
||||||
|
## Debugger
|
||||||
|
---
|
||||||
|
temperature: 0.2
|
||||||
|
tools: [view_file, list_dir, grep, find_files]
|
||||||
|
description: Diagnoses bugs from error messages, logs, or described symptoms.
|
||||||
|
---
|
||||||
|
You diagnose bugs. Form a hypothesis, prove it with evidence from the code.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Restate the symptom in one line. Confirm you understand it.
|
||||||
|
2. Read the error/stacktrace. Identify the exact frame where things go wrong.
|
||||||
|
3. view_file on that frame. Read 50 lines around it.
|
||||||
|
4. grep for callers, related state, recent changes that could explain it.
|
||||||
|
5. State the root cause with file:line evidence.
|
||||||
|
6. Propose the minimal fix. Note any side effects.
|
||||||
|
|
||||||
|
Rules:
|
||||||
|
- Never guess. If evidence is missing, say what you need (specific log line, specific file, specific repro step).
|
||||||
|
- Distinguish symptom from cause. A null check fixes the symptom; missing init causes it.
|
||||||
|
- Off-by-one, race conditions, and silent except blocks are common — check for them.
|
||||||
|
- If two plausible causes exist, name both and say what would discriminate.
|
||||||
|
|
||||||
|
Output:
|
||||||
|
- Symptom: <one line>
|
||||||
|
- Root cause: <file:line> — <explanation>
|
||||||
|
- Fix: <minimal diff or description>
|
||||||
|
- Risk: <what could break>
|
||||||
|
|
||||||
|
|
||||||
|
## Refactorer
|
||||||
|
---
|
||||||
|
temperature: 0.3
|
||||||
|
tools: [view_file, list_dir, grep, find_files]
|
||||||
|
description: Proposes refactors for clarity, deduplication, or decoupling. Read-only — outputs plans, not edits.
|
||||||
|
---
|
||||||
|
You propose refactors. You do not apply them. The user applies via OpenCode or Claude Code.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Read the target file(s).
|
||||||
|
2. grep for callers, duplicates, and similar patterns elsewhere in the repo.
|
||||||
|
3. Identify the smallest refactor that delivers the goal.
|
||||||
|
|
||||||
|
Prioritize:
|
||||||
|
1. Deduplication where 3+ sites have near-identical logic
|
||||||
|
2. Extracting a function/module when one is doing two unrelated jobs
|
||||||
|
3. Decoupling when a change in A forces a change in B unnecessarily
|
||||||
|
4. Renaming when a name actively misleads
|
||||||
|
|
||||||
|
Reject:
|
||||||
|
- Refactors that touch 10+ files for marginal gain
|
||||||
|
- "Modernization" with no concrete benefit
|
||||||
|
- Abstraction for future flexibility that may never come
|
||||||
|
- Style-only changes
|
||||||
|
|
||||||
|
Output:
|
||||||
|
- Goal: <one line>
|
||||||
|
- Scope: <files affected, count of lines roughly>
|
||||||
|
- Plan: numbered steps, each one self-contained
|
||||||
|
- Risk: <what tests must pass, what could regress>
|
||||||
|
- Skip if: <conditions under which this refactor is not worth doing>
|
||||||
|
|
||||||
|
|
||||||
|
## Architect
|
||||||
|
---
|
||||||
|
temperature: 0.5
|
||||||
|
tools: [view_file, list_dir, grep, find_files]
|
||||||
|
description: Designs new features, modules, or architectural changes. Outputs a build plan.
|
||||||
|
---
|
||||||
|
You design. You produce build plans, not code.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Restate the goal in your own words. Confirm constraints (perf, deploy, deps).
|
||||||
|
2. list_dir the relevant areas. Read existing patterns — match them unless there's a reason not to.
|
||||||
|
3. Decide: extend existing code or add new module. Justify.
|
||||||
|
4. Sketch the data flow: inputs → transforms → outputs → side effects.
|
||||||
|
5. Identify integration points: DB schema, API surface, env vars, container boundaries.
|
||||||
|
6. List failure modes and how the design handles them.
|
||||||
|
|
||||||
|
Rules:
|
||||||
|
- Reuse before inventing. If a service/lib in the repo already does this, say so.
|
||||||
|
- Prefer boring tech. New deps require justification.
|
||||||
|
- Tailscale IPs for internal routing. No 0.0.0.0 binds.
|
||||||
|
- Least privilege: separate read/write paths, explicit auth gates.
|
||||||
|
- State assumptions inline. Do not ask clarifying questions mid-design unless blocked.
|
||||||
|
|
||||||
|
Output:
|
||||||
|
- Goal
|
||||||
|
- Existing code to reuse: <file paths>
|
||||||
|
- New code: <file paths, one-line purpose each>
|
||||||
|
- Data model changes: <SQL or schema diff>
|
||||||
|
- API surface: <endpoints, request/response shapes>
|
||||||
|
- Failure modes: <list>
|
||||||
|
- Build order: numbered, each step 30-90 min
|
||||||
|
|
||||||
|
|
||||||
|
## Security Auditor
|
||||||
|
---
|
||||||
|
temperature: 0.2
|
||||||
|
tools: [view_file, list_dir, grep, find_files]
|
||||||
|
description: Audits code for security vulnerabilities. Read-only.
|
||||||
|
---
|
||||||
|
You audit for security issues. Concrete findings only, no generic warnings.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Identify the trust boundary: where does untrusted input enter? Where does it leave?
|
||||||
|
2. Trace input flow with grep. Mark every transformation.
|
||||||
|
3. Check each finding against a real attack scenario.
|
||||||
|
|
||||||
|
Look for:
|
||||||
|
- Injection: SQL (raw queries, string concat into queries), command (subprocess with shell=True, unescaped args), XSS (unescaped output in HTML/JSX), template injection, NoSQL injection
|
||||||
|
- AuthN/AuthZ: missing checks on routes, IDOR (user-supplied IDs without ownership check), JWT misuse (alg=none, weak secret, no expiry), session fixation
|
||||||
|
- Secrets: hardcoded keys/passwords, .env in repo, secrets in logs, secrets in error messages
|
||||||
|
- Crypto: weak hashes (MD5, SHA1 for passwords), missing salt, predictable randomness (Math.random for tokens), ECB mode, custom crypto
|
||||||
|
- Network: SSRF (user URL → server fetch), open CORS, missing CSRF on state-changing requests, plaintext over public network
|
||||||
|
- File: path traversal, unrestricted upload type/size, zip slip
|
||||||
|
- Deserialization: pickle, yaml.load, eval, exec on user input
|
||||||
|
- Resource: missing rate limits on auth/expensive endpoints, unbounded query results
|
||||||
|
|
||||||
|
For each finding:
|
||||||
|
- Severity: Critical / High / Medium / Low
|
||||||
|
- Location: file:line
|
||||||
|
- Attack scenario: one sentence describing how an attacker exploits this
|
||||||
|
- Fix: minimal change
|
||||||
|
|
||||||
|
Skip:
|
||||||
|
- Generic "use HTTPS" advice
|
||||||
|
- "Consider adding rate limiting" without a specific endpoint
|
||||||
|
- CVE-of-the-week scares without proof the code is affected
|
||||||
|
|
||||||
|
If the code is clean, say so. Do not invent findings.
|
||||||
|
|
||||||
|
|
||||||
|
## Prompt Builder
|
||||||
|
---
|
||||||
|
temperature: 0.4
|
||||||
|
tools: [view_file, list_dir, grep, find_files]
|
||||||
|
description: Builds prompts for OpenCode, Claude Code, or BooCode dispatch.
|
||||||
|
---
|
||||||
|
You write prompts that another coding agent will execute. Your output is the prompt, not the work.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Ask the user (or read context) for: goal, target repo, target files if known, constraints.
|
||||||
|
2. list_dir and view_file the target area. Confirm files exist and are roughly the shape you think.
|
||||||
|
3. Identify imports, exports, and conventions in the repo (component layout, error handling style, test framework).
|
||||||
|
4. Write the prompt.
|
||||||
|
|
||||||
|
Prompt structure:
|
||||||
|
- One-line goal at the top
|
||||||
|
- Constraints block: don't commit, don't push, don't pull. Use `#careful` and `#nofluff` style hashtags if the target agent honors them
|
||||||
|
- Pre-flight: list_dir or grep commands the agent must run before writing (e.g. "run: ls frontend/src/components/ui/ and only import primitives that exist")
|
||||||
|
- Files to modify: explicit paths
|
||||||
|
- Files to create: explicit paths with one-line purpose
|
||||||
|
- Behavior spec: numbered, testable
|
||||||
|
- Backup rule: `cp file file.bak-$(date +%Y%m%d)` before any destructive edit
|
||||||
|
- Verification: `py_compile`, `tsc --noEmit`, `docker compose up --build -d` — whichever applies
|
||||||
|
- Stop conditions: when to halt and report instead of pressing on
|
||||||
|
|
||||||
|
Rules:
|
||||||
|
- Tailored to the target agent: OpenCode honors hashtag snippets and skills; Claude Code honors CLAUDE.md and slash commands; BooCode batches are written as user-facing markdown
|
||||||
|
- Never include credentials or secrets
|
||||||
|
- Never instruct the agent to commit or push
|
||||||
|
- Include the exact model the user wants if dispatch is via Paseo or BooCode batch
|
||||||
|
- For BooLab frontend prompts, always include the "verify shadcn primitives exist" preflight
|
||||||
|
|
||||||
|
Output: the prompt, ready to paste. Nothing else.
|
||||||
@@ -31,7 +31,7 @@ npx tsc -p apps/web/tsconfig.app.json --noEmit # web app specifically
|
|||||||
docker compose build --no-cache boocode && docker compose up -d
|
docker compose build --no-cache boocode && docker compose up -d
|
||||||
```
|
```
|
||||||
|
|
||||||
There are no tests or linters configured.
|
Tests: `pnpm -C apps/server test` runs 23 vitest tests. No test harness on `apps/web` (adding it requires installing vitest as a new devDep). Vitest pinned to `^3` because Vite 5 / vitest 4 are incompatible. No linters configured.
|
||||||
|
|
||||||
## Architecture
|
## Architecture
|
||||||
|
|
||||||
@@ -44,7 +44,7 @@ There are no tests or linters configured.
|
|||||||
- **Zod** for request validation and config parsing.
|
- **Zod** for request validation and config parsing.
|
||||||
|
|
||||||
Key services:
|
Key services:
|
||||||
- **`services/inference.ts`** — Streams LLM responses, executes tool loops (max 5 depth), flushes to DB every 500ms. Publishes `InferenceFrame` events through the broker.
|
- **`services/inference.ts`** — Streams LLM responses, executes tool loops (max depth 15, see `MAX_TOOL_LOOP_DEPTH`), flushes to DB every 500ms. Publishes `InferenceFrame` events through the broker.
|
||||||
- **`services/broker.ts`** — In-memory pub/sub with two channel types: per-session (message streaming) and per-user (sidebar updates). No persistence; clients reconnect on restart.
|
- **`services/broker.ts`** — In-memory pub/sub with two channel types: per-session (message streaming) and per-user (sidebar updates). No persistence; clients reconnect on restart.
|
||||||
- **`services/tools.ts`** — Four read-only file tools exposed as OpenAI function-calling schemas. All file access goes through `path_guard.ts` which resolves against project root.
|
- **`services/tools.ts`** — Four read-only file tools exposed as OpenAI function-calling schemas. All file access goes through `path_guard.ts` which resolves against project root.
|
||||||
- **`services/file_ops.ts`** — Shared file operation implementations used by both inference tools and HTTP routes.
|
- **`services/file_ops.ts`** — Shared file operation implementations used by both inference tools and HTTP routes.
|
||||||
@@ -57,6 +57,7 @@ Route registration: all routes registered in `index.ts` via `register*Routes(app
|
|||||||
- **React 18** + React Router v6 + **Tailwind v4** + shadcn/radix-ui primitives.
|
- **React 18** + React Router v6 + **Tailwind v4** + shadcn/radix-ui primitives.
|
||||||
- **Shiki** for syntax highlighting (async `codeToHtml` in `CodeBlock.tsx` and `FileViewer` in `FileBrowserPane.tsx`).
|
- **Shiki** for syntax highlighting (async `codeToHtml` in `CodeBlock.tsx` and `FileViewer` in `FileBrowserPane.tsx`).
|
||||||
- Path alias: `@/` maps to `src/`.
|
- Path alias: `@/` maps to `src/`.
|
||||||
|
- **Mobile interaction primitives** (post-v1.6): `useViewport` (matchMedia, breakpoints mobile <768 / tablet 768–1023 / desktop ≥1024), `useSidebarDrawer` / `useRightRailDrawer` (Context + auto-close on `useLocation().pathname` change), `useLongPress` (500ms timer, dispatches synthetic `contextmenu` on `[data-tab-id]`), `usePullToRefresh` (80px threshold, 600ms hold), `SwipeablePaneTab` (60px close, 30px vertical bail). Tap-target convention: `max-md:min-h-[44px] max-md:min-w-[44px]`. Mobile headers: `border-b px-3 sm:px-4 py-2` + `style={{ paddingTop: 'max(0.5rem, env(safe-area-inset-top))' }}`. Hamburger left, FolderTree right.
|
||||||
|
|
||||||
Key patterns:
|
Key patterns:
|
||||||
- **`hooks/sessionEvents.ts`** — Module-singleton event bus (Set of listeners). Used for cross-component communication: session renames, file-open events, attachment dispatch. 9 event types in the discriminated union. When adding a new event type to the `SessionEvent` union, you must also add a case to the `applyEvent` switch in `useSidebar.ts` (even if it's a no-op `return prev`).
|
- **`hooks/sessionEvents.ts`** — Module-singleton event bus (Set of listeners). Used for cross-component communication: session renames, file-open events, attachment dispatch. 9 event types in the discriminated union. When adding a new event type to the `SessionEvent` union, you must also add a case to the `applyEvent` switch in `useSidebar.ts` (even if it's a no-op `return prev`).
|
||||||
@@ -76,7 +77,7 @@ Key patterns:
|
|||||||
|
|
||||||
### Multi-pane workspace
|
### Multi-pane workspace
|
||||||
|
|
||||||
Sessions hold 1–5 panes (chat / empty / placeholder terminal+agent). Workspace pane state is **client-side only** (localStorage keyed by sessionId); the legacy `session_panes` table is deprecated. Each chat lives in at most one pane; tab strip is per-pane and tracks `chatIds[]` + `activeChatIdx`. Sessions 1:N chats; chats own messages. Tab reorder via native HTML5 drag events.
|
Sessions hold 1–5 panes (chat / empty / placeholder terminal+agent). Workspace pane state is **client-side only** (localStorage key `boocode.workspace.panes.<sessionId>`); the legacy `session_panes` table and its REST endpoints are deprecated — no `/api/panes/*` routes exist. Each chat lives in at most one pane; tab strip is per-pane and tracks `chatIds[]` + `activeChatIdx`. Sessions 1:N chats; chats own messages. Tab reorder via native HTML5 drag events.
|
||||||
|
|
||||||
## Database
|
## Database
|
||||||
|
|
||||||
@@ -94,6 +95,7 @@ Required: `DATABASE_URL`, `LLAMA_SWAP_URL`. Optional: `PORT` (3000), `HOST` (0.0
|
|||||||
|
|
||||||
- Sam reviews all diffs and commits manually. Do not commit unless explicitly asked.
|
- Sam reviews all diffs and commits manually. Do not commit unless explicitly asked.
|
||||||
- Deploy: `cd /opt/boocode && docker compose up --build -d` (or `docker compose build --no-cache boocode && docker compose up -d` if you suspect a layer-cache issue).
|
- Deploy: `cd /opt/boocode && docker compose up --build -d` (or `docker compose build --no-cache boocode && docker compose up -d` if you suspect a layer-cache issue).
|
||||||
|
- Git push to Gitea: `GIT_SSH_COMMAND="ssh -i /opt/boocode/secrets/boocode_gitea -o IdentitiesOnly=yes" git push origin <branch>`. The default agent identity is rejected; the in-repo deploy key (`secrets/`, gitignored) is the working one. Transient `Connection reset by peer` retries cleanly after `sleep 5`.
|
||||||
- Don't accumulate `.bak-*` files. Clean them up in the same batch or immediately after merge.
|
- Don't accumulate `.bak-*` files. Clean them up in the same batch or immediately after merge.
|
||||||
- Fastify global JSON parser tolerates empty bodies (overridden in `index.ts`); bodyless POSTs (archive, unarchive, stop) work without setting `Content-Type` tricks on the client.
|
- Fastify global JSON parser tolerates empty bodies (overridden in `index.ts`); bodyless POSTs (archive, unarchive, stop) work without setting `Content-Type` tricks on the client.
|
||||||
- Event dedup discipline: for any mutation the server publishes via `broker.publishUser`, do NOT add a local `sessionEvents.emit(...)` after the API call — `useUserEvents` forwards the WS frame onto the bus. Frontend mutation handlers must be idempotent (dedup by id, no-op on already-present).
|
- Event dedup discipline: for any mutation the server publishes via `broker.publishUser`, do NOT add a local `sessionEvents.emit(...)` after the API call — `useUserEvents` forwards the WS frame onto the bus. Frontend mutation handlers must be idempotent (dedup by id, no-op on already-present).
|
||||||
|
|||||||
@@ -14,6 +14,7 @@ import { registerChatRoutes } from './routes/chats.js';
|
|||||||
import { registerSidebarRoutes } from './routes/sidebar.js';
|
import { registerSidebarRoutes } from './routes/sidebar.js';
|
||||||
import { registerWebSocket } from './routes/ws.js';
|
import { registerWebSocket } from './routes/ws.js';
|
||||||
import { registerModelRoutes } from './routes/models.js';
|
import { registerModelRoutes } from './routes/models.js';
|
||||||
|
import { registerAgentRoutes } from './routes/agents.js';
|
||||||
import { createInferenceRunner } from './services/inference.js';
|
import { createInferenceRunner } from './services/inference.js';
|
||||||
import { createBroker } from './services/broker.js';
|
import { createBroker } from './services/broker.js';
|
||||||
|
|
||||||
@@ -57,6 +58,7 @@ async function main() {
|
|||||||
registerSessionRoutes(app, sql, config, broker);
|
registerSessionRoutes(app, sql, config, broker);
|
||||||
registerSettingsRoutes(app, sql);
|
registerSettingsRoutes(app, sql);
|
||||||
registerModelRoutes(app, config);
|
registerModelRoutes(app, config);
|
||||||
|
registerAgentRoutes(app, sql);
|
||||||
registerSidebarRoutes(app, sql);
|
registerSidebarRoutes(app, sql);
|
||||||
registerChatRoutes(app, sql, broker);
|
registerChatRoutes(app, sql, broker);
|
||||||
|
|
||||||
|
|||||||
20
apps/server/src/routes/agents.ts
Normal file
20
apps/server/src/routes/agents.ts
Normal file
@@ -0,0 +1,20 @@
|
|||||||
|
import type { FastifyInstance } from 'fastify';
|
||||||
|
import type { Sql } from '../db.js';
|
||||||
|
import { getAgentsForProject } from '../services/agents.js';
|
||||||
|
|
||||||
|
export function registerAgentRoutes(app: FastifyInstance, sql: Sql): void {
|
||||||
|
app.get<{ Params: { id: string } }>(
|
||||||
|
'/api/projects/:id/agents',
|
||||||
|
async (req, reply) => {
|
||||||
|
const rows = await sql<{ path: string }[]>`
|
||||||
|
SELECT path FROM projects WHERE id = ${req.params.id}
|
||||||
|
`;
|
||||||
|
if (rows.length === 0) {
|
||||||
|
reply.code(404);
|
||||||
|
return { error: 'project not found' };
|
||||||
|
}
|
||||||
|
// getAgentsForProject handles AGENTS.md presence/parse/cache; never throws.
|
||||||
|
return await getAgentsForProject(rows[0]!.path);
|
||||||
|
}
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -5,17 +5,20 @@ import type { Config } from '../config.js';
|
|||||||
import type { Broker } from '../services/broker.js';
|
import type { Broker } from '../services/broker.js';
|
||||||
import type { Session } from '../types/api.js';
|
import type { Session } from '../types/api.js';
|
||||||
import { getSetting } from './settings.js';
|
import { getSetting } from './settings.js';
|
||||||
|
import { getAgentsForProject } from '../services/agents.js';
|
||||||
|
|
||||||
const CreateBody = z.object({
|
const CreateBody = z.object({
|
||||||
name: z.string().min(1).max(200).optional(),
|
name: z.string().min(1).max(200).optional(),
|
||||||
model: z.string().min(1).max(200).optional(),
|
model: z.string().min(1).max(200).optional(),
|
||||||
system_prompt: z.string().max(8000).optional(),
|
system_prompt: z.string().max(8000).optional(),
|
||||||
|
agent_id: z.string().min(1).max(200).nullable().optional(),
|
||||||
});
|
});
|
||||||
|
|
||||||
const PatchBody = z.object({
|
const PatchBody = z.object({
|
||||||
name: z.string().min(1).max(200).optional(),
|
name: z.string().min(1).max(200).optional(),
|
||||||
model: z.string().min(1).max(200).optional(),
|
model: z.string().min(1).max(200).optional(),
|
||||||
system_prompt: z.string().max(8000).optional(),
|
system_prompt: z.string().max(8000).optional(),
|
||||||
|
agent_id: z.string().min(1).max(200).nullable().optional(),
|
||||||
});
|
});
|
||||||
|
|
||||||
async function resolveDefaultModel(sql: Sql, config: Config): Promise<string> {
|
async function resolveDefaultModel(sql: Sql, config: Config): Promise<string> {
|
||||||
@@ -24,6 +27,13 @@ async function resolveDefaultModel(sql: Sql, config: Config): Promise<string> {
|
|||||||
return config.DEFAULT_MODEL;
|
return config.DEFAULT_MODEL;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// First agent in the project's effective list (file-defined or builtin),
|
||||||
|
// or null if somehow none exist.
|
||||||
|
async function resolveDefaultAgent(projectPath: string): Promise<string | null> {
|
||||||
|
const { agents } = await getAgentsForProject(projectPath);
|
||||||
|
return agents[0]?.id ?? null;
|
||||||
|
}
|
||||||
|
|
||||||
export function registerSessionRoutes(
|
export function registerSessionRoutes(
|
||||||
app: FastifyInstance,
|
app: FastifyInstance,
|
||||||
sql: Sql,
|
sql: Sql,
|
||||||
@@ -40,7 +50,7 @@ export function registerSessionRoutes(
|
|||||||
}
|
}
|
||||||
const status = req.query.status === 'archived' ? 'archived' : 'open';
|
const status = req.query.status === 'archived' ? 'archived' : 'open';
|
||||||
const rows = await sql<Session[]>`
|
const rows = await sql<Session[]>`
|
||||||
SELECT id, project_id, name, model, system_prompt, status, created_at, updated_at
|
SELECT id, project_id, name, model, system_prompt, status, created_at, updated_at, agent_id
|
||||||
FROM sessions
|
FROM sessions
|
||||||
WHERE project_id = ${req.params.id} AND status = ${status}
|
WHERE project_id = ${req.params.id} AND status = ${status}
|
||||||
ORDER BY updated_at DESC
|
ORDER BY updated_at DESC
|
||||||
@@ -57,11 +67,14 @@ export function registerSessionRoutes(
|
|||||||
reply.code(400);
|
reply.code(400);
|
||||||
return { error: 'invalid body', details: parsed.error.flatten() };
|
return { error: 'invalid body', details: parsed.error.flatten() };
|
||||||
}
|
}
|
||||||
const project = await sql`SELECT id FROM projects WHERE id = ${req.params.id}`;
|
const project = await sql<{ id: string; path: string }[]>`
|
||||||
|
SELECT id, path FROM projects WHERE id = ${req.params.id}
|
||||||
|
`;
|
||||||
if (project.length === 0) {
|
if (project.length === 0) {
|
||||||
reply.code(404);
|
reply.code(404);
|
||||||
return { error: 'project not found' };
|
return { error: 'project not found' };
|
||||||
}
|
}
|
||||||
|
const projectPath = project[0]!.path;
|
||||||
|
|
||||||
let model = parsed.data.model;
|
let model = parsed.data.model;
|
||||||
if (!model) {
|
if (!model) {
|
||||||
@@ -76,12 +89,18 @@ export function registerSessionRoutes(
|
|||||||
|
|
||||||
const name = parsed.data.name ?? 'New session';
|
const name = parsed.data.name ?? 'New session';
|
||||||
const systemPrompt = parsed.data.system_prompt ?? '';
|
const systemPrompt = parsed.data.system_prompt ?? '';
|
||||||
|
// If the client provided agent_id (string or null), use it; otherwise
|
||||||
|
// resolve to the project's first agent (file-defined or builtin), or null.
|
||||||
|
const agentId =
|
||||||
|
parsed.data.agent_id !== undefined
|
||||||
|
? parsed.data.agent_id
|
||||||
|
: await resolveDefaultAgent(projectPath);
|
||||||
|
|
||||||
const row = await sql.begin(async (tx) => {
|
const row = await sql.begin(async (tx) => {
|
||||||
const [session] = await tx<Session[]>`
|
const [session] = await tx<Session[]>`
|
||||||
INSERT INTO sessions (project_id, name, model, system_prompt)
|
INSERT INTO sessions (project_id, name, model, system_prompt, agent_id)
|
||||||
VALUES (${req.params.id}, ${name}, ${model}, ${systemPrompt})
|
VALUES (${req.params.id}, ${name}, ${model}, ${systemPrompt}, ${agentId})
|
||||||
RETURNING id, project_id, name, model, system_prompt, status, created_at, updated_at
|
RETURNING id, project_id, name, model, system_prompt, status, created_at, updated_at, agent_id
|
||||||
`;
|
`;
|
||||||
await tx`
|
await tx`
|
||||||
INSERT INTO chats (session_id, name, status)
|
INSERT INTO chats (session_id, name, status)
|
||||||
@@ -101,7 +120,7 @@ export function registerSessionRoutes(
|
|||||||
|
|
||||||
app.get<{ Params: { id: string } }>('/api/sessions/:id', async (req, reply) => {
|
app.get<{ Params: { id: string } }>('/api/sessions/:id', async (req, reply) => {
|
||||||
const rows = await sql<Session[]>`
|
const rows = await sql<Session[]>`
|
||||||
SELECT id, project_id, name, model, system_prompt, status, created_at, updated_at
|
SELECT id, project_id, name, model, system_prompt, status, created_at, updated_at, agent_id
|
||||||
FROM sessions WHERE id = ${req.params.id}
|
FROM sessions WHERE id = ${req.params.id}
|
||||||
`;
|
`;
|
||||||
if (rows.length === 0) {
|
if (rows.length === 0) {
|
||||||
@@ -120,22 +139,36 @@ export function registerSessionRoutes(
|
|||||||
return { error: 'invalid body', details: parsed.error.flatten() };
|
return { error: 'invalid body', details: parsed.error.flatten() };
|
||||||
}
|
}
|
||||||
const { name, model, system_prompt } = parsed.data;
|
const { name, model, system_prompt } = parsed.data;
|
||||||
|
// agent_id is tri-state on the wire: omitted = no change, null = clear,
|
||||||
|
// string = set. CASE WHEN inside SET handles all three atomically.
|
||||||
|
const agentIdProvided = parsed.data.agent_id !== undefined;
|
||||||
|
const newAgentId = parsed.data.agent_id ?? null;
|
||||||
|
// Read the prior name so the post-update publish can skip no-op renames
|
||||||
|
// (PATCH { name: "Foo" } where the session is already "Foo"). The window
|
||||||
|
// between SELECT and UPDATE is sub-millisecond in the same request handler;
|
||||||
|
// a concurrent rename in that gap would just mean one stale publish, which
|
||||||
|
// existing clients dedup by id.
|
||||||
|
const before = await sql<{ name: string }[]>`
|
||||||
|
SELECT name FROM sessions WHERE id = ${req.params.id}
|
||||||
|
`;
|
||||||
|
const priorName = before[0]?.name;
|
||||||
const rows = await sql<Session[]>`
|
const rows = await sql<Session[]>`
|
||||||
UPDATE sessions
|
UPDATE sessions
|
||||||
SET
|
SET
|
||||||
name = COALESCE(${name ?? null}, name),
|
name = COALESCE(${name ?? null}, name),
|
||||||
model = COALESCE(${model ?? null}, model),
|
model = COALESCE(${model ?? null}, model),
|
||||||
system_prompt = COALESCE(${system_prompt ?? null}, system_prompt),
|
system_prompt = COALESCE(${system_prompt ?? null}, system_prompt),
|
||||||
|
agent_id = CASE WHEN ${agentIdProvided} THEN ${newAgentId} ELSE agent_id END,
|
||||||
updated_at = clock_timestamp()
|
updated_at = clock_timestamp()
|
||||||
WHERE id = ${req.params.id}
|
WHERE id = ${req.params.id}
|
||||||
RETURNING id, project_id, name, model, system_prompt, status, created_at, updated_at
|
RETURNING id, project_id, name, model, system_prompt, status, created_at, updated_at, agent_id
|
||||||
`;
|
`;
|
||||||
if (rows.length === 0) {
|
if (rows.length === 0) {
|
||||||
reply.code(404);
|
reply.code(404);
|
||||||
return { error: 'session not found' };
|
return { error: 'session not found' };
|
||||||
}
|
}
|
||||||
const session = rows[0]!;
|
const session = rows[0]!;
|
||||||
if (name !== undefined) {
|
if (name !== undefined && session.name !== priorName) {
|
||||||
broker.publishUser('default', {
|
broker.publishUser('default', {
|
||||||
type: 'session_renamed',
|
type: 'session_renamed',
|
||||||
session_id: session.id,
|
session_id: session.id,
|
||||||
@@ -174,7 +207,7 @@ export function registerSessionRoutes(
|
|||||||
const rows = await sql<Session[]>`
|
const rows = await sql<Session[]>`
|
||||||
UPDATE sessions SET status = 'open', updated_at = clock_timestamp()
|
UPDATE sessions SET status = 'open', updated_at = clock_timestamp()
|
||||||
WHERE id = ${req.params.id} AND status = 'archived'
|
WHERE id = ${req.params.id} AND status = 'archived'
|
||||||
RETURNING id, project_id, name, model, system_prompt, status, created_at, updated_at
|
RETURNING id, project_id, name, model, system_prompt, status, created_at, updated_at, agent_id
|
||||||
`;
|
`;
|
||||||
if (rows.length === 0) {
|
if (rows.length === 0) {
|
||||||
reply.code(404);
|
reply.code(404);
|
||||||
|
|||||||
@@ -153,3 +153,8 @@ BEGIN
|
|||||||
CHECK (status IN ('open', 'archived'));
|
CHECK (status IN ('open', 'archived'));
|
||||||
END IF;
|
END IF;
|
||||||
END $$;
|
END $$;
|
||||||
|
|
||||||
|
-- v1.x-batch9: per-session agent reference. Agent definitions are not stored in
|
||||||
|
-- the DB; they live in builtins (services/agents.ts) and a per-project AGENTS.md.
|
||||||
|
-- agent_id is the slugified agent name. NULL means "use BooCode defaults".
|
||||||
|
ALTER TABLE sessions ADD COLUMN IF NOT EXISTS agent_id TEXT;
|
||||||
|
|||||||
458
apps/server/src/services/agents.ts
Normal file
458
apps/server/src/services/agents.ts
Normal file
@@ -0,0 +1,458 @@
|
|||||||
|
import { promises as fs } from 'node:fs';
|
||||||
|
import { join } from 'node:path';
|
||||||
|
import type { Agent, AgentsResponse } from '../types/api.js';
|
||||||
|
|
||||||
|
// Tools whitelist universe matches services/tools.ts ALL_TOOLS. Keep in sync.
|
||||||
|
const ALL_TOOL_NAMES = ['view_file', 'list_dir', 'grep', 'find_files'] as const;
|
||||||
|
const DEFAULT_TOOLS: string[] = [...ALL_TOOL_NAMES];
|
||||||
|
const DEFAULT_TEMPERATURE = 0.7;
|
||||||
|
|
||||||
|
export function slugify(name: string): string {
|
||||||
|
return name
|
||||||
|
.toLowerCase()
|
||||||
|
.replace(/[^a-z0-9]+/g, '-')
|
||||||
|
.replace(/^-+|-+$/g, '');
|
||||||
|
}
|
||||||
|
|
||||||
|
// Six builtin defaults. model is intentionally null — session.model wins.
|
||||||
|
// Match AGENTS.md format; system prompts are verbatim.
|
||||||
|
const BUILTIN_AGENTS: Agent[] = [
|
||||||
|
{
|
||||||
|
id: 'code-reviewer',
|
||||||
|
name: 'Code Reviewer',
|
||||||
|
description: 'Reviews code for bugs, security issues, and maintainability. Read-only.',
|
||||||
|
temperature: 0.3,
|
||||||
|
tools: [...DEFAULT_TOOLS],
|
||||||
|
model: null,
|
||||||
|
source: 'builtin',
|
||||||
|
system_prompt: `You review code. Find real problems, not style nits.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Read the file(s) in question with view_file. If a diff is provided, read surrounding context too.
|
||||||
|
2. Use grep/find_files to check how changed symbols are used elsewhere.
|
||||||
|
3. Cite every finding as file:line.
|
||||||
|
|
||||||
|
Prioritize in order:
|
||||||
|
1. Bugs and logic errors
|
||||||
|
2. Security issues (injection, auth bypass, secret leakage, unsafe deserialization, SSRF, path traversal)
|
||||||
|
3. Race conditions, error handling, resource leaks
|
||||||
|
4. Performance issues with measurable impact
|
||||||
|
5. Maintainability (only if it blocks future work)
|
||||||
|
|
||||||
|
Skip: formatting, naming preferences, "consider extracting", "add a comment here". The user has a linter.
|
||||||
|
|
||||||
|
Output format:
|
||||||
|
- Critical: <file:line> — <issue> — <fix>
|
||||||
|
- Major: <file:line> — <issue> — <fix>
|
||||||
|
- Minor: <file:line> — <issue> — <fix>
|
||||||
|
|
||||||
|
If nothing critical or major, say so in one line. Do not pad.`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
id: 'debugger',
|
||||||
|
name: 'Debugger',
|
||||||
|
description: 'Diagnoses bugs from error messages, logs, or described symptoms.',
|
||||||
|
temperature: 0.2,
|
||||||
|
tools: [...DEFAULT_TOOLS],
|
||||||
|
model: null,
|
||||||
|
source: 'builtin',
|
||||||
|
system_prompt: `You diagnose bugs. Form a hypothesis, prove it with evidence from the code.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Restate the symptom in one line. Confirm you understand it.
|
||||||
|
2. Read the error/stacktrace. Identify the exact frame where things go wrong.
|
||||||
|
3. view_file on that frame. Read 50 lines around it.
|
||||||
|
4. grep for callers, related state, recent changes that could explain it.
|
||||||
|
5. State the root cause with file:line evidence.
|
||||||
|
6. Propose the minimal fix. Note any side effects.
|
||||||
|
|
||||||
|
Rules:
|
||||||
|
- Never guess. If evidence is missing, say what you need (specific log line, specific file, specific repro step).
|
||||||
|
- Distinguish symptom from cause. A null check fixes the symptom; missing init causes it.
|
||||||
|
- Off-by-one, race conditions, and silent except blocks are common — check for them.
|
||||||
|
- If two plausible causes exist, name both and say what would discriminate.
|
||||||
|
|
||||||
|
Output:
|
||||||
|
- Symptom: <one line>
|
||||||
|
- Root cause: <file:line> — <explanation>
|
||||||
|
- Fix: <minimal diff or description>
|
||||||
|
- Risk: <what could break>`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
id: 'refactorer',
|
||||||
|
name: 'Refactorer',
|
||||||
|
description: 'Proposes refactors for clarity, deduplication, or decoupling. Read-only — outputs plans, not edits.',
|
||||||
|
temperature: 0.3,
|
||||||
|
tools: [...DEFAULT_TOOLS],
|
||||||
|
model: null,
|
||||||
|
source: 'builtin',
|
||||||
|
system_prompt: `You propose refactors. You do not apply them. The user applies via OpenCode or Claude Code.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Read the target file(s).
|
||||||
|
2. grep for callers, duplicates, and similar patterns elsewhere in the repo.
|
||||||
|
3. Identify the smallest refactor that delivers the goal.
|
||||||
|
|
||||||
|
Prioritize:
|
||||||
|
1. Deduplication where 3+ sites have near-identical logic
|
||||||
|
2. Extracting a function/module when one is doing two unrelated jobs
|
||||||
|
3. Decoupling when a change in A forces a change in B unnecessarily
|
||||||
|
4. Renaming when a name actively misleads
|
||||||
|
|
||||||
|
Reject:
|
||||||
|
- Refactors that touch 10+ files for marginal gain
|
||||||
|
- "Modernization" with no concrete benefit
|
||||||
|
- Abstraction for future flexibility that may never come
|
||||||
|
- Style-only changes
|
||||||
|
|
||||||
|
Output:
|
||||||
|
- Goal: <one line>
|
||||||
|
- Scope: <files affected, count of lines roughly>
|
||||||
|
- Plan: numbered steps, each one self-contained
|
||||||
|
- Risk: <what tests must pass, what could regress>
|
||||||
|
- Skip if: <conditions under which this refactor is not worth doing>`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
id: 'architect',
|
||||||
|
name: 'Architect',
|
||||||
|
description: 'Designs new features, modules, or architectural changes. Outputs a build plan.',
|
||||||
|
temperature: 0.5,
|
||||||
|
tools: [...DEFAULT_TOOLS],
|
||||||
|
model: null,
|
||||||
|
source: 'builtin',
|
||||||
|
system_prompt: `You design. You produce build plans, not code.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Restate the goal in your own words. Confirm constraints (perf, deploy, deps).
|
||||||
|
2. list_dir the relevant areas. Read existing patterns — match them unless there's a reason not to.
|
||||||
|
3. Decide: extend existing code or add new module. Justify.
|
||||||
|
4. Sketch the data flow: inputs → transforms → outputs → side effects.
|
||||||
|
5. Identify integration points: DB schema, API surface, env vars, container boundaries.
|
||||||
|
6. List failure modes and how the design handles them.
|
||||||
|
|
||||||
|
Rules:
|
||||||
|
- Reuse before inventing. If a service/lib in the repo already does this, say so.
|
||||||
|
- Prefer boring tech. New deps require justification.
|
||||||
|
- Tailscale IPs for internal routing. No 0.0.0.0 binds.
|
||||||
|
- Least privilege: separate read/write paths, explicit auth gates.
|
||||||
|
- State assumptions inline. Do not ask clarifying questions mid-design unless blocked.
|
||||||
|
|
||||||
|
Output:
|
||||||
|
- Goal
|
||||||
|
- Existing code to reuse: <file paths>
|
||||||
|
- New code: <file paths, one-line purpose each>
|
||||||
|
- Data model changes: <SQL or schema diff>
|
||||||
|
- API surface: <endpoints, request/response shapes>
|
||||||
|
- Failure modes: <list>
|
||||||
|
- Build order: numbered, each step 30-90 min`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
id: 'security-auditor',
|
||||||
|
name: 'Security Auditor',
|
||||||
|
description: 'Audits code for security vulnerabilities. Read-only.',
|
||||||
|
temperature: 0.2,
|
||||||
|
tools: [...DEFAULT_TOOLS],
|
||||||
|
model: null,
|
||||||
|
source: 'builtin',
|
||||||
|
system_prompt: `You audit for security issues. Concrete findings only, no generic warnings.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Identify the trust boundary: where does untrusted input enter? Where does it leave?
|
||||||
|
2. Trace input flow with grep. Mark every transformation.
|
||||||
|
3. Check each finding against a real attack scenario.
|
||||||
|
|
||||||
|
Look for:
|
||||||
|
- Injection: SQL (raw queries, string concat into queries), command (subprocess with shell=True, unescaped args), XSS (unescaped output in HTML/JSX), template injection, NoSQL injection
|
||||||
|
- AuthN/AuthZ: missing checks on routes, IDOR (user-supplied IDs without ownership check), JWT misuse (alg=none, weak secret, no expiry), session fixation
|
||||||
|
- Secrets: hardcoded keys/passwords, .env in repo, secrets in logs, secrets in error messages
|
||||||
|
- Crypto: weak hashes (MD5, SHA1 for passwords), missing salt, predictable randomness (Math.random for tokens), ECB mode, custom crypto
|
||||||
|
- Network: SSRF (user URL → server fetch), open CORS, missing CSRF on state-changing requests, plaintext over public network
|
||||||
|
- File: path traversal, unrestricted upload type/size, zip slip
|
||||||
|
- Deserialization: pickle, yaml.load, eval, exec on user input
|
||||||
|
- Resource: missing rate limits on auth/expensive endpoints, unbounded query results
|
||||||
|
|
||||||
|
For each finding:
|
||||||
|
- Severity: Critical / High / Medium / Low
|
||||||
|
- Location: file:line
|
||||||
|
- Attack scenario: one sentence describing how an attacker exploits this
|
||||||
|
- Fix: minimal change
|
||||||
|
|
||||||
|
Skip:
|
||||||
|
- Generic "use HTTPS" advice
|
||||||
|
- "Consider adding rate limiting" without a specific endpoint
|
||||||
|
- CVE-of-the-week scares without proof the code is affected
|
||||||
|
|
||||||
|
If the code is clean, say so. Do not invent findings.`,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
id: 'prompt-builder',
|
||||||
|
name: 'Prompt Builder',
|
||||||
|
description: 'Builds prompts for OpenCode, Claude Code, or BooCode dispatch.',
|
||||||
|
temperature: 0.4,
|
||||||
|
tools: [...DEFAULT_TOOLS],
|
||||||
|
model: null,
|
||||||
|
source: 'builtin',
|
||||||
|
system_prompt: `You write prompts that another coding agent will execute. Your output is the prompt, not the work.
|
||||||
|
|
||||||
|
Process:
|
||||||
|
1. Ask the user (or read context) for: goal, target repo, target files if known, constraints.
|
||||||
|
2. list_dir and view_file the target area. Confirm files exist and are roughly the shape you think.
|
||||||
|
3. Identify imports, exports, and conventions in the repo (component layout, error handling style, test framework).
|
||||||
|
4. Write the prompt.
|
||||||
|
|
||||||
|
Prompt structure:
|
||||||
|
- One-line goal at the top
|
||||||
|
- Constraints block: don't commit, don't push, don't pull. Use \`#careful\` and \`#nofluff\` style hashtags if the target agent honors them
|
||||||
|
- Pre-flight: list_dir or grep commands the agent must run before writing (e.g. "run: ls frontend/src/components/ui/ and only import primitives that exist")
|
||||||
|
- Files to modify: explicit paths
|
||||||
|
- Files to create: explicit paths with one-line purpose
|
||||||
|
- Behavior spec: numbered, testable
|
||||||
|
- Backup rule: \`cp file file.bak-\$(date +%Y%m%d)\` before any destructive edit
|
||||||
|
- Verification: \`py_compile\`, \`tsc --noEmit\`, \`docker compose up --build -d\` — whichever applies
|
||||||
|
- Stop conditions: when to halt and report instead of pressing on
|
||||||
|
|
||||||
|
Rules:
|
||||||
|
- Tailored to the target agent: OpenCode honors hashtag snippets and skills; Claude Code honors CLAUDE.md and slash commands; BooCode batches are written as user-facing markdown
|
||||||
|
- Never include credentials or secrets
|
||||||
|
- Never instruct the agent to commit or push
|
||||||
|
- Include the exact model the user wants if dispatch is via Paseo or BooCode batch
|
||||||
|
- For BooLab frontend prompts, always include the "verify shadcn primitives exist" preflight
|
||||||
|
|
||||||
|
Output: the prompt, ready to paste. Nothing else.`,
|
||||||
|
},
|
||||||
|
];
|
||||||
|
|
||||||
|
// ---- AGENTS.md parser ------------------------------------------------------
|
||||||
|
|
||||||
|
interface ParsedFrontmatter {
|
||||||
|
temperature?: number;
|
||||||
|
tools?: string[];
|
||||||
|
description?: string;
|
||||||
|
model?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
function stripQuotes(s: string): string {
|
||||||
|
if (
|
||||||
|
s.length >= 2 &&
|
||||||
|
(s[0] === '"' || s[0] === "'") &&
|
||||||
|
s[0] === s[s.length - 1]
|
||||||
|
) {
|
||||||
|
return s.slice(1, -1);
|
||||||
|
}
|
||||||
|
return s;
|
||||||
|
}
|
||||||
|
|
||||||
|
function parseFrontmatter(yaml: string): { data: ParsedFrontmatter; errors: string[] } {
|
||||||
|
const data: ParsedFrontmatter = {};
|
||||||
|
const errors: string[] = [];
|
||||||
|
const lines = yaml.split('\n');
|
||||||
|
let arrayKey: 'tools' | null = null;
|
||||||
|
|
||||||
|
for (const rawLine of lines) {
|
||||||
|
const line = rawLine.trim();
|
||||||
|
if (line.length === 0) continue;
|
||||||
|
|
||||||
|
// Block-list continuation: "- value" under a key that was set to empty
|
||||||
|
if (arrayKey && line.startsWith('- ')) {
|
||||||
|
data[arrayKey]!.push(line.slice(2).trim());
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
arrayKey = null;
|
||||||
|
|
||||||
|
const colonIdx = line.indexOf(':');
|
||||||
|
if (colonIdx < 0) continue;
|
||||||
|
const key = line.slice(0, colonIdx).trim();
|
||||||
|
const valueRaw = line.slice(colonIdx + 1).trim();
|
||||||
|
|
||||||
|
if (key === 'temperature') {
|
||||||
|
const n = Number(valueRaw);
|
||||||
|
if (Number.isFinite(n)) data.temperature = n;
|
||||||
|
else errors.push(`temperature must be a number (got "${valueRaw}")`);
|
||||||
|
} else if (key === 'tools') {
|
||||||
|
if (valueRaw === '') {
|
||||||
|
data.tools = [];
|
||||||
|
arrayKey = 'tools';
|
||||||
|
} else if (valueRaw.startsWith('[') && valueRaw.endsWith(']')) {
|
||||||
|
const inner = valueRaw.slice(1, -1);
|
||||||
|
data.tools = inner
|
||||||
|
.split(',')
|
||||||
|
.map((s) => stripQuotes(s.trim()))
|
||||||
|
.filter((s) => s.length > 0);
|
||||||
|
} else {
|
||||||
|
// Loose form: "tools: a, b, c"
|
||||||
|
data.tools = valueRaw
|
||||||
|
.split(',')
|
||||||
|
.map((s) => stripQuotes(s.trim()))
|
||||||
|
.filter((s) => s.length > 0);
|
||||||
|
}
|
||||||
|
} else if (key === 'description') {
|
||||||
|
data.description = stripQuotes(valueRaw);
|
||||||
|
} else if (key === 'model') {
|
||||||
|
data.model = stripQuotes(valueRaw);
|
||||||
|
}
|
||||||
|
// Unknown keys silently ignored — forward-compat.
|
||||||
|
}
|
||||||
|
|
||||||
|
return { data, errors };
|
||||||
|
}
|
||||||
|
|
||||||
|
interface ParseResult {
|
||||||
|
agents: Agent[];
|
||||||
|
error: string | null;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function parseAgentsMd(content: string): ParseResult {
|
||||||
|
const errors: string[] = [];
|
||||||
|
const agents: Agent[] = [];
|
||||||
|
|
||||||
|
// Split into per-agent sections by lines that exactly match "## <name>".
|
||||||
|
// Lines starting with "### " (level-3 headings) are not section boundaries.
|
||||||
|
const sections: { name: string; body: string }[] = [];
|
||||||
|
let currentName: string | null = null;
|
||||||
|
let currentLines: string[] = [];
|
||||||
|
|
||||||
|
for (const line of content.split('\n')) {
|
||||||
|
const h2 = /^##\s+(.+?)\s*$/.exec(line);
|
||||||
|
const h3 = line.startsWith('### ');
|
||||||
|
if (h2 && !h3) {
|
||||||
|
if (currentName !== null) {
|
||||||
|
sections.push({ name: currentName, body: currentLines.join('\n') });
|
||||||
|
}
|
||||||
|
currentName = h2[1]!.trim();
|
||||||
|
currentLines = [];
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (currentName !== null) {
|
||||||
|
currentLines.push(line);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (currentName !== null) {
|
||||||
|
sections.push({ name: currentName, body: currentLines.join('\n') });
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const section of sections) {
|
||||||
|
const lines = section.body.split('\n');
|
||||||
|
// Opening "---" fence must be the first non-empty line (blank lines allowed).
|
||||||
|
let openIdx = -1;
|
||||||
|
for (let i = 0; i < lines.length; i++) {
|
||||||
|
const t = lines[i]!.trim();
|
||||||
|
if (t === '') continue;
|
||||||
|
if (t === '---') {
|
||||||
|
openIdx = i;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
if (openIdx < 0) {
|
||||||
|
errors.push(`agent "${section.name}": missing opening --- fence after heading`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
let closeIdx = -1;
|
||||||
|
for (let i = openIdx + 1; i < lines.length; i++) {
|
||||||
|
if (lines[i]!.trim() === '---') {
|
||||||
|
closeIdx = i;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (closeIdx < 0) {
|
||||||
|
errors.push(`agent "${section.name}": missing closing --- fence`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
const yamlText = lines.slice(openIdx + 1, closeIdx).join('\n');
|
||||||
|
const systemPrompt = lines.slice(closeIdx + 1).join('\n').trim();
|
||||||
|
|
||||||
|
const { data: fm, errors: fmErrors } = parseFrontmatter(yamlText);
|
||||||
|
if (fmErrors.length > 0) {
|
||||||
|
errors.push(`agent "${section.name}": ${fmErrors.join('; ')}`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
const filteredTools = Array.isArray(fm.tools)
|
||||||
|
? fm.tools.filter((t): t is string =>
|
||||||
|
(ALL_TOOL_NAMES as readonly string[]).includes(t)
|
||||||
|
)
|
||||||
|
: DEFAULT_TOOLS;
|
||||||
|
|
||||||
|
agents.push({
|
||||||
|
id: slugify(section.name),
|
||||||
|
name: section.name,
|
||||||
|
description: fm.description ?? '',
|
||||||
|
system_prompt: systemPrompt,
|
||||||
|
temperature: typeof fm.temperature === 'number' ? fm.temperature : DEFAULT_TEMPERATURE,
|
||||||
|
tools: filteredTools,
|
||||||
|
model: typeof fm.model === 'string' && fm.model.length > 0 ? fm.model : null,
|
||||||
|
source: 'file',
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
return { agents, error: errors.length > 0 ? errors.join('; ') : null };
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---- mtime-keyed cache + public API ----------------------------------------
|
||||||
|
|
||||||
|
interface CacheEntry {
|
||||||
|
mtimeMs: number;
|
||||||
|
result: AgentsResponse;
|
||||||
|
}
|
||||||
|
|
||||||
|
const cache = new Map<string, CacheEntry>();
|
||||||
|
|
||||||
|
// Test/admin: force re-parse on next call for a project (or all projects).
|
||||||
|
export function invalidateAgentsCache(projectPath?: string): void {
|
||||||
|
if (projectPath === undefined) {
|
||||||
|
cache.clear();
|
||||||
|
} else {
|
||||||
|
cache.delete(projectPath);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function getAgentsForProject(projectPath: string): Promise<AgentsResponse> {
|
||||||
|
const agentsPath = join(projectPath, 'AGENTS.md');
|
||||||
|
let mtimeMs: number;
|
||||||
|
try {
|
||||||
|
const s = await fs.stat(agentsPath);
|
||||||
|
mtimeMs = s.mtimeMs;
|
||||||
|
} catch {
|
||||||
|
// No AGENTS.md → builtins, no parse error
|
||||||
|
cache.delete(projectPath);
|
||||||
|
return { agents: BUILTIN_AGENTS, parse_error: null };
|
||||||
|
}
|
||||||
|
|
||||||
|
const cached = cache.get(projectPath);
|
||||||
|
if (cached && cached.mtimeMs === mtimeMs) {
|
||||||
|
return cached.result;
|
||||||
|
}
|
||||||
|
|
||||||
|
let content: string;
|
||||||
|
try {
|
||||||
|
content = await fs.readFile(agentsPath, 'utf8');
|
||||||
|
} catch {
|
||||||
|
cache.delete(projectPath);
|
||||||
|
return { agents: BUILTIN_AGENTS, parse_error: null };
|
||||||
|
}
|
||||||
|
|
||||||
|
const parsed = parseAgentsMd(content);
|
||||||
|
let result: AgentsResponse;
|
||||||
|
if (parsed.error) {
|
||||||
|
// Parse error: surface in API, fall back to builtins
|
||||||
|
result = { agents: BUILTIN_AGENTS, parse_error: parsed.error };
|
||||||
|
} else if (parsed.agents.length === 0) {
|
||||||
|
// Empty / no headings → builtins
|
||||||
|
result = { agents: BUILTIN_AGENTS, parse_error: null };
|
||||||
|
} else {
|
||||||
|
// At least one valid agent → file-defined agents win, builtins hidden
|
||||||
|
result = { agents: parsed.agents, parse_error: null };
|
||||||
|
}
|
||||||
|
|
||||||
|
cache.set(projectPath, { mtimeMs, result });
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function getAgentById(
|
||||||
|
projectPath: string,
|
||||||
|
agentId: string
|
||||||
|
): Promise<Agent | null> {
|
||||||
|
const { agents } = await getAgentsForProject(projectPath);
|
||||||
|
return agents.find((a) => a.id === agentId) ?? null;
|
||||||
|
}
|
||||||
|
|
||||||
|
export { BUILTIN_AGENTS };
|
||||||
@@ -144,4 +144,23 @@ export async function maybeAutoNameChat(
|
|||||||
updated_at: updated[0]!.updated_at,
|
updated_at: updated[0]!.updated_at,
|
||||||
});
|
});
|
||||||
ctx.log.info({ chatId, name }, 'chat auto-named');
|
ctx.log.info({ chatId, name }, 'chat auto-named');
|
||||||
|
|
||||||
|
// Propagate to the parent session if it's still on its default name.
|
||||||
|
// The WHERE guard makes the check atomic — if the user has already
|
||||||
|
// renamed (or a prior chat already propagated), this UPDATE matches
|
||||||
|
// zero rows and we do nothing. First chat wins; manual renames win.
|
||||||
|
const renamedSession = await ctx.sql<{ id: string; name: string }[]>`
|
||||||
|
UPDATE sessions
|
||||||
|
SET name = ${name}
|
||||||
|
WHERE id = ${sessionId} AND name = 'New session'
|
||||||
|
RETURNING id, name
|
||||||
|
`;
|
||||||
|
if (renamedSession.length > 0) {
|
||||||
|
ctx.publishUser({
|
||||||
|
type: 'session_renamed',
|
||||||
|
session_id: sessionId,
|
||||||
|
name,
|
||||||
|
});
|
||||||
|
ctx.log.info({ sessionId, name }, 'session auto-named from chat');
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,10 +1,11 @@
|
|||||||
import type { FastifyBaseLogger } from 'fastify';
|
import type { FastifyBaseLogger } from 'fastify';
|
||||||
import type { Sql } from '../db.js';
|
import type { Sql } from '../db.js';
|
||||||
import type { Config } from '../config.js';
|
import type { Config } from '../config.js';
|
||||||
import type { Message, Project, Session, ToolCall, UserStreamFrame } from '../types/api.js';
|
import type { Agent, Message, Project, Session, ToolCall, UserStreamFrame } from '../types/api.js';
|
||||||
import { ALL_TOOLS, TOOLS_BY_NAME, toolJsonSchemas } from './tools.js';
|
import { ALL_TOOLS, TOOLS_BY_NAME, toolJsonSchemas, type ToolJsonSchema } from './tools.js';
|
||||||
import { PathScopeError, resolveProjectRoot } from './path_guard.js';
|
import { PathScopeError, resolveProjectRoot } from './path_guard.js';
|
||||||
import { maybeAutoNameChat } from './auto_name.js';
|
import { maybeAutoNameChat } from './auto_name.js';
|
||||||
|
import { getAgentById } from './agents.js';
|
||||||
|
|
||||||
const BASE_SYSTEM_PROMPT = (projectPath: string) =>
|
const BASE_SYSTEM_PROMPT = (projectPath: string) =>
|
||||||
`You are BooCode Chat, a code investigation assistant. The user is working on a project located at ${projectPath}. Use the file-read tools (view_file, list_dir, grep, find_files) to investigate code when needed. Be concise. Cite file paths and line numbers when discussing code. Do not hallucinate file contents — read the file first. Tool results may be truncated; if so, narrow your query rather than guessing.`;
|
`You are BooCode Chat, a code investigation assistant. The user is working on a project located at ${projectPath}. Use the file-read tools (view_file, list_dir, grep, find_files) to investigate code when needed. Be concise. Cite file paths and line numbers when discussing code. Do not hallucinate file contents — read the file first. Tool results may be truncated; if so, narrow your query rather than guessing.`;
|
||||||
@@ -91,16 +92,32 @@ export interface InferenceContext {
|
|||||||
publishUser: (frame: UserStreamFrame) => void;
|
publishUser: (frame: UserStreamFrame) => void;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Resolution order: base prompt < agent.system_prompt < session.system_prompt.
|
||||||
|
// Agent prompts layer on top of the base; session prompt is the most specific
|
||||||
|
// override and stacks last so callers can append per-session instructions.
|
||||||
|
export function buildSystemPrompt(
|
||||||
|
project: Project,
|
||||||
|
session: Session,
|
||||||
|
agent: Agent | null
|
||||||
|
): string {
|
||||||
|
let out = BASE_SYSTEM_PROMPT(project.path);
|
||||||
|
if (agent && agent.system_prompt.trim().length > 0) {
|
||||||
|
out += '\n\n' + agent.system_prompt.trim();
|
||||||
|
}
|
||||||
|
if (session.system_prompt && session.system_prompt.trim().length > 0) {
|
||||||
|
out += '\n\n' + session.system_prompt.trim();
|
||||||
|
}
|
||||||
|
return out;
|
||||||
|
}
|
||||||
|
|
||||||
export function buildMessagesPayload(
|
export function buildMessagesPayload(
|
||||||
session: Session,
|
session: Session,
|
||||||
project: Project,
|
project: Project,
|
||||||
history: Message[]
|
history: Message[],
|
||||||
|
agent: Agent | null = null
|
||||||
): OpenAiMessage[] {
|
): OpenAiMessage[] {
|
||||||
const out: OpenAiMessage[] = [];
|
const out: OpenAiMessage[] = [];
|
||||||
let systemPrompt = BASE_SYSTEM_PROMPT(project.path);
|
const systemPrompt = buildSystemPrompt(project, session, agent);
|
||||||
if (session.system_prompt && session.system_prompt.trim().length > 0) {
|
|
||||||
systemPrompt += '\n\n' + session.system_prompt.trim();
|
|
||||||
}
|
|
||||||
out.push({ role: 'system', content: systemPrompt });
|
out.push({ role: 'system', content: systemPrompt });
|
||||||
|
|
||||||
// Find the latest compact marker — only send messages from that point onwards
|
// Find the latest compact marker — only send messages from that point onwards
|
||||||
@@ -161,7 +178,7 @@ async function loadContext(
|
|||||||
chatId: string
|
chatId: string
|
||||||
): Promise<{ session: Session; project: Project; history: Message[] } | null> {
|
): Promise<{ session: Session; project: Project; history: Message[] } | null> {
|
||||||
const sessionRows = await sql<Session[]>`
|
const sessionRows = await sql<Session[]>`
|
||||||
SELECT id, project_id, name, model, system_prompt, status, created_at, updated_at
|
SELECT id, project_id, name, model, system_prompt, status, created_at, updated_at, agent_id
|
||||||
FROM sessions WHERE id = ${sessionId}
|
FROM sessions WHERE id = ${sessionId}
|
||||||
`;
|
`;
|
||||||
if (sessionRows.length === 0) return null;
|
if (sessionRows.length === 0) return null;
|
||||||
@@ -217,11 +234,18 @@ interface StreamResult {
|
|||||||
nCtx: number | null;
|
nCtx: number | null;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
interface StreamOptions {
|
||||||
|
// null = omit tools entirely (compact phase); [] = caller stripped all tools
|
||||||
|
// (rare; we still omit from the request body to avoid OpenAI 400).
|
||||||
|
tools: ToolJsonSchema[] | null;
|
||||||
|
temperature?: number;
|
||||||
|
}
|
||||||
|
|
||||||
async function streamCompletion(
|
async function streamCompletion(
|
||||||
ctx: InferenceContext,
|
ctx: InferenceContext,
|
||||||
model: string,
|
model: string,
|
||||||
messages: OpenAiMessage[],
|
messages: OpenAiMessage[],
|
||||||
includeTools: boolean,
|
opts: StreamOptions,
|
||||||
onDelta: (content: string) => void,
|
onDelta: (content: string) => void,
|
||||||
signal?: AbortSignal
|
signal?: AbortSignal
|
||||||
): Promise<StreamResult> {
|
): Promise<StreamResult> {
|
||||||
@@ -231,10 +255,13 @@ async function streamCompletion(
|
|||||||
stream: true,
|
stream: true,
|
||||||
stream_options: { include_usage: true },
|
stream_options: { include_usage: true },
|
||||||
};
|
};
|
||||||
if (includeTools) {
|
if (opts.tools && opts.tools.length > 0) {
|
||||||
body['tools'] = toolJsonSchemas();
|
body['tools'] = opts.tools;
|
||||||
body['tool_choice'] = 'auto';
|
body['tool_choice'] = 'auto';
|
||||||
}
|
}
|
||||||
|
if (typeof opts.temperature === 'number') {
|
||||||
|
body['temperature'] = opts.temperature;
|
||||||
|
}
|
||||||
|
|
||||||
const res = await fetch(`${ctx.config.LLAMA_SWAP_URL}/v1/chat/completions`, {
|
const res = await fetch(`${ctx.config.LLAMA_SWAP_URL}/v1/chat/completions`, {
|
||||||
method: 'POST',
|
method: 'POST',
|
||||||
@@ -366,7 +393,8 @@ async function executeStreamPhase(
|
|||||||
args: TurnArgs,
|
args: TurnArgs,
|
||||||
session: Session,
|
session: Session,
|
||||||
messages: OpenAiMessage[],
|
messages: OpenAiMessage[],
|
||||||
state: StreamPhaseState
|
state: StreamPhaseState,
|
||||||
|
agent: Agent | null
|
||||||
): Promise<StreamResult> {
|
): Promise<StreamResult> {
|
||||||
const { sessionId, chatId, assistantMessageId, signal } = args;
|
const { sessionId, chatId, assistantMessageId, signal } = args;
|
||||||
|
|
||||||
@@ -407,12 +435,20 @@ async function executeStreamPhase(
|
|||||||
}, DB_FLUSH_INTERVAL_MS);
|
}, DB_FLUSH_INTERVAL_MS);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Tool whitelist: if an agent is set, filter the global tool list to only the
|
||||||
|
// tool names it allows. Unknown names in agent.tools are dropped silently
|
||||||
|
// (handled here by intersection). When no agent: send all tools.
|
||||||
|
const effectiveTools: ToolJsonSchema[] = agent
|
||||||
|
? toolJsonSchemas().filter((t) => agent.tools.includes(t.function.name))
|
||||||
|
: toolJsonSchemas();
|
||||||
|
const effectiveTemperature = agent?.temperature;
|
||||||
|
|
||||||
try {
|
try {
|
||||||
return await streamCompletion(
|
return await streamCompletion(
|
||||||
ctx,
|
ctx,
|
||||||
session.model,
|
session.model,
|
||||||
messages,
|
messages,
|
||||||
true,
|
{ tools: effectiveTools, temperature: effectiveTemperature },
|
||||||
(delta) => {
|
(delta) => {
|
||||||
state.accumulated += delta;
|
state.accumulated += delta;
|
||||||
ctx.publish(sessionId, {
|
ctx.publish(sessionId, {
|
||||||
@@ -657,12 +693,18 @@ async function runAssistantTurn(
|
|||||||
}
|
}
|
||||||
const { session, project, history } = loaded;
|
const { session, project, history } = loaded;
|
||||||
const projectRoot = await resolveProjectRoot(project.path);
|
const projectRoot = await resolveProjectRoot(project.path);
|
||||||
const messages = buildMessagesPayload(session, project, history);
|
// Agent resolution is per-turn so PATCH agent_id mid-conversation takes
|
||||||
|
// effect on the next message. Unknown agent_id returns null silently —
|
||||||
|
// session falls back to base prompt + all tools + default temperature.
|
||||||
|
const agent = session.agent_id
|
||||||
|
? await getAgentById(project.path, session.agent_id)
|
||||||
|
: null;
|
||||||
|
const messages = buildMessagesPayload(session, project, history, agent);
|
||||||
|
|
||||||
const state: StreamPhaseState = { accumulated: '', startedAt: null };
|
const state: StreamPhaseState = { accumulated: '', startedAt: null };
|
||||||
let result: StreamResult;
|
let result: StreamResult;
|
||||||
try {
|
try {
|
||||||
result = await executeStreamPhase(ctx, args, session, messages, state);
|
result = await executeStreamPhase(ctx, args, session, messages, state, agent);
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
await handleAbortOrError(ctx, args, state.accumulated, err);
|
await handleAbortOrError(ctx, args, state.accumulated, err);
|
||||||
return;
|
return;
|
||||||
@@ -720,7 +762,7 @@ async function runCompact(
|
|||||||
ctx,
|
ctx,
|
||||||
session.model,
|
session.model,
|
||||||
messagesForSummary,
|
messagesForSummary,
|
||||||
false,
|
{ tools: null },
|
||||||
(delta) => {
|
(delta) => {
|
||||||
content += delta;
|
content += delta;
|
||||||
ctx.publish(sessionId, {
|
ctx.publish(sessionId, {
|
||||||
|
|||||||
@@ -28,6 +28,27 @@ export interface Session {
|
|||||||
status: SessionStatus;
|
status: SessionStatus;
|
||||||
created_at: string;
|
created_at: string;
|
||||||
updated_at: string;
|
updated_at: string;
|
||||||
|
agent_id: string | null;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Agent sources: 'builtin' = baked-in default (services/agents.ts),
|
||||||
|
// 'file' = parsed from project's AGENTS.md.
|
||||||
|
export type AgentSource = 'builtin' | 'file';
|
||||||
|
|
||||||
|
export interface Agent {
|
||||||
|
id: string; // slug of name; stable handle stored in sessions.agent_id
|
||||||
|
name: string;
|
||||||
|
description: string;
|
||||||
|
system_prompt: string;
|
||||||
|
temperature: number;
|
||||||
|
tools: string[]; // whitelist of tool names; empty = no tools allowed
|
||||||
|
model: string | null; // null means "session.model wins"
|
||||||
|
source: AgentSource;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface AgentsResponse {
|
||||||
|
agents: Agent[];
|
||||||
|
parse_error: string | null; // present (non-null) when AGENTS.md exists but failed to parse
|
||||||
}
|
}
|
||||||
|
|
||||||
// KEEP IN SYNC: apps/server/src/schema.sql chats_status_chk
|
// KEEP IN SYNC: apps/server/src/schema.sql chats_status_chk
|
||||||
|
|||||||
@@ -8,6 +8,7 @@ import type {
|
|||||||
SidebarResponse,
|
SidebarResponse,
|
||||||
ListDirResult,
|
ListDirResult,
|
||||||
ViewFileResult,
|
ViewFileResult,
|
||||||
|
AgentsResponse,
|
||||||
} from './types';
|
} from './types';
|
||||||
|
|
||||||
export class ApiError extends Error {
|
export class ApiError extends Error {
|
||||||
@@ -93,7 +94,7 @@ export const api = {
|
|||||||
request<Session[]>(`/api/projects/${projectId}/sessions${status ? `?status=${status}` : ''}`),
|
request<Session[]>(`/api/projects/${projectId}/sessions${status ? `?status=${status}` : ''}`),
|
||||||
create: (
|
create: (
|
||||||
projectId: string,
|
projectId: string,
|
||||||
body: { name?: string; model?: string; system_prompt?: string }
|
body: { name?: string; model?: string; system_prompt?: string; agent_id?: string | null }
|
||||||
) =>
|
) =>
|
||||||
request<Session>(`/api/projects/${projectId}/sessions`, {
|
request<Session>(`/api/projects/${projectId}/sessions`, {
|
||||||
method: 'POST',
|
method: 'POST',
|
||||||
@@ -102,7 +103,7 @@ export const api = {
|
|||||||
get: (id: string) => request<Session>(`/api/sessions/${id}`),
|
get: (id: string) => request<Session>(`/api/sessions/${id}`),
|
||||||
update: (
|
update: (
|
||||||
id: string,
|
id: string,
|
||||||
body: Partial<Pick<Session, 'name' | 'model' | 'system_prompt'>>
|
body: Partial<Pick<Session, 'name' | 'model' | 'system_prompt' | 'agent_id'>>
|
||||||
) =>
|
) =>
|
||||||
request<Session>(`/api/sessions/${id}`, {
|
request<Session>(`/api/sessions/${id}`, {
|
||||||
method: 'PATCH',
|
method: 'PATCH',
|
||||||
@@ -179,6 +180,11 @@ export const api = {
|
|||||||
|
|
||||||
models: () => request<ModelInfo[]>('/api/models'),
|
models: () => request<ModelInfo[]>('/api/models'),
|
||||||
|
|
||||||
|
agents: {
|
||||||
|
list: (projectId: string) =>
|
||||||
|
request<AgentsResponse>(`/api/projects/${projectId}/agents`),
|
||||||
|
},
|
||||||
|
|
||||||
settings: {
|
settings: {
|
||||||
get: () => request<Record<string, unknown>>('/api/settings'),
|
get: () => request<Record<string, unknown>>('/api/settings'),
|
||||||
patch: (body: Record<string, unknown>) =>
|
patch: (body: Record<string, unknown>) =>
|
||||||
|
|||||||
@@ -27,6 +27,25 @@ export interface Session {
|
|||||||
status: SessionStatus;
|
status: SessionStatus;
|
||||||
created_at: string;
|
created_at: string;
|
||||||
updated_at: string;
|
updated_at: string;
|
||||||
|
agent_id: string | null;
|
||||||
|
}
|
||||||
|
|
||||||
|
export type AgentSource = 'builtin' | 'file';
|
||||||
|
|
||||||
|
export interface Agent {
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
description: string;
|
||||||
|
system_prompt: string;
|
||||||
|
temperature: number;
|
||||||
|
tools: string[];
|
||||||
|
model: string | null;
|
||||||
|
source: AgentSource;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface AgentsResponse {
|
||||||
|
agents: Agent[];
|
||||||
|
parse_error: string | null;
|
||||||
}
|
}
|
||||||
|
|
||||||
export const CHAT_STATUSES = ['open', 'archived'] as const;
|
export const CHAT_STATUSES = ['open', 'archived'] as const;
|
||||||
|
|||||||
108
apps/web/src/components/AgentPicker.tsx
Normal file
108
apps/web/src/components/AgentPicker.tsx
Normal file
@@ -0,0 +1,108 @@
|
|||||||
|
import { useEffect, useState } from 'react';
|
||||||
|
import { Check, ChevronDown } from 'lucide-react';
|
||||||
|
import { toast } from 'sonner';
|
||||||
|
import { api } from '@/api/client';
|
||||||
|
import type { Agent } from '@/api/types';
|
||||||
|
import {
|
||||||
|
DropdownMenu,
|
||||||
|
DropdownMenuContent,
|
||||||
|
DropdownMenuItem,
|
||||||
|
DropdownMenuSeparator,
|
||||||
|
DropdownMenuTrigger,
|
||||||
|
} from '@/components/ui/dropdown-menu';
|
||||||
|
|
||||||
|
interface Props {
|
||||||
|
projectId: string;
|
||||||
|
value: string | null;
|
||||||
|
onChange: (agentId: string | null) => void | Promise<void>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function AgentPicker({ projectId, value, onChange }: Props) {
|
||||||
|
const [agents, setAgents] = useState<Agent[] | null>(null);
|
||||||
|
const [error, setError] = useState<string | null>(null);
|
||||||
|
const [open, setOpen] = useState(false);
|
||||||
|
|
||||||
|
// Load on mount (and on projectId change) so the trigger shows the agent
|
||||||
|
// name immediately, not the raw id. AGENTS.md parse errors surface as a
|
||||||
|
// toast once per load.
|
||||||
|
useEffect(() => {
|
||||||
|
let cancelled = false;
|
||||||
|
setAgents(null);
|
||||||
|
setError(null);
|
||||||
|
api.agents
|
||||||
|
.list(projectId)
|
||||||
|
.then((res) => {
|
||||||
|
if (cancelled) return;
|
||||||
|
setAgents(res.agents);
|
||||||
|
if (res.parse_error) {
|
||||||
|
toast.error(`AGENTS.md parse error: ${res.parse_error}`);
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.catch((err) => {
|
||||||
|
if (cancelled) return;
|
||||||
|
setError(err instanceof Error ? err.message : 'failed to load agents');
|
||||||
|
});
|
||||||
|
return () => {
|
||||||
|
cancelled = true;
|
||||||
|
};
|
||||||
|
}, [projectId]);
|
||||||
|
|
||||||
|
const selectedAgent = agents?.find((a) => a.id === value) ?? null;
|
||||||
|
const triggerLabel = value === null
|
||||||
|
? 'No agent'
|
||||||
|
: selectedAgent?.name ?? value;
|
||||||
|
|
||||||
|
return (
|
||||||
|
<DropdownMenu open={open} onOpenChange={setOpen}>
|
||||||
|
<DropdownMenuTrigger asChild>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
className="text-xs text-muted-foreground hover:text-foreground flex items-center gap-1 px-1.5 py-0.5 rounded hover:bg-muted/60"
|
||||||
|
title={selectedAgent?.description ?? undefined}
|
||||||
|
>
|
||||||
|
<span className="truncate max-w-[160px]">{triggerLabel}</span>
|
||||||
|
<ChevronDown className="size-3 opacity-70" />
|
||||||
|
</button>
|
||||||
|
</DropdownMenuTrigger>
|
||||||
|
<DropdownMenuContent align="start" className="max-h-80 overflow-y-auto w-72">
|
||||||
|
{error && (
|
||||||
|
<div className="px-2 py-1.5 text-xs text-destructive">{error}</div>
|
||||||
|
)}
|
||||||
|
{agents === null && !error && (
|
||||||
|
<div className="px-2 py-1.5 text-xs text-muted-foreground">Loading…</div>
|
||||||
|
)}
|
||||||
|
{agents !== null && (
|
||||||
|
<>
|
||||||
|
<DropdownMenuItem
|
||||||
|
onSelect={() => void onChange(null)}
|
||||||
|
className="text-xs"
|
||||||
|
>
|
||||||
|
<Check className={`size-3 ${value === null ? 'opacity-100' : 'opacity-0'}`} />
|
||||||
|
<span className="font-medium">No agent</span>
|
||||||
|
</DropdownMenuItem>
|
||||||
|
{agents.length > 0 && <DropdownMenuSeparator />}
|
||||||
|
{agents.map((a) => (
|
||||||
|
<DropdownMenuItem
|
||||||
|
key={a.id}
|
||||||
|
onSelect={() => void onChange(a.id)}
|
||||||
|
className="text-xs flex-col items-start gap-0.5"
|
||||||
|
>
|
||||||
|
<div className="flex items-center gap-1.5">
|
||||||
|
<Check
|
||||||
|
className={`size-3 ${a.id === value ? 'opacity-100' : 'opacity-0'}`}
|
||||||
|
/>
|
||||||
|
<span className="font-medium">{a.name}</span>
|
||||||
|
</div>
|
||||||
|
{a.description && (
|
||||||
|
<span className="text-muted-foreground pl-[18px] truncate w-full">
|
||||||
|
{a.description}
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
</DropdownMenuItem>
|
||||||
|
))}
|
||||||
|
</>
|
||||||
|
)}
|
||||||
|
</DropdownMenuContent>
|
||||||
|
</DropdownMenu>
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -1,29 +1,47 @@
|
|||||||
import { useCallback, useEffect, useRef, useState, type KeyboardEvent } from 'react';
|
import { useCallback, useEffect, useRef, useState, type DragEvent, type KeyboardEvent } from 'react';
|
||||||
import { Send } from 'lucide-react';
|
import { Send } from 'lucide-react';
|
||||||
import { toast } from 'sonner';
|
import { toast } from 'sonner';
|
||||||
import { Textarea } from '@/components/ui/textarea';
|
import { Textarea } from '@/components/ui/textarea';
|
||||||
import { Button } from '@/components/ui/button';
|
import { Button } from '@/components/ui/button';
|
||||||
import { flattenToMessage, inferLanguage, type Attachment } from '@/lib/attachments';
|
import {
|
||||||
|
flattenToMessage,
|
||||||
|
inferLanguage,
|
||||||
|
looksBinary,
|
||||||
|
MAX_FILE_SIZE_BYTES,
|
||||||
|
PASTE_INLINE_MAX_LINES,
|
||||||
|
type Attachment,
|
||||||
|
} from '@/lib/attachments';
|
||||||
import { AttachmentChip } from '@/components/AttachmentChip';
|
import { AttachmentChip } from '@/components/AttachmentChip';
|
||||||
import { AttachmentPreviewModal } from '@/components/AttachmentPreviewModal';
|
import { AttachmentPreviewModal } from '@/components/AttachmentPreviewModal';
|
||||||
import { FileMentionPopover } from '@/components/FileMentionPopover';
|
import { FileMentionPopover } from '@/components/FileMentionPopover';
|
||||||
|
import { DropOverlay } from '@/components/DropOverlay';
|
||||||
|
import { AgentPicker } from '@/components/AgentPicker';
|
||||||
import { api } from '@/api/client';
|
import { api } from '@/api/client';
|
||||||
import { sessionEvents } from '@/hooks/sessionEvents';
|
import { sessionEvents } from '@/hooks/sessionEvents';
|
||||||
import { useViewport } from '@/hooks/useViewport';
|
import { useViewport } from '@/hooks/useViewport';
|
||||||
|
|
||||||
|
const MAX_ATTACHMENTS = 10;
|
||||||
|
|
||||||
interface Props {
|
interface Props {
|
||||||
disabled?: boolean;
|
disabled?: boolean;
|
||||||
projectId: string;
|
projectId: string;
|
||||||
|
// Batch 9: optional so callers that pre-date the agent picker still compile.
|
||||||
|
// When omitted, the toolbar row is hidden entirely.
|
||||||
|
agentId?: string | null;
|
||||||
|
onAgentChange?: (agentId: string | null) => void | Promise<void>;
|
||||||
onSend: (content: string) => void | Promise<void>;
|
onSend: (content: string) => void | Promise<void>;
|
||||||
onForceSend?: (content: string) => void | Promise<void>;
|
onForceSend?: (content: string) => void | Promise<void>;
|
||||||
}
|
}
|
||||||
|
|
||||||
export function ChatInput({ disabled, projectId, onSend, onForceSend }: Props) {
|
export function ChatInput({ disabled, projectId, agentId, onAgentChange, onSend, onForceSend }: Props) {
|
||||||
const { isMobile } = useViewport();
|
const { isMobile } = useViewport();
|
||||||
const [value, setValue] = useState('');
|
const [value, setValue] = useState('');
|
||||||
const [busy, setBusy] = useState(false);
|
const [busy, setBusy] = useState(false);
|
||||||
const [attachments, setAttachments] = useState<Attachment[]>([]);
|
const [attachments, setAttachments] = useState<Attachment[]>([]);
|
||||||
const [previewAttachment, setPreviewAttachment] = useState<Attachment | null>(null);
|
const [previewAttachment, setPreviewAttachment] = useState<Attachment | null>(null);
|
||||||
|
const [isDraggingOver, setIsDraggingOver] = useState(false);
|
||||||
|
const dropRootRef = useRef<HTMLDivElement | null>(null);
|
||||||
|
const pasteCounterRef = useRef(0);
|
||||||
const [mentionState, setMentionState] = useState<{
|
const [mentionState, setMentionState] = useState<{
|
||||||
open: boolean;
|
open: boolean;
|
||||||
query: string;
|
query: string;
|
||||||
@@ -35,8 +53,8 @@ export function ChatInput({ disabled, projectId, onSend, onForceSend }: Props) {
|
|||||||
|
|
||||||
function addAttachment(a: Attachment) {
|
function addAttachment(a: Attachment) {
|
||||||
setAttachments(prev => {
|
setAttachments(prev => {
|
||||||
if (prev.length >= 10) {
|
if (prev.length >= MAX_ATTACHMENTS) {
|
||||||
toast.error('Max 10 attachments per message');
|
toast.error(`Max ${MAX_ATTACHMENTS} attachments per message`);
|
||||||
return prev;
|
return prev;
|
||||||
}
|
}
|
||||||
return [...prev, a];
|
return [...prev, a];
|
||||||
@@ -185,6 +203,162 @@ export function ChatInput({ disabled, projectId, onSend, onForceSend }: Props) {
|
|||||||
|
|
||||||
const closeMention = useCallback(() => setMentionState(null), []);
|
const closeMention = useCallback(() => setMentionState(null), []);
|
||||||
|
|
||||||
|
// ---- Drag & drop (F1 + F3 + F4) ----------------------------------------
|
||||||
|
// The drop zone is the outer ChatInput container (ref'd as dropRootRef).
|
||||||
|
// onDragLeave only clears the highlight when the cursor leaves the
|
||||||
|
// container, not when it crosses into a child element.
|
||||||
|
|
||||||
|
async function processDroppedFile(file: File) {
|
||||||
|
// Size gate
|
||||||
|
if (file.size > MAX_FILE_SIZE_BYTES) {
|
||||||
|
const mb = (file.size / (1024 * 1024)).toFixed(1);
|
||||||
|
toast.error(`File ${file.name} is too large (${mb} MB). Limit is 5 MB.`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Read once as ArrayBuffer so we can do byte-level binary detection
|
||||||
|
// before deciding whether to decode as text.
|
||||||
|
let buf: ArrayBuffer;
|
||||||
|
try {
|
||||||
|
buf = await file.arrayBuffer();
|
||||||
|
} catch (err) {
|
||||||
|
toast.error(`Failed to read ${file.name}: ${err instanceof Error ? err.message : String(err)}`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (looksBinary(buf)) {
|
||||||
|
toast.error(`${file.name} appears to be binary.`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const text = new TextDecoder('utf-8', { fatal: false }).decode(buf);
|
||||||
|
addAttachment({
|
||||||
|
id: crypto.randomUUID(),
|
||||||
|
kind: 'file',
|
||||||
|
filename: file.name,
|
||||||
|
language: inferLanguage(file.name),
|
||||||
|
content: text,
|
||||||
|
source: 'drop',
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
function isFolderItem(item: DataTransferItem | undefined): boolean {
|
||||||
|
if (!item) return false;
|
||||||
|
// webkitGetAsEntry is non-standard but supported in Chromium + Safari.
|
||||||
|
// If unavailable, we conservatively treat the entry as a file.
|
||||||
|
const entry =
|
||||||
|
typeof item.webkitGetAsEntry === 'function' ? item.webkitGetAsEntry() : null;
|
||||||
|
if (entry && entry.isDirectory) return true;
|
||||||
|
// Heuristic fallback: folders dragged from Finder have type === '' and
|
||||||
|
// a 0-byte File. The empty-type alone isn't reliable for files (some
|
||||||
|
// plaintext drops also lack a type), so we only flag when the entry
|
||||||
|
// explicitly says directory.
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
async function handleDroppedItems(dt: DataTransfer) {
|
||||||
|
// Snapshot items first because reading files inside the loop can
|
||||||
|
// detach the DataTransfer between awaits.
|
||||||
|
const itemsArray: { file: File | null; isFolder: boolean }[] = [];
|
||||||
|
if (dt.items && dt.items.length > 0) {
|
||||||
|
for (let i = 0; i < dt.items.length; i++) {
|
||||||
|
const it = dt.items[i];
|
||||||
|
if (!it || it.kind !== 'file') continue;
|
||||||
|
const folder = isFolderItem(it);
|
||||||
|
const file = folder ? null : it.getAsFile();
|
||||||
|
itemsArray.push({ file, isFolder: folder });
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
for (let i = 0; i < dt.files.length; i++) {
|
||||||
|
const f = dt.files[i];
|
||||||
|
if (f) itemsArray.push({ file: f, isFolder: false });
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
let remainingSlots = MAX_ATTACHMENTS - attachments.length;
|
||||||
|
let folderRejected = false;
|
||||||
|
for (const { file, isFolder } of itemsArray) {
|
||||||
|
if (isFolder) {
|
||||||
|
if (!folderRejected) {
|
||||||
|
toast.error('Folders are not supported');
|
||||||
|
folderRejected = true;
|
||||||
|
}
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (!file) continue;
|
||||||
|
if (remainingSlots <= 0) {
|
||||||
|
toast.error(`Attachment limit reached (${MAX_ATTACHMENTS}).`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
await processDroppedFile(file);
|
||||||
|
remainingSlots -= 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
function onDragEnter(e: DragEvent<HTMLDivElement>) {
|
||||||
|
if (disabled || busy) return;
|
||||||
|
e.preventDefault();
|
||||||
|
setIsDraggingOver(true);
|
||||||
|
}
|
||||||
|
|
||||||
|
function onDragOver(e: DragEvent<HTMLDivElement>) {
|
||||||
|
if (disabled || busy) return;
|
||||||
|
e.preventDefault();
|
||||||
|
e.dataTransfer.dropEffect = 'copy';
|
||||||
|
}
|
||||||
|
|
||||||
|
function onDragLeave(e: DragEvent<HTMLDivElement>) {
|
||||||
|
// Only clear when the cursor actually leaves the root container.
|
||||||
|
// relatedTarget is the element being entered; if it's inside the root,
|
||||||
|
// ignore — we're just crossing into a child.
|
||||||
|
const root = dropRootRef.current;
|
||||||
|
if (!root) return;
|
||||||
|
const related = e.relatedTarget as Node | null;
|
||||||
|
if (related && root.contains(related)) return;
|
||||||
|
setIsDraggingOver(false);
|
||||||
|
}
|
||||||
|
|
||||||
|
function onDrop(e: DragEvent<HTMLDivElement>) {
|
||||||
|
e.preventDefault();
|
||||||
|
setIsDraggingOver(false);
|
||||||
|
if (disabled || busy) return;
|
||||||
|
void handleDroppedItems(e.dataTransfer);
|
||||||
|
}
|
||||||
|
// ---- end Drag & drop -----------------------------------------------------
|
||||||
|
|
||||||
|
// ---- Paste-as-attachment (F2) -------------------------------------------
|
||||||
|
// Pasting >PASTE_INLINE_MAX_LINES lines of text becomes a chip rather than
|
||||||
|
// inline content. Image pastes are rejected with a toast. If both text and
|
||||||
|
// image are present (e.g. screenshot tool that sets both), prefer text.
|
||||||
|
|
||||||
|
function onPaste(e: React.ClipboardEvent<HTMLTextAreaElement>) {
|
||||||
|
const cd = e.clipboardData;
|
||||||
|
if (!cd) return;
|
||||||
|
const text = cd.getData('text/plain');
|
||||||
|
const hasImage = Array.from(cd.items ?? []).some((it) =>
|
||||||
|
it.type.startsWith('image/'),
|
||||||
|
);
|
||||||
|
if (text) {
|
||||||
|
const lineCount = text.split('\n').length;
|
||||||
|
if (lineCount > PASTE_INLINE_MAX_LINES) {
|
||||||
|
e.preventDefault();
|
||||||
|
pasteCounterRef.current += 1;
|
||||||
|
addAttachment({
|
||||||
|
id: crypto.randomUUID(),
|
||||||
|
kind: 'paste',
|
||||||
|
filename: `pasted-${pasteCounterRef.current}.txt`,
|
||||||
|
language: 'plaintext',
|
||||||
|
content: text,
|
||||||
|
source: 'paste',
|
||||||
|
});
|
||||||
|
}
|
||||||
|
// <= threshold: let default paste insert inline.
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (hasImage) {
|
||||||
|
e.preventDefault();
|
||||||
|
toast.error('Image paste is not supported. Drop a file or paste text.');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// ---- end Paste-as-attachment --------------------------------------------
|
||||||
|
|
||||||
function onKeyDown(e: KeyboardEvent<HTMLTextAreaElement>) {
|
function onKeyDown(e: KeyboardEvent<HTMLTextAreaElement>) {
|
||||||
if (mentionState?.open) return;
|
if (mentionState?.open) return;
|
||||||
// IME safety: never act on Enter while an IME composition is in flight
|
// IME safety: never act on Enter while an IME composition is in flight
|
||||||
@@ -228,7 +402,16 @@ export function ChatInput({ disabled, projectId, onSend, onForceSend }: Props) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="border-t" style={{ paddingBottom: 'env(safe-area-inset-bottom)' }}>
|
<div
|
||||||
|
ref={dropRootRef}
|
||||||
|
className="border-t relative"
|
||||||
|
style={{ paddingBottom: 'env(safe-area-inset-bottom)' }}
|
||||||
|
onDragEnter={onDragEnter}
|
||||||
|
onDragOver={onDragOver}
|
||||||
|
onDragLeave={onDragLeave}
|
||||||
|
onDrop={onDrop}
|
||||||
|
>
|
||||||
|
<DropOverlay visible={isDraggingOver} />
|
||||||
<div className="max-w-[1000px] mx-auto w-full">
|
<div className="max-w-[1000px] mx-auto w-full">
|
||||||
{attachments.length > 0 && (
|
{attachments.length > 0 && (
|
||||||
<div className="flex flex-wrap gap-1.5 px-4 pt-3">
|
<div className="flex flex-wrap gap-1.5 px-4 pt-3">
|
||||||
@@ -242,12 +425,25 @@ export function ChatInput({ disabled, projectId, onSend, onForceSend }: Props) {
|
|||||||
))}
|
))}
|
||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
|
{/* Batch 9 toolbar — agent picker. Sits above the input row so it
|
||||||
|
doesn't compete with the send button for vertical alignment.
|
||||||
|
When Batch 7 lands, ModelPicker and the + button join this row. */}
|
||||||
|
{onAgentChange && (
|
||||||
|
<div className="px-4 pt-2 flex items-center gap-1.5">
|
||||||
|
<AgentPicker
|
||||||
|
projectId={projectId}
|
||||||
|
value={agentId ?? null}
|
||||||
|
onChange={onAgentChange}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
<div className="px-4 py-3 flex items-end gap-2">
|
<div className="px-4 py-3 flex items-end gap-2">
|
||||||
<Textarea
|
<Textarea
|
||||||
ref={textareaRef}
|
ref={textareaRef}
|
||||||
value={value}
|
value={value}
|
||||||
onChange={handleChange}
|
onChange={handleChange}
|
||||||
onKeyDown={onKeyDown}
|
onKeyDown={onKeyDown}
|
||||||
|
onPaste={onPaste}
|
||||||
placeholder={
|
placeholder={
|
||||||
isMobile
|
isMobile
|
||||||
? 'Ask about this project. Tap send to submit.'
|
? 'Ask about this project. Tap send to submit.'
|
||||||
|
|||||||
18
apps/web/src/components/DropOverlay.tsx
Normal file
18
apps/web/src/components/DropOverlay.tsx
Normal file
@@ -0,0 +1,18 @@
|
|||||||
|
interface Props {
|
||||||
|
visible: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Visual cue layered over the ChatInput while a drag is in progress.
|
||||||
|
// Pointer-events: none so the underlying drop handler still receives the
|
||||||
|
// drop event. Renders nothing when not visible (cheap and out of layout).
|
||||||
|
export function DropOverlay({ visible }: Props) {
|
||||||
|
if (!visible) return null;
|
||||||
|
return (
|
||||||
|
<div
|
||||||
|
className="absolute inset-0 z-10 pointer-events-none flex items-center justify-center rounded border-2 border-dashed border-primary bg-background/85"
|
||||||
|
aria-hidden="true"
|
||||||
|
>
|
||||||
|
<div className="text-sm font-medium text-primary">Drop to attach</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
@@ -20,9 +20,12 @@ import { cn } from '@/lib/utils';
|
|||||||
interface Props {
|
interface Props {
|
||||||
sessionId: string;
|
sessionId: string;
|
||||||
projectId: string;
|
projectId: string;
|
||||||
|
// Batch 9: threaded down to ChatPane → ChatInput → AgentPicker.
|
||||||
|
agentId?: string | null;
|
||||||
|
onAgentChange?: (agentId: string | null) => void | Promise<void>;
|
||||||
}
|
}
|
||||||
|
|
||||||
export function Workspace({ sessionId, projectId }: Props) {
|
export function Workspace({ sessionId, projectId, agentId, onAgentChange }: Props) {
|
||||||
const {
|
const {
|
||||||
panes,
|
panes,
|
||||||
activePaneIdx,
|
activePaneIdx,
|
||||||
@@ -219,7 +222,14 @@ export function Workspace({ sessionId, projectId }: Props) {
|
|||||||
|
|
||||||
<div className="flex-1 min-h-0 overflow-hidden">
|
<div className="flex-1 min-h-0 overflow-hidden">
|
||||||
{pane.kind === 'chat' && pane.chatId ? (
|
{pane.kind === 'chat' && pane.chatId ? (
|
||||||
<ChatPane sessionId={sessionId} chatId={pane.chatId} projectId={projectId} sessionChats={chats} />
|
<ChatPane
|
||||||
|
sessionId={sessionId}
|
||||||
|
chatId={pane.chatId}
|
||||||
|
projectId={projectId}
|
||||||
|
agentId={agentId}
|
||||||
|
onAgentChange={onAgentChange}
|
||||||
|
sessionChats={chats}
|
||||||
|
/>
|
||||||
) : (
|
) : (
|
||||||
<SessionLandingPage
|
<SessionLandingPage
|
||||||
sessionId={sessionId}
|
sessionId={sessionId}
|
||||||
|
|||||||
@@ -18,10 +18,13 @@ interface Props {
|
|||||||
sessionId: string;
|
sessionId: string;
|
||||||
chatId: string;
|
chatId: string;
|
||||||
projectId: string;
|
projectId: string;
|
||||||
|
// Batch 9: optional, threaded down to ChatInput's agent picker.
|
||||||
|
agentId?: string | null;
|
||||||
|
onAgentChange?: (agentId: string | null) => void | Promise<void>;
|
||||||
sessionChats?: import('@/api/types').Chat[];
|
sessionChats?: import('@/api/types').Chat[];
|
||||||
}
|
}
|
||||||
|
|
||||||
export function ChatPane({ sessionId, chatId, projectId, sessionChats }: Props) {
|
export function ChatPane({ sessionId, chatId, projectId, agentId, onAgentChange, sessionChats }: Props) {
|
||||||
const stream = useSessionStream(sessionId);
|
const stream = useSessionStream(sessionId);
|
||||||
const lastErrorRef = useRef<string | null>(null);
|
const lastErrorRef = useRef<string | null>(null);
|
||||||
const [queue, setQueue] = useState<string[]>([]);
|
const [queue, setQueue] = useState<string[]>([]);
|
||||||
@@ -167,7 +170,14 @@ export function ChatPane({ sessionId, chatId, projectId, sessionChats }: Props)
|
|||||||
|
|
||||||
<div className="relative">
|
<div className="relative">
|
||||||
<ChatContextPopover stats={contextStats} />
|
<ChatContextPopover stats={contextStats} />
|
||||||
<ChatInput disabled={false} projectId={projectId} onSend={handleSend} onForceSend={streaming ? handleForceSend : undefined} />
|
<ChatInput
|
||||||
|
disabled={false}
|
||||||
|
projectId={projectId}
|
||||||
|
agentId={agentId}
|
||||||
|
onAgentChange={onAgentChange}
|
||||||
|
onSend={handleSend}
|
||||||
|
onForceSend={streaming ? handleForceSend : undefined}
|
||||||
|
/>
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
|
|||||||
@@ -8,6 +8,34 @@ export type Attachment = {
|
|||||||
source: '@' | 'line-select' | 'drop' | 'paste';
|
source: '@' | 'line-select' | 'drop' | 'paste';
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// v1.7: caps shared between drag-drop and paste-as-attachment so both paths
|
||||||
|
// reject the same way. Match the existing 10-attachment cap in
|
||||||
|
// ChatInput.addAttachment.
|
||||||
|
export const MAX_FILE_SIZE_BYTES = 5 * 1024 * 1024; // 5 MB
|
||||||
|
export const PASTE_INLINE_MAX_LINES = 8;
|
||||||
|
|
||||||
|
// First-8KB null-byte scan. Returns true if the content looks binary.
|
||||||
|
// Accepts a string (post-decode), an ArrayBuffer (pre-decode), or a Uint8Array.
|
||||||
|
// For binary files like PNG, scanning bytes is more reliable than scanning
|
||||||
|
// post-UTF-8-decode strings because invalid sequences may be replaced rather
|
||||||
|
// than preserved.
|
||||||
|
export function looksBinary(content: string | ArrayBuffer | Uint8Array): boolean {
|
||||||
|
const SCAN_BYTES = 8192;
|
||||||
|
if (typeof content === 'string') {
|
||||||
|
const max = Math.min(content.length, SCAN_BYTES);
|
||||||
|
for (let i = 0; i < max; i++) {
|
||||||
|
if (content.charCodeAt(i) === 0) return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
const bytes = content instanceof Uint8Array ? content : new Uint8Array(content);
|
||||||
|
const max = Math.min(bytes.length, SCAN_BYTES);
|
||||||
|
for (let i = 0; i < max; i++) {
|
||||||
|
if (bytes[i] === 0) return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
export const LANG_MAP: Record<string, string> = {
|
export const LANG_MAP: Record<string, string> = {
|
||||||
ts: 'typescript', tsx: 'tsx', js: 'javascript', jsx: 'jsx',
|
ts: 'typescript', tsx: 'tsx', js: 'javascript', jsx: 'jsx',
|
||||||
mjs: 'javascript', cjs: 'javascript',
|
mjs: 'javascript', cjs: 'javascript',
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import { useEffect, useState } from 'react';
|
import { useEffect, useState } from 'react';
|
||||||
import { ChevronDown, ChevronRight, Folder, RotateCcw } from 'lucide-react';
|
import { ChevronDown, ChevronRight, Folder, FolderTree, Menu, RotateCcw } from 'lucide-react';
|
||||||
import { toast } from 'sonner';
|
import { toast } from 'sonner';
|
||||||
import { Button } from '@/components/ui/button';
|
import { Button } from '@/components/ui/button';
|
||||||
import { AddProjectModal } from '@/components/AddProjectModal';
|
import { AddProjectModal } from '@/components/AddProjectModal';
|
||||||
@@ -8,6 +8,9 @@ import { api } from '@/api/client';
|
|||||||
import type { Project } from '@/api/types';
|
import type { Project } from '@/api/types';
|
||||||
import { sessionEvents } from '@/hooks/sessionEvents';
|
import { sessionEvents } from '@/hooks/sessionEvents';
|
||||||
import { useSidebar } from '@/hooks/useSidebar';
|
import { useSidebar } from '@/hooks/useSidebar';
|
||||||
|
import { useSidebarDrawer } from '@/hooks/useSidebarDrawer';
|
||||||
|
import { useRightRailDrawer } from '@/hooks/useRightRailDrawer';
|
||||||
|
import { useViewport } from '@/hooks/useViewport';
|
||||||
|
|
||||||
export function Home() {
|
export function Home() {
|
||||||
const { data } = useSidebar();
|
const { data } = useSidebar();
|
||||||
@@ -15,6 +18,9 @@ export function Home() {
|
|||||||
const [createOpen, setCreateOpen] = useState(false);
|
const [createOpen, setCreateOpen] = useState(false);
|
||||||
const [archived, setArchived] = useState<Project[] | null>(null);
|
const [archived, setArchived] = useState<Project[] | null>(null);
|
||||||
const [showArchived, setShowArchived] = useState(false);
|
const [showArchived, setShowArchived] = useState(false);
|
||||||
|
const { setOpen: setSidebarOpen } = useSidebarDrawer();
|
||||||
|
const { toggle: toggleRightRail } = useRightRailDrawer();
|
||||||
|
const { isMobile } = useViewport();
|
||||||
|
|
||||||
const empty = data ? data.projects.length === 0 : false;
|
const empty = data ? data.projects.length === 0 : false;
|
||||||
|
|
||||||
@@ -70,8 +76,32 @@ export function Home() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="flex-1 flex flex-col items-center px-6 py-12 overflow-y-auto">
|
<div className="flex-1 flex flex-col min-h-0">
|
||||||
<div className="w-full max-w-md space-y-6">
|
{isMobile && (
|
||||||
|
<header
|
||||||
|
className="border-b px-3 sm:px-4 py-2 flex items-center gap-1.5 shrink-0 text-sm"
|
||||||
|
style={{ paddingTop: 'max(0.5rem, env(safe-area-inset-top))' }}
|
||||||
|
>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
onClick={() => setSidebarOpen(true)}
|
||||||
|
className="inline-flex items-center justify-center -ml-1 min-w-[44px] min-h-[44px] rounded text-muted-foreground hover:bg-muted hover:text-foreground shrink-0"
|
||||||
|
aria-label="Open sidebar"
|
||||||
|
>
|
||||||
|
<Menu className="size-5" />
|
||||||
|
</button>
|
||||||
|
<button
|
||||||
|
type="button"
|
||||||
|
onClick={toggleRightRail}
|
||||||
|
className="inline-flex items-center justify-center -mr-1 ml-auto min-w-[44px] min-h-[44px] rounded text-muted-foreground hover:bg-muted hover:text-foreground shrink-0"
|
||||||
|
aria-label="Toggle file browser"
|
||||||
|
>
|
||||||
|
<FolderTree className="size-5" />
|
||||||
|
</button>
|
||||||
|
</header>
|
||||||
|
)}
|
||||||
|
<div className="flex-1 flex flex-col items-center px-6 py-12 overflow-y-auto">
|
||||||
|
<div className="w-full max-w-md space-y-6">
|
||||||
<div className="text-center space-y-3">
|
<div className="text-center space-y-3">
|
||||||
{empty ? (
|
{empty ? (
|
||||||
<>
|
<>
|
||||||
@@ -127,9 +157,10 @@ export function Home() {
|
|||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
)}
|
)}
|
||||||
|
</div>
|
||||||
|
<AddProjectModal open={addOpen} onOpenChange={setAddOpen} onAdded={() => {}} />
|
||||||
|
<CreateProjectModal open={createOpen} onOpenChange={setCreateOpen} />
|
||||||
</div>
|
</div>
|
||||||
<AddProjectModal open={addOpen} onOpenChange={setAddOpen} onAdded={() => {}} />
|
|
||||||
<CreateProjectModal open={createOpen} onOpenChange={setCreateOpen} />
|
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -193,7 +193,15 @@ export function Session() {
|
|||||||
</header>
|
</header>
|
||||||
|
|
||||||
{id && session && (
|
{id && session && (
|
||||||
<Workspace sessionId={id} projectId={session.project_id} />
|
<Workspace
|
||||||
|
sessionId={id}
|
||||||
|
projectId={session.project_id}
|
||||||
|
agentId={session.agent_id}
|
||||||
|
onAgentChange={async (agent_id) => {
|
||||||
|
const updated = await api.sessions.update(session.id, { agent_id });
|
||||||
|
setSession(updated);
|
||||||
|
}}
|
||||||
|
/>
|
||||||
)}
|
)}
|
||||||
</div>
|
</div>
|
||||||
);
|
);
|
||||||
|
|||||||
Reference in New Issue
Block a user