Jason Woltje
d7f04d1148
All checks were successful
ci/woodpecker/push/woodpecker Pipeline was successful
Implement intent classification for natural language queries in the brain module. Features: - Hybrid classification approach: rule-based (fast, <100ms) with optional LLM fallback - 10 intent types: query_tasks, query_events, query_projects, create_task, create_event, update_task, update_event, briefing, search, unknown - Entity extraction: dates, times, priorities, statuses, people - Pattern-based matching with priority system (higher priority = checked first) - Optional LLM classification for ambiguous queries - POST /api/brain/classify endpoint Implementation: - IntentClassificationService with classify(), classifyWithRules(), classifyWithLlm(), extractEntities() - Comprehensive regex patterns for common query types - Entity extraction for dates, times, priorities, statuses, mentions - Type-safe interfaces for IntentType, IntentClassification, ExtractedEntity, IntentPattern - ClassifyIntentDto and IntentClassificationResultDto for API validation - Integrated with existing LlmService (optional dependency) Testing: - 60 comprehensive tests covering all intent types - Edge cases: empty queries, special characters, case sensitivity, multiple whitespace - Entity extraction tests with position tracking - LLM fallback tests with error handling - 100% test coverage - All tests passing (60/60) - TDD approach: tests written first Quality: - No explicit any types - Explicit return types on all functions - No TypeScript errors - Build successful - Follows existing code patterns - Quality Rails compliance: All lint checks pass Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
93 lines
4.1 KiB
TypeScript
93 lines
4.1 KiB
TypeScript
import { Controller, Get, Post, Body, Query, UseGuards } from "@nestjs/common";
|
|
import { BrainService } from "./brain.service";
|
|
import { IntentClassificationService } from "./intent-classification.service";
|
|
import {
|
|
BrainQueryDto,
|
|
BrainContextDto,
|
|
ClassifyIntentDto,
|
|
IntentClassificationResultDto,
|
|
} from "./dto";
|
|
import { AuthGuard } from "../auth/guards/auth.guard";
|
|
import { WorkspaceGuard, PermissionGuard } from "../common/guards";
|
|
import { Workspace, Permission, RequirePermission } from "../common/decorators";
|
|
|
|
/**
|
|
* @description Controller for AI/brain operations on workspace data.
|
|
* Provides endpoints for querying, searching, and getting context across
|
|
* tasks, events, and projects within a workspace.
|
|
*/
|
|
@Controller("brain")
|
|
@UseGuards(AuthGuard, WorkspaceGuard, PermissionGuard)
|
|
export class BrainController {
|
|
constructor(
|
|
private readonly brainService: BrainService,
|
|
private readonly intentClassificationService: IntentClassificationService
|
|
) {}
|
|
|
|
/**
|
|
* @description Query workspace entities with flexible filtering options.
|
|
* Allows filtering tasks, events, and projects by various criteria.
|
|
* @param queryDto - Query parameters including entity types, filters, and search term
|
|
* @param workspaceId - The workspace ID (injected from request context)
|
|
* @returns Filtered tasks, events, and projects with metadata
|
|
* @throws UnauthorizedException if user lacks workspace access
|
|
* @throws ForbiddenException if user lacks required permissions
|
|
*/
|
|
@Post("query")
|
|
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
async query(@Body() queryDto: BrainQueryDto, @Workspace() workspaceId: string) {
|
|
return this.brainService.query(Object.assign({}, queryDto, { workspaceId }));
|
|
}
|
|
|
|
/**
|
|
* @description Get current workspace context for AI operations.
|
|
* Returns a summary of active tasks, overdue items, upcoming events, and projects.
|
|
* @param contextDto - Context options specifying which entities to include
|
|
* @param workspaceId - The workspace ID (injected from request context)
|
|
* @returns Workspace context with summary counts and optional detailed entity lists
|
|
* @throws UnauthorizedException if user lacks workspace access
|
|
* @throws ForbiddenException if user lacks required permissions
|
|
* @throws NotFoundException if workspace does not exist
|
|
*/
|
|
@Get("context")
|
|
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
async getContext(@Query() contextDto: BrainContextDto, @Workspace() workspaceId: string) {
|
|
return this.brainService.getContext(Object.assign({}, contextDto, { workspaceId }));
|
|
}
|
|
|
|
/**
|
|
* @description Search across all workspace entities by text.
|
|
* Performs case-insensitive search on titles, descriptions, and locations.
|
|
* @param searchTerm - Text to search for across all entity types
|
|
* @param limit - Maximum number of results per entity type (max: 100, default: 20)
|
|
* @param workspaceId - The workspace ID (injected from request context)
|
|
* @returns Matching tasks, events, and projects with metadata
|
|
* @throws UnauthorizedException if user lacks workspace access
|
|
* @throws ForbiddenException if user lacks required permissions
|
|
*/
|
|
@Get("search")
|
|
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
async search(
|
|
@Query("q") searchTerm: string,
|
|
@Query("limit") limit: string,
|
|
@Workspace() workspaceId: string
|
|
) {
|
|
const parsedLimit = limit ? Math.min(parseInt(limit, 10) || 20, 100) : 20;
|
|
return this.brainService.search(workspaceId, searchTerm || "", parsedLimit);
|
|
}
|
|
|
|
/**
|
|
* @description Classify a natural language query into a structured intent.
|
|
* Uses hybrid classification: rule-based (fast) with optional LLM fallback.
|
|
* @param dto - Classification request with query and optional useLlm flag
|
|
* @returns Intent classification with confidence, entities, and method used
|
|
* @throws UnauthorizedException if user lacks workspace access
|
|
* @throws ForbiddenException if user lacks required permissions
|
|
*/
|
|
@Post("classify")
|
|
@RequirePermission(Permission.WORKSPACE_ANY)
|
|
async classifyIntent(@Body() dto: ClassifyIntentDto): Promise<IntentClassificationResultDto> {
|
|
return this.intentClassificationService.classify(dto.query, dto.useLlm);
|
|
}
|
|
}
|