opcode-pg-memory 2.2.0

package/dist/src/topic/segment-manager.d.ts

@@ -0,0 +1,165 @@
+import { Pool } from 'pg';
+/**
+ * Represents a topic segment boundary within a session.
+ * Segments partition a session into coherent topic units,
+ * preventing entity contamination across different topics.
+ */
+export interface TopicSegmentInfo {
+    id: string;
+    sessionMapId: string;
+    segmentIndex: number;
+    startMessageExternalId: string;
+    endMessageExternalId?: string;
+    summary?: string;
+    embedding?: number[];
+    closedAt?: Date;
+}
+/**
+ * Configuration for the TopicManager's sliding window
+ * topic boundary detection algorithm.
+ */
+export interface TopicManagerConfig {
+    /** Number of recent events to keep in the sliding window for comparison. Default: 3 */
+    windowSize: number;
+    /** Cosine similarity threshold below which a new topic segment is created. Default: 0.3 */
+    mutationThreshold: number;
+}
+/**
+ * Standard event shape accepted by classifyEvent().
+ * Compatible with tool.execute.after, message.updated, and general events.
+ */
+export type StandardEvent = {
+    type: string;
+    session?: {
+        id: string;
+    };
+    messageId?: string;
+    tool?: {
+        name: string;
+        parameters?: Record<string, any>;
+    };
+    result?: {
+        success: boolean;
+        data?: any;
+        error?: string;
+    };
+    message?: {
+        id: string;
+        content?: string;
+        role?: string;
+    };
+    [key: string]: any;
+};
+/**
+ * Intra-session topic boundary detector using sliding window embedding similarity.
+ *
+ * How it works:
+ * 1. Each incoming event is embedded (via the shared EmbeddingService).
+ * 2. The embedding is added to a fixed-size window buffer.
+ * 3. Average cosine similarity between window items and the current segment's
+ *    centroid embedding is computed.
+ * 4. If similarity drops below `mutationThreshold`, the current segment is closed
+ *    and a new one begins. Otherwise, the segment centroid is updated via
+ *    moving average (0.7 current + 0.3 new).
+ *
+ * This prevents entity contamination across different topics within the same
+ * OpenCode session.
+ */
+export declare class TopicManager {
+    /** Internal UUID from the sessions table (session_map_id). */
+    private sessionMapId;
+    /** External OpenCode session ID (as seen in event.session.id). */
+    private opencodeSessionId;
+    /** All topic segments for this session, ordered by segmentIndex. */
+    private segments;
+    /** The currently open (unclosed) segment, if any. */
+    private currentSegment;
+    /**
+     * Fixed-size sliding window of recent event embeddings.
+     * Used to detect topic drift by comparing against the segment centroid.
+     */
+    private windowBuffer;
+    /** Maximum number of items in the window buffer. */
+    private readonly windowSize;
+    /** Cosine similarity threshold for topic boundary detection. */
+    private readonly mutationThreshold;
+    /** PostgreSQL connection pool. */
+    private pool;
+    constructor(pool: Pool, sessionMapId: string, opencodeSessionId: string, config?: Partial<TopicManagerConfig>);
+    /**
+     * Classify an incoming event: decide whether it belongs to the current
+     * topic segment or triggers a new one.
+     *
+     * This is the primary entry point for the event handler.
+     * Call this for every tool.execute.after and message.updated event.
+     *
+     * @returns The TopicSegmentInfo this event belongs to.
+     */
+    classifyEvent(event: StandardEvent): Promise<TopicSegmentInfo>;
+    /**
+     * Create a new topic segment in the database and track it in memory.
+     */
+    createNewSegment(sessionMapId: string, index: number, startMessageId: string): Promise<TopicSegmentInfo>;
+    /**
+     * Close the current topic segment: generate a summary, compute its
+     * embedding, and persist both to the database.
+     *
+     * After calling this, `currentSegment` is set to null.
+     */
+    closeCurrentSegment(endMessageId: string): Promise<void>;
+    /**
+     * Close all currently open segments (should be called when the
+     * OpenCode session ends or is compacted).
+     *
+     * Updates the session's last_active_at timestamp.
+     */
+    closeAllPendingSegments(): Promise<void>;
+    /**
+     * Generate a 1-sentence summary for a segment by inspecting its
+     * observations and producing a heuristic or LLM-generated description.
+     *
+     * Uses a heuristic by default (tool name aggregation). The LLM path
+     * can be wired in later for higher-quality summaries.
+     */
+    retrieveSegmentSummary(segmentId: string): Promise<string>;
+    /**
+     * Factory method: resolve the internal session_map_id from the
+     * OpenCode session ID, load any existing segments, and return a
+     * fully initialized TopicManager.
+     *
+     * Typical usage:
+     *   const tm = await TopicManager.forSession(pool, event.session.id);
+     */
+    static forSession(pool: Pool, opencodeSessionId: string, config?: Partial<TopicManagerConfig>): Promise<TopicManager>;
+    /** Returns the internal session UUID (session_map_id). */
+    getSessionMapId(): string;
+    /** Returns the OpenCode external session ID. */
+    getOpenCodeSessionId(): string;
+    /** Returns the current segment ID, or null if no segment is open. */
+    getCurrentSegmentId(): string | null;
+    /** Returns the current segment, or null. */
+    getCurrentSegment(): TopicSegmentInfo | null;
+    /** Returns all segments for this session (including closed ones). */
+    getSegments(): TopicSegmentInfo[];
+    /**
+     * Extract a representative text string from an event for embedding.
+     *
+     * - For tool.execute.after events: combines tool name with a snippet
+     *   of the result output.
+     * - For message events: uses the message content directly.
+     * - For all others: falls back to the event type string.
+     */
+    private extractTextFromEvent;
+    /**
+     * Update a segment's centroid embedding using a moving average.
+     *
+     * Formula: 0.7 * current + 0.3 * new, then normalized to unit length.
+     *
+     * This ensures the centroid slowly tracks topic drift within a segment
+     * while remaining anchored to the dominant theme.
+     *
+     * Gracefully handles mismatched vector lengths and null/empty inputs.
+     */
+    private updateSegmentEmbedding;
+}
+//# sourceMappingURL=segment-manager.d.ts.map

