lazlo-ai 1.0.0

package/dist/chains/rag.d.ts

@@ -0,0 +1,23 @@
+/**
+ * RAG Chain - Retrieval-Augmented Generation
+ */
+import { BaseChatModel, BaseRetriever, Document } from '../core/types.js';
+export declare class RetrievalQAChain {
+    private llm;
+    private retriever;
+    private promptTemplate;
+    constructor(options: {
+        llm: BaseChatModel;
+        retriever: BaseRetriever;
+        promptTemplate?: string;
+    });
+    invoke(query: string): Promise<{
+        text: string;
+        documents: Document[];
+    }>;
+}
+/**
+ * Factory function for RAG chain
+ */
+export declare function createRetrievalChain(llm: BaseChatModel, retriever: BaseRetriever): RetrievalQAChain;
+//# sourceMappingURL=rag.d.ts.map

package/dist/chains/rag.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag.d.ts","sourceRoot":"","sources":["../../src/chains/rag.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,aAAa,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AAY1E,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,GAAG,CAAgB;IAC3B,OAAO,CAAC,SAAS,CAAgB;IACjC,OAAO,CAAC,cAAc,CAAS;gBAEnB,OAAO,EAAE;QACnB,GAAG,EAAE,aAAa,CAAC;QACnB,SAAS,EAAE,aAAa,CAAC;QACzB,cAAc,CAAC,EAAE,MAAM,CAAC;KACzB;IAMK,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,EAAE,QAAQ,EAAE,CAAA;KAAE,CAAC;CAuB9E;AAED;;GAEG;AACH,wBAAgB,oBAAoB,CAClC,GAAG,EAAE,aAAa,EAClB,SAAS,EAAE,aAAa,GACvB,gBAAgB,CAElB"}

package/dist/chains/rag.js

@@ -0,0 +1,47 @@
+/**
+ * RAG Chain - Retrieval-Augmented Generation
+ */
+import { logger } from '../utils/logger.js';
+const DEFAULT_PROMPT = `You are a helpful assistant. Use the following context to answer the query.
+
+Context:
+{context}
+
+Query: {query}
+
+Answer:`;
+export class RetrievalQAChain {
+    llm;
+    retriever;
+    promptTemplate;
+    constructor(options) {
+        this.llm = options.llm;
+        this.retriever = options.retriever;
+        this.promptTemplate = options.promptTemplate ?? DEFAULT_PROMPT;
+    }
+    async invoke(query) {
+        const docs = await this.retriever.getRelevantDocuments(query);
+        // Combine documents into context
+        const contextParts = docs.map(doc => doc.pageContent);
+        const context = contextParts.join('\n\n---\n\n');
+        // Format prompt
+        const prompt = this.promptTemplate
+            .replace('{context}', context)
+            .replace('{query}', query);
+        logger.debug(`[RetrievalQAChain] Retrieved ${docs.length} documents. Invoking LLM.`);
+        const response = await this.llm.invoke([
+            { role: 'user', content: prompt }
+        ]);
+        return {
+            text: response.content,
+            documents: docs,
+        };
+    }
+}
+/**
+ * Factory function for RAG chain
+ */
+export function createRetrievalChain(llm, retriever) {
+    return new RetrievalQAChain({ llm, retriever });
+}
+//# sourceMappingURL=rag.js.map

package/dist/chains/rag.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rag.js","sourceRoot":"","sources":["../../src/chains/rag.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,MAAM,cAAc,GAAG;;;;;;;QAOf,CAAC;AAET,MAAM,OAAO,gBAAgB;IACnB,GAAG,CAAgB;IACnB,SAAS,CAAgB;IACzB,cAAc,CAAS;IAE/B,YAAY,OAIX;QACC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,GAAG,CAAC;QACvB,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,IAAI,cAAc,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAa;QACxB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;QAE9D,iCAAiC;QACjC,MAAM,YAAY,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;QACtD,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAEjD,gBAAgB;QAChB,MAAM,MAAM,GAAG,IAAI,CAAC,cAAc;aAC/B,OAAO,CAAC,WAAW,EAAE,OAAO,CAAC;aAC7B,OAAO,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAE7B,MAAM,CAAC,KAAK,CAAC,gCAAgC,IAAI,CAAC,MAAM,2BAA2B,CAAC,CAAC;QAErF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC;YACrC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE;SAClC,CAAC,CAAC;QAEH,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,OAAO;YACtB,SAAS,EAAE,IAAI;SAChB,CAAC;IACJ,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,oBAAoB,CAClC,GAAkB,EAClB,SAAwB;IAExB,OAAO,IAAI,gBAAgB,CAAC,EAAE,GAAG,EAAE,SAAS,EAAE,CAAC,CAAC;AAClD,CAAC"}

