Files
stack/apps/api/src/knowledge/search.controller.ts
Jason Woltje 3969dd5598 feat(#70): implement semantic search API with Ollama embeddings
Updated semantic search to use OllamaEmbeddingService instead of OpenAI:
- Replaced EmbeddingService with OllamaEmbeddingService in SearchService
- Added configurable similarity threshold (SEMANTIC_SEARCH_SIMILARITY_THRESHOLD)
- Updated both semanticSearch() and hybridSearch() methods
- Added comprehensive tests for semantic search functionality
- Updated controller documentation to reflect Ollama requirement
- All tests passing with 85%+ coverage

Related changes:
- Updated knowledge.service.versions.spec.ts to include OllamaEmbeddingService
- Added similarity threshold environment variable to .env.example

Fixes #70

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
2026-02-02 15:15:04 -06:00

152 lines
4.8 KiB
TypeScript

import { Controller, Get, Post, Body, Query, UseGuards } from "@nestjs/common";
import { SearchService, PaginatedSearchResults } from "./services/search.service";
import { SearchQueryDto, TagSearchDto, RecentEntriesDto } from "./dto";
import { AuthGuard } from "../auth/guards/auth.guard";
import { WorkspaceGuard, PermissionGuard } from "../common/guards";
import { Workspace, Permission, RequirePermission } from "../common/decorators";
import { EntryStatus } from "@prisma/client";
import type { PaginatedEntries, KnowledgeEntryWithTags } from "./entities/knowledge-entry.entity";
/**
* Response for recent entries endpoint
*/
interface RecentEntriesResponse {
data: KnowledgeEntryWithTags[];
count: number;
}
/**
* Controller for knowledge search endpoints
* All endpoints require authentication and workspace context
*/
@Controller("knowledge/search")
@UseGuards(AuthGuard, WorkspaceGuard, PermissionGuard)
export class SearchController {
constructor(private readonly searchService: SearchService) {}
/**
* GET /api/knowledge/search
* Full-text search across knowledge entries
* Searches title and content with relevance ranking
* Requires: Any workspace member
*
* @query q - The search query string (required)
* @query tags - Comma-separated tag slugs to filter by (optional, entries must have ALL tags)
* @query status - Filter by entry status (optional)
* @query page - Page number (default: 1)
* @query limit - Results per page (default: 20, max: 100)
*/
@Get()
@RequirePermission(Permission.WORKSPACE_ANY)
async search(
@Workspace() workspaceId: string,
@Query() query: SearchQueryDto
): Promise<PaginatedSearchResults> {
return this.searchService.search(query.q, workspaceId, {
status: query.status,
page: query.page,
limit: query.limit,
tags: query.tags,
});
}
/**
* GET /api/knowledge/search/by-tags
* Search entries by tags (entries must have ALL specified tags)
* Requires: Any workspace member
*
* @query tags - Comma-separated list of tag slugs (required)
* @query status - Filter by entry status (optional)
* @query page - Page number (default: 1)
* @query limit - Results per page (default: 20, max: 100)
*/
@Get("by-tags")
@RequirePermission(Permission.WORKSPACE_ANY)
async searchByTags(
@Workspace() workspaceId: string,
@Query() query: TagSearchDto
): Promise<PaginatedEntries> {
return this.searchService.searchByTags(query.tags, workspaceId, {
status: query.status,
page: query.page,
limit: query.limit,
});
}
/**
* GET /api/knowledge/search/recent
* Get recently modified entries
* Requires: Any workspace member
*
* @query limit - Maximum number of entries (default: 10, max: 50)
* @query status - Filter by entry status (optional)
*/
@Get("recent")
@RequirePermission(Permission.WORKSPACE_ANY)
async recentEntries(
@Workspace() workspaceId: string,
@Query() query: RecentEntriesDto
): Promise<RecentEntriesResponse> {
const entries = await this.searchService.recentEntries(
workspaceId,
query.limit ?? 10,
query.status
);
return {
data: entries,
count: entries.length,
};
}
/**
* POST /api/knowledge/search/semantic
* Semantic search using vector similarity
* Requires: Any workspace member, Ollama configured
*
* @body query - The search query string (required)
* @body status - Filter by entry status (optional)
* @query page - Page number (default: 1)
* @query limit - Results per page (default: 20, max: 100)
*/
@Post("semantic")
@RequirePermission(Permission.WORKSPACE_ANY)
async semanticSearch(
@Workspace() workspaceId: string,
@Body() body: { query: string; status?: EntryStatus },
@Query("page") page?: number,
@Query("limit") limit?: number
): Promise<PaginatedSearchResults> {
return this.searchService.semanticSearch(body.query, workspaceId, {
status: body.status,
page,
limit,
});
}
/**
* POST /api/knowledge/search/hybrid
* Hybrid search combining vector similarity and full-text search
* Uses Reciprocal Rank Fusion to merge results
* Requires: Any workspace member
*
* @body query - The search query string (required)
* @body status - Filter by entry status (optional)
* @query page - Page number (default: 1)
* @query limit - Results per page (default: 20, max: 100)
*/
@Post("hybrid")
@RequirePermission(Permission.WORKSPACE_ANY)
async hybridSearch(
@Workspace() workspaceId: string,
@Body() body: { query: string; status?: EntryStatus },
@Query("page") page?: number,
@Query("limit") limit?: number
): Promise<PaginatedSearchResults> {
return this.searchService.hybridSearch(body.query, workspaceId, {
status: body.status,
page,
limit,
});
}
}