package/dist/src/topic/segment-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"segment-manager.d.ts","sourceRoot":"","sources":["../../../src/topic/segment-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,IAAI,CAAC;AAK1B;;;;GAIG;AACH,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,EAAE,MAAM,CAAC;IACrB,sBAAsB,EAAE,MAAM,CAAC;IAC/B,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,CAAC,EAAE,IAAI,CAAC;CACjB;AAED;;;GAGG;AACH,MAAM,WAAW,kBAAkB;IACjC,uFAAuF;IACvF,UAAU,EAAE,MAAM,CAAC;IACnB,2FAA2F;IAC3F,iBAAiB,EAAE,MAAM,CAAC;CAC3B;AAED;;;GAGG;AACH,MAAM,MAAM,aAAa,GAAG;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE;QACL,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KAClC,CAAC;IACF,MAAM,CAAC,EAAE;QACP,OAAO,EAAE,OAAO,CAAC;QACjB,IAAI,CAAC,EAAE,GAAG,CAAC;QACX,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,OAAO,CAAC,EAAE;QACR,EAAE,EAAE,MAAM,CAAC;QACX,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,IAAI,CAAC,EAAE,MAAM,CAAC;KACf,CAAC;IACF,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;CACpB,CAAC;AAyDF;;;;;;;;;;;;;;GAcG;AACH,qBAAa,YAAY;IACvB,8DAA8D;IAC9D,OAAO,CAAC,YAAY,CAAS;IAE7B,kEAAkE;IAClE,OAAO,CAAC,iBAAiB,CAAS;IAElC,oEAAoE;IACpE,OAAO,CAAC,QAAQ,CAAqB;IAErC,qDAAqD;IACrD,OAAO,CAAC,cAAc,CAA0B;IAEhD;;;OAGG;IACH,OAAO,CAAC,YAAY,CAAoD;IAExE,oDAAoD;IACpD,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IAEpC,gEAAgE;IAChE,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAS;IAE3C,kCAAkC;IAClC,OAAO,CAAC,IAAI,CAAO;gBAGjB,IAAI,EAAE,IAAI,EACV,YAAY,EAAE,MAAM,EACpB,iBAAiB,EAAE,MAAM,EACzB,MAAM,CAAC,EAAE,OAAO,CAAC,kBAAkB,CAAC;IActC;;;;;;;;OAQG;IACG,aAAa,CAAC,KAAK,EAAE,aAAa,GAAG,OAAO,CAAC,gBAAgB,CAAC;IAkGpE;;OAEG;IACG,gBAAgB,CACpB,YAAY,EAAE,MAAM,EACpB,KAAK,EAAE,MAAM,EACb,cAAc,EAAE,MAAM,GACrB,OAAO,CAAC,gBAAgB,CAAC;IAoB5B;;;;;OAKG;IACG,mBAAmB,CAAC,YAAY,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAmD9D;;;;;OAKG;IACG,uBAAuB,IAAI,OAAO,CAAC,IAAI,CAAC;IAkB9C;;;;;;OAMG;IACG,sBAAsB,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAmEhE;;;;;;;OAOG;WACU,UAAU,CACrB,IAAI,EAAE,IAAI,EACV,iBAAiB,EAAE,MAAM,EACzB,MAAM,CAAC,EAAE,OAAO,CAAC,kBAAkB,CAAC,GACnC,OAAO,CAAC,YAAY,CAAC;IAkDxB,0DAA0D;IAC1D,eAAe,IAAI,MAAM;IAIzB,gDAAgD;IAChD,oBAAoB,IAAI,MAAM;IAI9B,qEAAqE;IACrE,mBAAmB,IAAI,MAAM,GAAG,IAAI;IAIpC,4CAA4C;IAC5C,iBAAiB,IAAI,gBAAgB,GAAG,IAAI;IAI5C,qEAAqE;IACrE,WAAW,IAAI,gBAAgB,EAAE;IAMjC;;;;;;;OAOG;IACH,OAAO,CAAC,oBAAoB;IAqC5B;;;;;;;;;OASG;IACH,OAAO,CAAC,sBAAsB;CAa/B"}