package/dist/core/types.d.ts

@@ -0,0 +1,130 @@
+/**
+ * Lazlo Core Types & Protocols
+ *
+ * The foundation of Lazlo - protocol definitions that any LLM, tool, or memory
+ * implementation must satisfy. No massive base classes, just duck typing.
+ */
+export interface Message {
+    role: 'system' | 'user' | 'assistant' | 'tool';
+    content: string;
+    name?: string;
+    tool_call_id?: string;
+}
+export interface ToolCall {
+    id: string;
+    type: 'function';
+    function: {
+        name: string;
+        arguments: string | Record<string, unknown>;
+    };
+}
+export interface ChatResponse {
+    content: string;
+    tool_calls?: ToolCall[];
+    usage?: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
+    model?: string;
+    finish_reason?: 'stop' | 'length' | 'tool_calls' | 'content_filter' | null;
+}
+export interface StreamChunk {
+    delta: string;
+    finish_reason?: string;
+    index?: number;
+}
+export interface BaseLLM {
+    invoke(prompt: string, options?: LLMInvokeOptions): Promise<string>;
+    ainvoke(prompt: string, options?: LLMInvokeOptions): Promise<string>;
+    stream?(prompt: string, options?: LLMInvokeOptions): AsyncGenerator<string>;
+    astream?(prompt: string, options?: LLMInvokeOptions): AsyncGenerator<string>;
+}
+export interface BaseChatModel {
+    readonly supportsTools: boolean;
+    readonly supportsStructuredOutput: boolean;
+    invoke(messages: Message[], options?: ChatInvokeOptions): Promise<ChatResponse>;
+    ainvoke?(messages: Message[], options?: ChatInvokeOptions): Promise<ChatResponse>;
+    stream?(messages: Message[], options?: ChatInvokeOptions): AsyncGenerator<StreamChunk>;
+    astream?(messages: Message[], options?: ChatInvokeOptions): AsyncGenerator<StreamChunk>;
+}
+export interface BaseTool {
+    name: string;
+    description: string;
+    argsSchema?: Record<string, unknown>;
+    invoke(input: string | Record<string, unknown>): Promise<string>;
+    ainvoke?(input: string | Record<string, unknown>): Promise<string>;
+}
+export interface BaseMemory {
+    loadMemoryVariables(inputs: Record<string, unknown>): Promise<Record<string, unknown>>;
+    saveContext(inputs: Record<string, unknown>, outputs: Record<string, string>): Promise<void>;
+    clear(): Promise<void>;
+}
+export interface BaseRetriever {
+    getRelevantDocuments(query: string): Promise<Document[]>;
+}
+export interface BaseVectorStore {
+    addTexts(texts: string[], metadatas?: Record<string, unknown>[]): Promise<string[]>;
+    similaritySearch(query: string, k?: number): Promise<Document[]>;
+    addDocuments?(documents: Document[]): Promise<string[]>;
+}
+export interface BaseEmbeddings {
+    embedQuery(text: string): Promise<number[]>;
+    embedDocuments(texts: string[]): Promise<number[][]>;
+}
+export interface Document {
+    pageContent: string;
+    metadata: Record<string, unknown>;
+}
+export interface LLMInvokeOptions {
+    model?: string;
+    temperature?: number;
+    maxTokens?: number;
+    topP?: number;
+    frequencyPenalty?: number;
+    presencePenalty?: number;
+    stop?: string | string[];
+    [key: string]: unknown;
+}
+export interface ChatInvokeOptions extends LLMInvokeOptions {
+    tools?: ToolDefinition[];
+    responseFormat?: ResponseFormat;
+}
+export interface ToolDefinition {
+    type: 'function';
+    function: {
+        name: string;
+        description: string;
+        parameters: Record<string, unknown>;
+    };
+}
+export interface ResponseFormat {
+    type: 'text' | 'json_object' | 'json_schema';
+    json_schema?: {
+        name: string;
+        schema: Record<string, unknown>;
+        strict?: boolean;
+    };
+}
+export interface ChainInputs {
+    [key: string]: unknown;
+}
+export interface ChainOutputs {
+    [key: string]: unknown;
+}
+export interface Chain {
+    invoke(inputs: ChainInputs): Promise<ChainOutputs>;
+    ainvoke?(inputs: ChainInputs): Promise<ChainOutputs>;
+}
+export interface ModelPricing {
+    input: number;
+    output: number;
+}
+export type JsonPrimitive = string | number | boolean | null;
+export type JsonValue = JsonPrimitive | JsonObject | JsonArray;
+export interface JsonObject {
+    [key: string]: JsonValue;
+}
+export interface JsonArray extends Array<JsonValue> {
+}
+//# sourceMappingURL=types.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAMH,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;IAC/C,OAAO,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,QAAQ;IACvB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KAC7C,CAAC;CACH;AAED,MAAM,WAAW,YAAY;IAC3B,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,QAAQ,EAAE,CAAC;IACxB,KAAK,CAAC,EAAE;QACN,aAAa,EAAE,MAAM,CAAC;QACtB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,YAAY,EAAE,MAAM,CAAC;KACtB,CAAC;IACF,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,aAAa,CAAC,EAAE,MAAM,GAAG,QAAQ,GAAG,YAAY,GAAG,gBAAgB,GAAG,IAAI,CAAC;CAC5E;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAMD,MAAM,WAAW,OAAO;IACtB,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,gBAAgB,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACpE,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,gBAAgB,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACrE,MAAM,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,gBAAgB,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;IAC5E,OAAO,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,gBAAgB,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;CAC9E;AAED,MAAM,WAAW,aAAa;IAC5B,QAAQ,CAAC,aAAa,EAAE,OAAO,CAAC;IAChC,QAAQ,CAAC,wBAAwB,EAAE,OAAO,CAAC;IAE3C,MAAM,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IAChF,OAAO,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IAElF,MAAM,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,cAAc,CAAC,WAAW,CAAC,CAAC;IACvF,OAAO,CAAC,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,OAAO,CAAC,EAAE,iBAAiB,GAAG,cAAc,CAAC,WAAW,CAAC,CAAC;CACzF;AAED,MAAM,WAAW,QAAQ;IACvB,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAErC,MAAM,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACjE,OAAO,CAAC,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;CACpE;AAED,MAAM,WAAW,UAAU;IACzB,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IACvF,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IAC7F,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AAED,MAAM,WAAW,aAAa;IAC5B,oBAAoB,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;CAC1D;AAED,MAAM,WAAW,eAAe;IAC9B,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,SAAS,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IACpF,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;IACjE,YAAY,CAAC,CAAC,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;CACzD;AAED,MAAM,WAAW,cAAc;IAC7B,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;IAC5C,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;CACtD;AAMD,MAAM,WAAW,QAAQ;IACvB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAMD,MAAM,WAAW,gBAAgB;IAC/B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;IACzB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,iBAAkB,SAAQ,gBAAgB;IACzD,KAAK,CAAC,EAAE,cAAc,EAAE,CAAC;IACzB,cAAc,CAAC,EAAE,cAAc,CAAC;CACjC;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;KACrC,CAAC;CACH;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,GAAG,aAAa,GAAG,aAAa,CAAC;IAC7C,WAAW,CAAC,EAAE;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAChC,MAAM,CAAC,EAAE,OAAO,CAAC;KAClB,CAAC;CACH;AAMD,MAAM,WAAW,WAAW;IAC1B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,YAAY;IAC3B,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,KAAK;IACpB,MAAM,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IACnD,OAAO,CAAC,CAAC,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;CACtD;AAMD,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAChB;AAMD,MAAM,MAAM,aAAa,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,IAAI,CAAC;AAC7D,MAAM,MAAM,SAAS,GAAG,aAAa,GAAG,UAAU,GAAG,SAAS,CAAC;AAC/D,MAAM,WAAW,UAAU;IAAG,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,CAAC;CAAE;AACzD,MAAM,WAAW,SAAU,SAAQ,KAAK,CAAC,SAAS,CAAC;CAAG"}