package/dist/src/types.d.ts

@@ -0,0 +1,405 @@
+export type EntityTier = 'permanent' | 'project' | 'session';
+export type RelationType = 'belongs_to' | 'depends_on' | 'references' | 'implements' | 'uses' | 'custom';
+export interface Entity {
+    id: string;
+    session_id: string;
+    topic_segment_id?: string;
+    name: string;
+    type: string;
+    tier: EntityTier;
+    weight: number;
+    description?: string;
+    embedding?: number[];
+    first_seen_at: Date;
+    last_seen_at: Date;
+    confidence: number;
+    metadata: Record<string, any>;
+}
+export interface Relation {
+    id: string;
+    source_entity_id: string;
+    target_entity_id: string;
+    relation_type: RelationType;
+    confidence: number;
+    description?: string;
+    created_at: Date;
+    session_id: string;
+    topic_segment_id?: string;
+}
+export interface Observation {
+    id: string;
+    session_id: string;
+    topic_segment_id?: string;
+    tool_name?: string;
+    tool_input_summary?: string;
+    tool_output_summary?: string;
+    embedding?: number[];
+    importance: number;
+    created_at: Date;
+    message_id?: string;
+    metadata: Record<string, any>;
+}
+export interface Reflection {
+    id: string;
+    session_id: string;
+    topic_segment_id?: string;
+    summary: string;
+    source_observation_ids: string[];
+    confidence: number;
+    pattern_type?: string;
+    created_at: Date;
+    embedding?: number[];
+    metadata: Record<string, any>;
+}
+export interface SemanticCache {
+    id: string;
+    query_hash: string;
+    query_text: string;
+    query_embedding: number[];
+    response_text: string;
+    hit_count: number;
+    last_hit_at: Date;
+    created_at: Date;
+    similarity_threshold: number;
+    is_pruned: boolean;
+    session_id?: string;
+}
+export interface SessionMap {
+    id: string;
+    opencode_session_id: string;
+    omo_task_id?: string;
+    project_id?: string;
+    model_context_limit: number;
+    created_at: Date;
+    last_active_at: Date;
+    metadata: Record<string, any>;
+}
+export interface TopicSegment {
+    id: string;
+    session_map_id: string;
+    segment_index: number;
+    summary?: string;
+    embedding?: number[];
+    start_message_external_id?: string;
+    end_message_external_id?: string;
+    created_at: Date;
+    closed_at?: Date;
+    observation_count: number;
+    metadata: Record<string, any>;
+}
+export interface TopicSegmentInfo {
+    id: string;
+    sessionMapId: string;
+    index: number;
+    summary?: string;
+    embedding?: number[];
+    startMessageId?: string;
+    endMessageId?: string;
+    observationCount: number;
+}
+export interface CallerContext {
+    type: 'user' | 'omo_agent';
+    current_goal?: string;
+    current_session_id?: string;
+}
+export interface EventSynchronizerConfig {
+    mode: 'hooks' | 'polling' | 'event';
+    pollingIntervalMs?: number;
+}
+export interface TopicConfig {
+    mutationThreshold: number;
+    windowSize: number;
+    enableSummaryGeneration: boolean;
+}
+export interface PluginConfig {
+    database: {
+        host: string;
+        port: number;
+        database: string;
+        user: string;
+        password: string;
+        ssl?: boolean;
+    };
+    embedding: {
+        model: string;
+        dimensions: number;
+        batchSize: number;
+    };
+    cache: {
+        initialThreshold: number;
+        adjustmentStep: number;
+        minThreshold: number;
+        maxThreshold: number;
+        enabled: boolean;
+    };
+    reflection: {
+        observationThreshold: number;
+        segmentThreshold: number;
+        modelSize: '7b' | '14b' | 'full';
+        offPeakHours: number[];
+        enabled: boolean;
+    };
+    tokenBudget: {
+        contextLimitRatio: number;
+        minTokens: number;
+        maxTokens: number;
+    };
+    retrieval: {
+        defaultStrategies: string[];
+        rerankEnabled: boolean;
+        maxResults: number;
+        weights: {
+            semantic: number;
+            recency: number;
+            importance: number;
+        };
+    };
+    topic?: TopicConfig;
+    eventSynchronizer?: EventSynchronizerConfig;
+}
+export interface ToolExecuteBeforeHookInput {
+    tool: string;
+    sessionID: string;
+    callID: string;
+}
+export interface ToolExecuteBeforeHookOutput {
+    args: any;
+}
+export interface ToolExecuteAfterHookInput {
+    tool: string;
+    sessionID: string;
+    callID: string;
+    args: any;
+}
+export interface ToolExecuteAfterHookOutput {
+    title: string;
+    output: string;
+    metadata: any;
+}
+export interface SessionCompactingHookInput {
+    sessionID: string;
+}
+export interface SessionCompactingHookOutput {
+    context: string[];
+    prompt?: string;
+}
+export interface SessionCreatedEvent {
+    sessionID: string;
+}
+export interface SessionCompletedEvent {
+    sessionID: string;
+}
+export type StandardEvent = {
+    type: 'session.created';
+    sessionID: string;
+} | {
+    type: 'tool.execute.after';
+    tool: string;
+    sessionID: string;
+    callID: string;
+    args: any;
+} | {
+    type: 'message.updated';
+    sessionID: string;
+    messageID: string;
+} | {
+    type: 'session.compacting';
+    sessionID: string;
+} | {
+    type: 'session.completed';
+    sessionID: string;
+};
+export interface OpenCodeSession {
+    id: string;
+    projectId?: string;
+    model: {
+        id: string;
+        contextLimit: number;
+        name: string;
+    };
+    messages: OpenCodeMessage[];
+}
+export interface OpenCodeMessage {
+    id: string;
+    parentID?: string;
+    role: 'user' | 'assistant' | 'tool';
+    mode?: string;
+    agent?: string;
+    path?: {
+        cwd: string;
+        root: string;
+    };
+    cost?: number;
+    tokens?: {
+        total: number;
+        input: number;
+        output: number;
+        reasoning: number;
+        cache: {
+            write: number;
+            read: number;
+        };
+    };
+    modelID?: string;
+    providerID?: string;
+    time?: {
+        created: number;
+        completed?: number;
+    };
+    finish?: string;
+    content?: string;
+    summary?: {
+        diffs: any[];
+    };
+    parts?: OpenCodeMessagePart[];
+    message_id?: string;
+    reasoning?: string;
+    tool_calls?: OpenCodeToolCall[];
+}
+export interface OpenCodeMessagePart {
+    type: 'reasoning' | 'text' | 'tool' | 'image' | 'audio';
+    text?: string;
+    time?: {
+        start: number;
+        end?: number;
+    };
+    id: string;
+    sessionID: string;
+    messageID: string;
+    tool?: {
+        name: string;
+        callID: string;
+        state: {
+            status: 'pending' | 'completed' | 'failed';
+            input: Record<string, any>;
+            output?: any;
+            filepath?: string;
+        };
+    };
+}
+export interface OpenCodeToolCall {
+    type: string;
+    tool: string;
+    callID: string;
+    input: Record<string, any>;
+    output?: any;
+    status: string;
+    filepath?: string;
+    exists?: boolean;
+    truncated?: boolean;
+}
+export interface SessionCreatedInput {
+    session: OpenCodeSession;
+}
+export interface SessionCreatedOutput {
+    context?: {
+        memories?: string[];
+        facts?: string[];
+    };
+}
+export interface ToolExecuteBeforeInput {
+    session: {
+        id: string;
+    };
+    tool: {
+        name: string;
+        parameters: Record<string, any>;
+    };
+    messageId: string;
+}
+export interface ToolExecuteBeforeOutput {
+    parameters?: Record<string, any>;
+}
+export interface ToolExecuteAfterInput {
+    session: {
+        id: string;
+    };
+    tool: {
+        name: string;
+        parameters: Record<string, any>;
+    };
+    result: {
+        success: boolean;
+        data?: any;
+        error?: string;
+    };
+    messageId: string;
+    executionTimeMs: number;
+}
+export interface ToolExecuteAfterOutput {
+}
+export interface MessageUpdatedInput {
+    session: {
+        id: string;
+    };
+    message: OpenCodeMessage;
+}
+export interface MessageUpdatedOutput {
+}
+export interface MessagePartUpdatedInput {
+    session: {
+        id: string;
+    };
+    message: {
+        id: string;
+        partIndex: number;
+        content: string;
+        isComplete: boolean;
+    };
+}
+export interface MessagePartUpdatedOutput {
+}
+export interface SessionCompactingInput {
+    session: {
+        id: string;
+    };
+    messagesToCompact: string[];
+    compactionStrategy: 'prune' | 'summarize' | 'archive';
+}
+export interface SessionCompactingOutput {
+    preserveMessageIds?: string[];
+}
+export interface SessionCompletedInput {
+    session: {
+        id: string;
+        projectId?: string;
+        messageCount: number;
+        durationMs: number;
+    };
+    summary?: string;
+}
+export interface SessionCompletedOutput {
+}
+export interface RetrievedFact {
+    id?: string;
+    type: 'entity' | 'observation' | 'reflection' | 'relation' | 'message';
+    content: string;
+    tier?: EntityTier;
+    tokens: number;
+    relevanceScore: number;
+    metadata: Record<string, any>;
+}
+export interface CacheResult {
+    hit: boolean;
+    response?: string;
+    similarity?: number;
+}
+export interface HindsightReflectInput {
+    session_id?: string;
+    omo_task_id?: string;
+    topic_segment_id?: string;
+    trigger_type?: 'threshold' | 'scheduled' | 'manual';
+    observation_threshold?: number;
+    model_size?: '7b' | '14b' | 'full';
+    aggregate?: boolean;
+}
+export interface HindsightReflectOutput {
+    generated_reflections: Reflection[];
+    token_usage: {
+        input: number;
+        output: number;
+        total: number;
+    };
+    duration_ms: number;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/src/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/types.ts"],"names":[],"mappings":"AAQA,MAAM,MAAM,UAAU,GAAG,WAAW,GAAG,SAAS,GAAG,SAAS,CAAC;AAC7D,MAAM,MAAM,YAAY,GAAG,YAAY,GAAG,YAAY,GAAG,YAAY,GAAG,YAAY,GAAG,MAAM,GAAG,QAAQ,CAAC;AAMzG,MAAM,WAAW,MAAM;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,UAAU,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,aAAa,EAAE,IAAI,CAAC;IACpB,YAAY,EAAE,IAAI,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC/B;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,gBAAgB,EAAE,MAAM,CAAC;IACzB,gBAAgB,EAAE,MAAM,CAAC;IACzB,aAAa,EAAE,YAAY,CAAC;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,EAAE,IAAI,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC3B;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,IAAI,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC/B;AAED,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,OAAO,EAAE,MAAM,CAAC;IAChB,sBAAsB,EAAE,MAAM,EAAE,CAAC;IACjC,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,IAAI,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC/B;AAED,MAAM,WAAW,aAAa;IAC5B,EAAE,EAAE,MAAM,CAAC;IACX,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,EAAE,MAAM,EAAE,CAAC;IAC1B,aAAa,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,WAAW,EAAE,IAAI,CAAC;IAClB,UAAU,EAAE,IAAI,CAAC;IACjB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,SAAS,EAAE,OAAO,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAMD,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,mBAAmB,EAAE,MAAM,CAAC;IAC5B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,mBAAmB,EAAE,MAAM,CAAC;IAC5B,UAAU,EAAE,IAAI,CAAC;IACjB,cAAc,EAAE,IAAI,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC/B;AAED,MAAM,WAAW,YAAY;IAC3B,EAAE,EAAE,MAAM,CAAC;IACX,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,yBAAyB,CAAC,EAAE,MAAM,CAAC;IACnC,uBAAuB,CAAC,EAAE,MAAM,CAAC;IACjC,UAAU,EAAE,IAAI,CAAC;IACjB,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC/B;AAGD,MAAM,WAAW,gBAAgB;IAC/B,EAAE,EAAE,MAAM,CAAC;IACX,YAAY,EAAE,MAAM,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,gBAAgB,EAAE,MAAM,CAAC;CAC1B;AAMD,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC7B;AAMD,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,OAAO,GAAG,SAAS,GAAG,OAAO,CAAC;IACpC,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,MAAM,WAAW,WAAW;IAC1B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,UAAU,EAAE,MAAM,CAAC;IACnB,uBAAuB,EAAE,OAAO,CAAC;CAClC;AAED,MAAM,WAAW,YAAY;IAC3B,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,MAAM,CAAC;QACjB,GAAG,CAAC,EAAE,OAAO,CAAC;KACf,CAAC;IACF,SAAS,EAAE;QACT,KAAK,EAAE,MAAM,CAAC;QACd,UAAU,EAAE,MAAM,CAAC;QACnB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,KAAK,EAAE;QACL,gBAAgB,EAAE,MAAM,CAAC;QACzB,cAAc,EAAE,MAAM,CAAC;QACvB,YAAY,EAAE,MAAM,CAAC;QACrB,YAAY,EAAE,MAAM,CAAC;QACrB,OAAO,EAAE,OAAO,CAAC;KAClB,CAAC;IACF,UAAU,EAAE;QACV,oBAAoB,EAAE,MAAM,CAAC;QAC7B,gBAAgB,EAAE,MAAM,CAAC;QACzB,SAAS,EAAE,IAAI,GAAG,KAAK,GAAG,MAAM,CAAC;QACjC,YAAY,EAAE,MAAM,EAAE,CAAC;QACvB,OAAO,EAAE,OAAO,CAAC;KAClB,CAAC;IACF,WAAW,EAAE;QACX,iBAAiB,EAAE,MAAM,CAAC;QAC1B,SAAS,EAAE,MAAM,CAAC;QAClB,SAAS,EAAE,MAAM,CAAC;KACnB,CAAC;IACF,SAAS,EAAE;QACT,iBAAiB,EAAE,MAAM,EAAE,CAAC;QAC5B,aAAa,EAAE,OAAO,CAAC;QACvB,UAAU,EAAE,MAAM,CAAC;QACnB,OAAO,EAAE;YACP,QAAQ,EAAE,MAAM,CAAC;YACjB,OAAO,EAAE,MAAM,CAAC;YAChB,UAAU,EAAE,MAAM,CAAC;SACpB,CAAC;KACH,CAAC;IACF,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,iBAAiB,CAAC,EAAE,uBAAuB,CAAC;CAC7C;AAWD,MAAM,WAAW,0BAA0B;IACzC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,2BAA2B;IAC1C,IAAI,EAAE,GAAG,CAAC;CACX;AAED,MAAM,WAAW,yBAAyB;IACxC,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,GAAG,CAAC;CACX;AAED,MAAM,WAAW,0BAA0B;IACzC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,GAAG,CAAC;CACf;AAED,MAAM,WAAW,0BAA0B;IACzC,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,2BAA2B;IAC1C,OAAO,EAAE,MAAM,EAAE,CAAC;IAClB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,mBAAmB;IAClC,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,qBAAqB;IACpC,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,MAAM,aAAa,GACrB;IAAE,IAAI,EAAE,iBAAiB,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GAC9C;IAAE,IAAI,EAAE,oBAAoB,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,GAAG,CAAA;CAAE,GAC1F;IAAE,IAAI,EAAE,iBAAiB,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GACjE;IAAE,IAAI,EAAE,oBAAoB,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GACjD;IAAE,IAAI,EAAE,mBAAmB,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,CAAC;AAUrD,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,EAAE;QACL,EAAE,EAAE,MAAM,CAAC;QACX,YAAY,EAAE,MAAM,CAAC;QACrB,IAAI,EAAE,MAAM,CAAC;KACd,CAAC;IACF,QAAQ,EAAE,eAAe,EAAE,CAAC;CAC7B;AAED,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;IACpC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,IAAI,CAAC,EAAE;QAAE,GAAG,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IACrC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,MAAM,CAAC,EAAE;QACP,KAAK,EAAE,MAAM,CAAC;QACd,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,EAAE,MAAM,CAAC;QAClB,KAAK,EAAE;YAAE,KAAK,EAAE,MAAM,CAAC;YAAC,IAAI,EAAE,MAAM,CAAA;SAAE,CAAC;KACxC,CAAC;IACF,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,IAAI,CAAC,EAAE;QAAE,OAAO,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAC/C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE;QAAE,KAAK,EAAE,GAAG,EAAE,CAAA;KAAE,CAAC;IAC3B,KAAK,CAAC,EAAE,mBAAmB,EAAE,CAAC;IAC9B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,gBAAgB,EAAE,CAAC;CACjC;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,EAAE,WAAW,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,OAAO,CAAC;IACxD,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,GAAG,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IACvC,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE;QACL,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,EAAE;YACL,MAAM,EAAE,SAAS,GAAG,WAAW,GAAG,QAAQ,CAAC;YAC3C,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAC3B,MAAM,CAAC,EAAE,GAAG,CAAC;YACb,QAAQ,CAAC,EAAE,MAAM,CAAC;SACnB,CAAC;KACH,CAAC;CACH;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC3B,MAAM,CAAC,EAAE,GAAG,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB;AAED,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE,eAAe,CAAC;CAC1B;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,CAAC,EAAE;QAAE,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,EAAE,CAAA;KAAE,CAAC;CACrD;AAED,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,CAAC;IACxB,IAAI,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,CAAC;IACxD,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,uBAAuB;IACtC,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAClC;AAED,MAAM,WAAW,qBAAqB;IACpC,OAAO,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,CAAC;IACxB,IAAI,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAA;KAAE,CAAC;IACxD,MAAM,EAAE;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,IAAI,CAAC,EAAE,GAAG,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IACzD,SAAS,EAAE,MAAM,CAAC;IAClB,eAAe,EAAE,MAAM,CAAC;CACzB;AAED,MAAM,WAAW,sBAAsB;CAEtC;AAED,MAAM,WAAW,mBAAmB;IAClC,OAAO,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,CAAC;IACxB,OAAO,EAAE,eAAe,CAAC;CAC1B;AAED,MAAM,WAAW,oBAAoB;CAEpC;AAED,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,CAAC;IACxB,OAAO,EAAE;QACP,EAAE,EAAE,MAAM,CAAC;QACX,SAAS,EAAE,MAAM,CAAC;QAClB,OAAO,EAAE,MAAM,CAAC;QAChB,UAAU,EAAE,OAAO,CAAC;KACrB,CAAC;CACH;AAED,MAAM,WAAW,wBAAwB;CAExC;AAED,MAAM,WAAW,sBAAsB;IACrC,OAAO,EAAE;QAAE,EAAE,EAAE,MAAM,CAAA;KAAE,CAAC;IACxB,iBAAiB,EAAE,MAAM,EAAE,CAAC;IAC5B,kBAAkB,EAAE,OAAO,GAAG,WAAW,GAAG,SAAS,CAAC;CACvD;AAED,MAAM,WAAW,uBAAuB;IACtC,kBAAkB,CAAC,EAAE,MAAM,EAAE,CAAC;CAC/B;AAED,MAAM,WAAW,qBAAqB;IACpC,OAAO,EAAE;QACP,EAAE,EAAE,MAAM,CAAC;QACX,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,YAAY,EAAE,MAAM,CAAC;QACrB,UAAU,EAAE,MAAM,CAAC;KACpB,CAAC;IACF,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,sBAAsB;CAEtC;AAMD,MAAM,WAAW,aAAa;IAC5B,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,IAAI,EAAE,QAAQ,GAAG,aAAa,GAAG,YAAY,GAAG,UAAU,GAAG,SAAS,CAAC;IACvE,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,UAAU,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,cAAc,EAAE,MAAM,CAAC;IACvB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAC/B;AAED,MAAM,WAAW,WAAW;IAC1B,GAAG,EAAE,OAAO,CAAC;IACb,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAMD,MAAM,WAAW,qBAAqB;IACpC,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,YAAY,CAAC,EAAE,WAAW,GAAG,WAAW,GAAG,QAAQ,CAAC;IACpD,qBAAqB,CAAC,EAAE,MAAM,CAAC;IAC/B,UAAU,CAAC,EAAE,IAAI,GAAG,KAAK,GAAG,MAAM,CAAC;IACnC,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB;AAED,MAAM,WAAW,sBAAsB;IACrC,qBAAqB,EAAE,UAAU,EAAE,CAAC;IACpC,WAAW,EAAE;QACX,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;IACF,WAAW,EAAE,MAAM,CAAC;CACrB"}

package/dist/src/utils/embedding.d.ts

@@ -0,0 +1,44 @@
+export interface EmbeddingConfig {
+    provider: 'openai' | 'deepseek' | 'ollama';
+    model: string;
+    dimensions: number;
+    batchSize: number;
+}
+export declare class EmbeddingService {
+    private client;
+    private config;
+    constructor(config: EmbeddingConfig);
+    /**
+     * 生成单个文本的 embedding
+     * 对于长文本，自动分段嵌入后合并
+     */
+    generateEmbedding(text: string): Promise<number[]>;
+    /**
+     * 批量生成 embeddings
+     */
+    generateEmbeddings(texts: string[]): Promise<number[][]>;
+    /**
+     * 嵌入单个文本块
+     */
+    private embedSingleChunk;
+    /**
+     * 调用本地 Ollama API 生成 embedding
+     */
+    private embedWithOllama;
+    /**
+     * 将长文本切分成重叠的块
+     */
+    private chunkText;
+    /**
+     * 平均合并多个 embeddings（向量的逐元素平均）
+     */
+    private averageEmbeddings;
+}
+export declare function createEmbeddingService(): EmbeddingService;
+/**
+ * 获取共享的 EmbeddingService 单例
+ * 惰性初始化：首次调用时创建，后续调用返回同一实例。
+ * 如果创建失败（如缺少 API Key），返回 null 而非抛出异常。
+ */
+export declare function getEmbeddingService(): EmbeddingService | null;
+//# sourceMappingURL=embedding.d.ts.map

package/dist/src/utils/embedding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding.d.ts","sourceRoot":"","sources":["../../../src/utils/embedding.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,QAAQ,GAAG,UAAU,GAAG,QAAQ,CAAC;IAC3C,KAAK,EAAE,MAAM,CAAC;IACd,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;CACnB;AAOD,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,MAAM,CAAkB;gBAEpB,MAAM,EAAE,eAAe;IAiBnC;;;OAGG;IACG,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IA2BxD;;OAEG;IACG,kBAAkB,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAW9D;;OAEG;YACW,gBAAgB;IAe9B;;OAEG;YACW,eAAe;IAwB7B;;OAEG;IACH,OAAO,CAAC,SAAS;IAajB;;OAEG;IACH,OAAO,CAAC,iBAAiB;CAoB1B;AAGD,wBAAgB,sBAAsB,IAAI,gBAAgB,CAYzD;AAKD;;;;GAIG;AACH,wBAAgB,mBAAmB,IAAI,gBAAgB,GAAG,IAAI,CAW7D"}