package/dist/core/types.js

@@ -0,0 +1,8 @@
+/**
+ * Lazlo Core Types & Protocols
+ *
+ * The foundation of Lazlo - protocol definitions that any LLM, tool, or memory
+ * implementation must satisfy. No massive base classes, just duck typing.
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/dist/core/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/core/types.ts"],"names":[],"mappings":"AAAA;;;;;GAKG"}

package/dist/document_loaders/index.d.ts

@@ -0,0 +1,61 @@
+/**
+ * Document Loaders
+ *
+ * Load documents from various sources: PDF, text, CSV, JSON, HTML
+ */
+export interface Document {
+    pageContent: string;
+    metadata: Record<string, unknown>;
+}
+export interface BaseLoader {
+    load(): Promise<Document[]>;
+    loadSync(): Document[];
+}
+/**
+ * Text File Loader
+ */
+export declare class TextLoader implements BaseLoader {
+    private filePath;
+    constructor(filePath: string);
+    load(): Promise<Document[]>;
+    loadSync(): Document[];
+    private parseContent;
+}
+/**
+ * CSV File Loader
+ */
+export declare class CSVLoader implements BaseLoader {
+    private filePath;
+    private options;
+    constructor(filePath: string, options?: {
+        delimiter?: string;
+        headers?: string[];
+        skipEmptyLines?: boolean;
+    });
+    load(): Promise<Document[]>;
+    loadSync(): Document[];
+    private parseCSV;
+    private parseLine;
+}
+/**
+ * JSON File Loader
+ */
+export declare class JSONLoader implements BaseLoader {
+    private filePath;
+    private options;
+    constructor(filePath: string, options?: {
+        key?: string;
+    });
+    load(): Promise<Document[]>;
+    loadSync(): Document[];
+    private parseJSON;
+}
+/**
+ * Auto-detect file type and load
+ */
+export declare function loadDocument(filePath: string, options?: any): Promise<Document[]>;
+/**
+ * Load multiple files
+ */
+export declare function loadDocuments(filePaths: string[], options?: any): Promise<Document[]>;
+//# sourceMappingURL=index.d.ts.map

package/dist/document_loaders/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/document_loaders/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAMH,MAAM,WAAW,QAAQ;IACvB,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC5B,QAAQ,IAAI,QAAQ,EAAE,CAAC;CACxB;AAED;;GAEG;AACH,qBAAa,UAAW,YAAW,UAAU;IAC/B,OAAO,CAAC,QAAQ;gBAAR,QAAQ,EAAE,MAAM;IAE9B,IAAI,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;IAKjC,QAAQ,IAAI,QAAQ,EAAE;IAKtB,OAAO,CAAC,YAAY;CASrB;AAED;;GAEG;AACH,qBAAa,SAAU,YAAW,UAAU;IAExC,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,OAAO;gBADP,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE;QACf,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;QACnB,cAAc,CAAC,EAAE,OAAO,CAAC;KACrB;IAGF,IAAI,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;IAKjC,QAAQ,IAAI,QAAQ,EAAE;IAKtB,OAAO,CAAC,QAAQ;IAiChB,OAAO,CAAC,SAAS;CAkBlB;AAED;;GAEG;AACH,qBAAa,UAAW,YAAW,UAAU;IAEzC,OAAO,CAAC,QAAQ;IAChB,OAAO,CAAC,OAAO;gBADP,QAAQ,EAAE,MAAM,EAChB,OAAO,GAAE;QACf,GAAG,CAAC,EAAE,MAAM,CAAC;KACT;IAGF,IAAI,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;IAKjC,QAAQ,IAAI,QAAQ,EAAE;IAKtB,OAAO,CAAC,SAAS;CAqClB;AAED;;GAEG;AACH,wBAAsB,YAAY,CAAC,QAAQ,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,GAAG,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC,CAmBvF;AAED;;GAEG;AACH,wBAAsB,aAAa,CACjC,SAAS,EAAE,MAAM,EAAE,EACnB,OAAO,CAAC,EAAE,GAAG,GACZ,OAAO,CAAC,QAAQ,EAAE,CAAC,CASrB"}

package/dist/document_loaders/index.js

@@ -0,0 +1,183 @@
+/**
+ * Document Loaders
+ *
+ * Load documents from various sources: PDF, text, CSV, JSON, HTML
+ */
+import { readFile } from 'fs/promises';
+import { readFileSync } from 'fs';
+import { extname } from 'path';
+/**
+ * Text File Loader
+ */
+export class TextLoader {
+    filePath;
+    constructor(filePath) {
+        this.filePath = filePath;
+    }
+    async load() {
+        const content = await readFile(this.filePath, 'utf-8');
+        return this.parseContent(content);
+    }
+    loadSync() {
+        const content = readFileSync(this.filePath, 'utf-8');
+        return this.parseContent(content);
+    }
+    parseContent(content) {
+        return [{
+                pageContent: content,
+                metadata: {
+                    source: this.filePath,
+                    type: 'text',
+                },
+            }];
+    }
+}
+/**
+ * CSV File Loader
+ */
+export class CSVLoader {
+    filePath;
+    options;
+    constructor(filePath, options = {}) {
+        this.filePath = filePath;
+        this.options = options;
+    }
+    async load() {
+        const content = await readFile(this.filePath, 'utf-8');
+        return this.parseCSV(content);
+    }
+    loadSync() {
+        const content = readFileSync(this.filePath, 'utf-8');
+        return this.parseCSV(content);
+    }
+    parseCSV(content) {
+        const delimiter = this.options.delimiter || ',';
+        const lines = content.split('\n').filter(line => line.trim());
+        if (lines.length === 0)
+            return [];
+        const headers = this.options.headers || this.parseLine(lines[0], delimiter);
+        const documents = [];
+        const startIndex = this.options.headers ? 0 : 1;
+        for (let i = startIndex; i < lines.length; i++) {
+            const values = this.parseLine(lines[i], delimiter);
+            if (this.options.skipEmptyLines && values.every(v => !v.trim()))
+                continue;
+            const obj = {};
+            headers.forEach((header, idx) => {
+                obj[header.trim()] = values[idx] || '';
+            });
+            documents.push({
+                pageContent: JSON.stringify(obj),
+                metadata: {
+                    source: this.filePath,
+                    type: 'csv',
+                    row: i - startIndex + 1,
+                },
+            });
+        }
+        return documents;
+    }
+    parseLine(line, delimiter) {
+        const result = [];
+        let current = '';
+        let inQuotes = false;
+        for (const char of line) {
+            if (char === '"') {
+                inQuotes = !inQuotes;
+            }
+            else if (char === delimiter && !inQuotes) {
+                result.push(current);
+                current = '';
+            }
+            else {
+                current += char;
+            }
+        }
+        result.push(current);
+        return result;
+    }
+}
+/**
+ * JSON File Loader
+ */
+export class JSONLoader {
+    filePath;
+    options;
+    constructor(filePath, options = {}) {
+        this.filePath = filePath;
+        this.options = options;
+    }
+    async load() {
+        const content = await readFile(this.filePath, 'utf-8');
+        return this.parseJSON(content);
+    }
+    loadSync() {
+        const content = readFileSync(this.filePath, 'utf-8');
+        return this.parseJSON(content);
+    }
+    parseJSON(content) {
+        const data = JSON.parse(content);
+        if (this.options.key) {
+            const array = data[this.options.key];
+            if (Array.isArray(array)) {
+                return array.map((item, idx) => ({
+                    pageContent: typeof item === 'string' ? item : JSON.stringify(item),
+                    metadata: {
+                        source: this.filePath,
+                        type: 'json',
+                        key: this.options.key,
+                        index: idx,
+                    },
+                }));
+            }
+        }
+        if (Array.isArray(data)) {
+            return data.map((item, idx) => ({
+                pageContent: typeof item === 'string' ? item : JSON.stringify(item),
+                metadata: {
+                    source: this.filePath,
+                    type: 'json',
+                    index: idx,
+                },
+            }));
+        }
+        return [{
+                pageContent: JSON.stringify(data),
+                metadata: {
+                    source: this.filePath,
+                    type: 'json',
+                },
+            }];
+    }
+}
+/**
+ * Auto-detect file type and load
+ */
+export async function loadDocument(filePath, options) {
+    const ext = extname(filePath).toLowerCase();
+    switch (ext) {
+        case '.txt':
+        case '.md':
+        case '.text':
+            return new TextLoader(filePath).load();
+        case '.csv':
+            return new CSVLoader(filePath, options?.csv).load();
+        case '.json':
+            return new JSONLoader(filePath, options?.json).load();
+        default:
+            // Default to text loader
+            return new TextLoader(filePath).load();
+    }
+}
+/**
+ * Load multiple files
+ */
+export async function loadDocuments(filePaths, options) {
+    const documents = [];
+    for (const filePath of filePaths) {
+        const docs = await loadDocument(filePath, options);
+        documents.push(...docs);
+    }
+    return documents;
+}
+//# sourceMappingURL=index.js.map

package/dist/document_loaders/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/document_loaders/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EAAQ,OAAO,EAAE,MAAM,MAAM,CAAC;AAYrC;;GAEG;AACH,MAAM,OAAO,UAAU;IACD;IAApB,YAAoB,QAAgB;QAAhB,aAAQ,GAAR,QAAQ,CAAQ;IAAG,CAAC;IAExC,KAAK,CAAC,IAAI;QACR,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IAED,QAAQ;QACN,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACrD,OAAO,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IAEO,YAAY,CAAC,OAAe;QAClC,OAAO,CAAC;gBACN,WAAW,EAAE,OAAO;gBACpB,QAAQ,EAAE;oBACR,MAAM,EAAE,IAAI,CAAC,QAAQ;oBACrB,IAAI,EAAE,MAAM;iBACb;aACF,CAAC,CAAC;IACL,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,SAAS;IAEV;IACA;IAFV,YACU,QAAgB,EAChB,UAIJ,EAAE;QALE,aAAQ,GAAR,QAAQ,CAAQ;QAChB,YAAO,GAAP,OAAO,CAIT;IACL,CAAC;IAEJ,KAAK,CAAC,IAAI;QACR,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,QAAQ;QACN,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACrD,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAEO,QAAQ,CAAC,OAAe;QAC9B,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,IAAI,GAAG,CAAC;QAChD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAE9D,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAElC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QAC5E,MAAM,SAAS,GAAe,EAAE,CAAC;QAEjC,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAEhD,KAAK,IAAI,CAAC,GAAG,UAAU,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC/C,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YACnD,IAAI,IAAI,CAAC,OAAO,CAAC,cAAc,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;gBAAE,SAAS;YAE1E,MAAM,GAAG,GAA2B,EAAE,CAAC;YACvC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;gBAC9B,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;YACzC,CAAC,CAAC,CAAC;YAEH,SAAS,CAAC,IAAI,CAAC;gBACb,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC;gBAChC,QAAQ,EAAE;oBACR,MAAM,EAAE,IAAI,CAAC,QAAQ;oBACrB,IAAI,EAAE,KAAK;oBACX,GAAG,EAAE,CAAC,GAAG,UAAU,GAAG,CAAC;iBACxB;aACF,CAAC,CAAC;QACL,CAAC;QAED,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,SAAS,CAAC,IAAY,EAAE,SAAiB;QAC/C,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,IAAI,OAAO,GAAG,EAAE,CAAC;QACjB,IAAI,QAAQ,GAAG,KAAK,CAAC;QAErB,KAAK,MAAM,IAAI,IAAI,IAAI,EAAE,CAAC;YACxB,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;gBACjB,QAAQ,GAAG,CAAC,QAAQ,CAAC;YACvB,CAAC;iBAAM,IAAI,IAAI,KAAK,SAAS,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAC3C,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACrB,OAAO,GAAG,EAAE,CAAC;YACf,CAAC;iBAAM,CAAC;gBACN,OAAO,IAAI,IAAI,CAAC;YAClB,CAAC;QACH,CAAC;QACD,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACrB,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,OAAO,UAAU;IAEX;IACA;IAFV,YACU,QAAgB,EAChB,UAEJ,EAAE;QAHE,aAAQ,GAAR,QAAQ,CAAQ;QAChB,YAAO,GAAP,OAAO,CAET;IACL,CAAC;IAEJ,KAAK,CAAC,IAAI;QACR,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACvD,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;IAED,QAAQ;QACN,MAAM,OAAO,GAAG,YAAY,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACrD,OAAO,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACjC,CAAC;IAEO,SAAS,CAAC,OAAe;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAEjC,IAAI,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;YACrB,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACrC,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC/B,WAAW,EAAE,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;oBACnE,QAAQ,EAAE;wBACR,MAAM,EAAE,IAAI,CAAC,QAAQ;wBACrB,IAAI,EAAE,MAAM;wBACZ,GAAG,EAAE,IAAI,CAAC,OAAO,CAAC,GAAG;wBACrB,KAAK,EAAE,GAAG;qBACX;iBACF,CAAC,CAAC,CAAC;YACN,CAAC;QACH,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YACxB,OAAO,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;gBAC9B,WAAW,EAAE,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;gBACnE,QAAQ,EAAE;oBACR,MAAM,EAAE,IAAI,CAAC,QAAQ;oBACrB,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,GAAG;iBACX;aACF,CAAC,CAAC,CAAC;QACN,CAAC;QAED,OAAO,CAAC;gBACN,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;gBACjC,QAAQ,EAAE;oBACR,MAAM,EAAE,IAAI,CAAC,QAAQ;oBACrB,IAAI,EAAE,MAAM;iBACb;aACF,CAAC,CAAC;IACL,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,QAAgB,EAAE,OAAa;IAChE,MAAM,GAAG,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IAE5C,QAAQ,GAAG,EAAE,CAAC;QACZ,KAAK,MAAM,CAAC;QACZ,KAAK,KAAK,CAAC;QACX,KAAK,OAAO;YACV,OAAO,IAAI,UAAU,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QAEzC,KAAK,MAAM;YACT,OAAO,IAAI,SAAS,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QAEtD,KAAK,OAAO;YACV,OAAO,IAAI,UAAU,CAAC,QAAQ,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;QAExD;YACE,yBAAyB;YACzB,OAAO,IAAI,UAAU,CAAC,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;IAC3C,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,SAAmB,EACnB,OAAa;IAEb,MAAM,SAAS,GAAe,EAAE,CAAC;IAEjC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,MAAM,IAAI,GAAG,MAAM,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QACnD,SAAS,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC"}

package/dist/embeddings/google.d.ts

@@ -0,0 +1,43 @@
+/**
+ * Google Embeddings
+ *
+ * Google AI Gemini embeddings for text vectorization
+ */
+export interface GoogleEmbeddingsOptions {
+    apiKey?: string;
+    model?: string;
+    taskType?: 'RETRIEVAL_QUERY' | 'RETRIEVAL_DOCUMENT' | 'SEMANTIC_SIMILARITY' | 'CLASSIFICATION' | 'CLUSTERING';
+    title?: string;
+}
+export interface EmbeddingResponse {
+    embedding: {
+        values: number[];
+    };
+}
+/**
+ * Google Embeddings
+ *
+ * Uses Google's Generative Language API for embeddings
+ */
+export declare class GoogleEmbeddings {
+    readonly model: string;
+    readonly taskType: string;
+    readonly title?: string;
+    private apiKey;
+    private apiUrl;
+    constructor(options?: GoogleEmbeddingsOptions);
+    /**
+     * Embed a single text
+     */
+    embedQuery(text: string): Promise<number[]>;
+    /**
+     * Embed multiple texts
+     */
+    embedDocuments(texts: string[]): Promise<number[][]>;
+    private embed;
+}
+/**
+ * Factory function
+ */
+export declare function createGoogleEmbeddings(options?: GoogleEmbeddingsOptions): GoogleEmbeddings;
+//# sourceMappingURL=google.d.ts.map

package/dist/embeddings/google.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../../src/embeddings/google.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,MAAM,WAAW,uBAAuB;IACtC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,iBAAiB,GAAG,oBAAoB,GAAG,qBAAqB,GAAG,gBAAgB,GAAG,YAAY,CAAC;IAC9G,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,iBAAiB;IAChC,SAAS,EAAE;QACT,MAAM,EAAE,MAAM,EAAE,CAAC;KAClB,CAAC;CACH;AAED;;;;GAIG;AACH,qBAAa,gBAAgB;IAC3B,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;IAC1B,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IACxB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,MAAM,CAAsD;gBAExD,OAAO,GAAE,uBAA4B;IAWjD;;OAEG;IACG,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAIjD;;OAEG;IACG,cAAc,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;YAI5C,KAAK;CAoDpB;AAED;;GAEG;AACH,wBAAgB,sBAAsB,CAAC,OAAO,CAAC,EAAE,uBAAuB,GAAG,gBAAgB,CAE1F"}

package/dist/embeddings/google.js

@@ -0,0 +1,90 @@
+/**
+ * Google Embeddings
+ *
+ * Google AI Gemini embeddings for text vectorization
+ */
+import axios from 'axios';
+import { logger } from '../utils/logger.js';
+/**
+ * Google Embeddings
+ *
+ * Uses Google's Generative Language API for embeddings
+ */
+export class GoogleEmbeddings {
+    model;
+    taskType;
+    title;
+    apiKey;
+    apiUrl = 'https://generativelanguage.googleapis.com/v1beta';
+    constructor(options = {}) {
+        this.apiKey = options.apiKey || (typeof process !== 'undefined' ? process.env?.GEMINI_API_KEY : '') || '';
+        this.model = options.model || 'text-embedding-004';
+        this.taskType = options.taskType || 'RETRIEVAL_DOCUMENT';
+        this.title = options.title;
+        if (!this.apiKey) {
+            logger.warn('[GoogleEmbeddings] No API key provided. Set GEMINI_API_KEY env var or pass apiKey option.');
+        }
+    }
+    /**
+     * Embed a single text
+     */
+    async embedQuery(text) {
+        return this.embed([text]).then(res => res[0]);
+    }
+    /**
+     * Embed multiple texts
+     */
+    async embedDocuments(texts) {
+        return this.embed(texts);
+    }
+    async embed(texts) {
+        if (!this.apiKey) {
+            throw new Error('Google API key required. Set GEMINI_API_KEY or pass apiKey option.');
+        }
+        const embeddings = [];
+        // Google API allows up to 100 inputs per request
+        const batchSize = 100;
+        for (let i = 0; i < texts.length; i += batchSize) {
+            const batch = texts.slice(i, i + batchSize);
+            try {
+                const requests = batch.map((content, idx) => ({
+                    model: `models/${this.model}`,
+                    content: { role: 'user', parts: [{ text: content }] },
+                    taskType: this.taskType,
+                    title: this.title,
+                }));
+                const response = await axios.post(`${this.apiUrl}/embedContent:batchEmbedContents?key=${this.apiKey}`, { requests }, {
+                    headers: { 'Content-Type': 'application/json' },
+                    timeout: 60000,
+                });
+                const results = response.data.embeddings || [];
+                for (let j = 0; j < results.length; j++) {
+                    const result = results[j];
+                    if (result.embedding?.values) {
+                        embeddings.push(result.embedding.values);
+                    }
+                    else {
+                        // Fallback: generate random embedding
+                        logger.warn(`[GoogleEmbeddings] Empty embedding for batch ${Math.floor(i / batchSize)}, index ${j}`);
+                        embeddings.push(new Array(768).fill(0));
+                    }
+                }
+            }
+            catch (error) {
+                logger.error(`[GoogleEmbeddings] Error embedding batch: ${error.message}`);
+                // Return zero embeddings on error
+                for (let j = 0; j < batch.length; j++) {
+                    embeddings.push(new Array(768).fill(0));
+                }
+            }
+        }
+        return embeddings;
+    }
+}
+/**
+ * Factory function
+ */
+export function createGoogleEmbeddings(options) {
+    return new GoogleEmbeddings(options);
+}
+//# sourceMappingURL=google.js.map

package/dist/embeddings/google.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google.js","sourceRoot":"","sources":["../../src/embeddings/google.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,KAAK,MAAM,OAAO,CAAC;AAC1B,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAe5C;;;;GAIG;AACH,MAAM,OAAO,gBAAgB;IAClB,KAAK,CAAS;IACd,QAAQ,CAAS;IACjB,KAAK,CAAU;IAChB,MAAM,CAAS;IACf,MAAM,GAAG,kDAAkD,CAAC;IAEpE,YAAY,UAAmC,EAAE;QAC/C,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,CAAC,OAAO,OAAO,KAAK,WAAW,CAAC,CAAC,CAAE,OAAe,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC;QACnH,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,oBAAoB,CAAC;QACnD,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,oBAAoB,CAAC;QACzD,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAE3B,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,CAAC,IAAI,CAAC,2FAA2F,CAAC,CAAC;QAC3G,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,IAAY;QAC3B,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,KAAe;QAClC,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAEO,KAAK,CAAC,KAAK,CAAC,KAAe;QACjC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,oEAAoE,CAAC,CAAC;QACxF,CAAC;QAED,MAAM,UAAU,GAAe,EAAE,CAAC;QAElC,iDAAiD;QACjD,MAAM,SAAS,GAAG,GAAG,CAAC;QAEtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,IAAI,SAAS,EAAE,CAAC;YACjD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,CAAC;YAE5C,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBAC5C,KAAK,EAAE,UAAU,IAAI,CAAC,KAAK,EAAE;oBAC7B,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAE;oBACrD,QAAQ,EAAE,IAAI,CAAC,QAAQ;oBACvB,KAAK,EAAE,IAAI,CAAC,KAAK;iBAClB,CAAC,CAAC,CAAC;gBAEJ,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,GAAG,IAAI,CAAC,MAAM,wCAAwC,IAAI,CAAC,MAAM,EAAE,EACnE,EAAE,QAAQ,EAAE,EACZ;oBACE,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;oBAC/C,OAAO,EAAE,KAAK;iBACf,CACF,CAAC;gBAEF,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,UAAU,IAAI,EAAE,CAAC;gBAC/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBACxC,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;oBAC1B,IAAI,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,CAAC;wBAC7B,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;oBAC3C,CAAC;yBAAM,CAAC;wBACN,sCAAsC;wBACtC,MAAM,CAAC,IAAI,CAAC,gDAAgD,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,SAAS,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;wBACrG,UAAU,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;oBAC1C,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,MAAM,CAAC,KAAK,CAAC,6CAA6C,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;gBAC3E,kCAAkC;gBAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBACtC,UAAU,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;CACF;AAED;;GAEG;AACH,MAAM,UAAU,sBAAsB,CAAC,OAAiC;IACtE,OAAO,IAAI,gBAAgB,CAAC,OAAO,CAAC,CAAC;AACvC,CAAC"}