@librechat/agents 3.2.31 → 3.2.33

package/dist/esm/tools/search/rerankers.mjs

@@ -1,172 +1,153 @@
-import axios from 'axios';
-import { createDefaultLogger, formatErrorForLog } from './utils.mjs';
-
-const DEFAULT_JINA_API_URL = 'https://api.jina.ai/v1/rerank';
-const getDefaultJinaApiUrl = () => process.env.JINA_API_URL != null && process.env.JINA_API_URL !== ''
-    ? process.env.JINA_API_URL
-    : DEFAULT_JINA_API_URL;
-class BaseReranker {
-    apiKey;
-    logger;
-    constructor(logger) {
-        // Each specific reranker will set its API key
-        this.logger = logger || createDefaultLogger();
-    }
-    getDefaultRanking(documents, topK) {
-        return documents
-            .slice(0, Math.min(topK, documents.length))
-            .map((doc) => ({ text: doc, score: 0 }));
-    }
-}
-class JinaReranker extends BaseReranker {
-    apiUrl;
-    constructor({ apiKey = process.env.JINA_API_KEY, apiUrl = getDefaultJinaApiUrl(), logger, }) {
-        super(logger);
-        this.apiKey = apiKey;
-        this.apiUrl = apiUrl;
-    }
-    async rerank(query, documents, topK = 5) {
-        this.logger.debug(`Reranking ${documents.length} chunks with Jina using API URL: ${this.apiUrl}`);
-        try {
-            if (this.apiKey == null || this.apiKey === '') {
-                this.logger.warn('JINA_API_KEY is not set. Using default ranking.');
-                return this.getDefaultRanking(documents, topK);
-            }
-            const requestData = {
-                model: 'jina-reranker-v2-base-multilingual',
-                query: query,
-                top_n: topK,
-                documents: documents,
-                return_documents: true,
-            };
-            const response = await axios.post(this.apiUrl, requestData, {
-                headers: {
-                    'Content-Type': 'application/json',
-                    Authorization: `Bearer ${this.apiKey}`,
-                },
-            });
-            this.logger.debug('Jina API Model:', response.data?.model);
-            this.logger.debug('Jina API Usage:', response.data?.usage);
-            if (response.data && response.data.results.length) {
-                return response.data.results.map((result) => {
-                    const docIndex = result.index;
-                    const score = result.relevance_score;
-                    let text = '';
-                    // If return_documents is true, the document field will be present
-                    if (result.document != null) {
-                        const doc = result.document;
-                        if (typeof doc === 'object' && 'text' in doc) {
-                            text = doc.text;
-                        }
-                        else if (typeof doc === 'string') {
-                            text = doc;
-                        }
-                    }
-                    else {
-                        // Otherwise, use the index to get the document
-                        text = documents[docIndex];
-                    }
-                    return { text, score };
-                });
-            }
-            else {
-                this.logger.warn('Unexpected response format from Jina API. Using default ranking.');
-                return this.getDefaultRanking(documents, topK);
-            }
-        }
-        catch (error) {
-            this.logger.error('Error using Jina reranker', formatErrorForLog(error));
-            // Fallback to default ranking on error
-            return this.getDefaultRanking(documents, topK);
-        }
-    }
-}
-class CohereReranker extends BaseReranker {
-    constructor({ apiKey = process.env.COHERE_API_KEY, logger, }) {
-        super(logger);
-        this.apiKey = apiKey;
-    }
-    async rerank(query, documents, topK = 5) {
-        this.logger.debug(`Reranking ${documents.length} chunks with Cohere`);
-        try {
-            if (this.apiKey == null || this.apiKey === '') {
-                this.logger.warn('COHERE_API_KEY is not set. Using default ranking.');
-                return this.getDefaultRanking(documents, topK);
-            }
-            const requestData = {
-                model: 'rerank-v3.5',
-                query: query,
-                top_n: topK,
-                documents: documents,
-            };
-            const response = await axios.post('https://api.cohere.com/v2/rerank', requestData, {
-                headers: {
-                    'Content-Type': 'application/json',
-                    Authorization: `Bearer ${this.apiKey}`,
-                },
-            });
-            this.logger.debug('Cohere API ID:', response.data?.id);
-            this.logger.debug('Cohere API Meta:', response.data?.meta);
-            if (response.data && response.data.results.length) {
-                return response.data.results.map((result) => {
-                    const docIndex = result.index;
-                    const score = result.relevance_score;
-                    const text = documents[docIndex];
-                    return { text, score };
-                });
-            }
-            else {
-                this.logger.warn('Unexpected response format from Cohere API. Using default ranking.');
-                return this.getDefaultRanking(documents, topK);
-            }
-        }
-        catch (error) {
-            this.logger.error('Error using Cohere reranker', formatErrorForLog(error));
-            // Fallback to default ranking on error
-            return this.getDefaultRanking(documents, topK);
-        }
-    }
-}
-class InfinityReranker extends BaseReranker {
-    constructor(logger) {
-        super(logger);
-        // No API key needed for the placeholder implementation
-    }
-    async rerank(query, documents, topK = 5) {
-        this.logger.debug(`Reranking ${documents.length} chunks with Infinity (placeholder)`);
-        // This would be replaced with actual Infinity reranker implementation
-        return this.getDefaultRanking(documents, topK);
-    }
-}
+import { createDefaultLogger, formatErrorForLog } from "./utils.mjs";
+import axios from "axios";
+//#region src/tools/search/rerankers.ts
+const DEFAULT_JINA_API_URL = "https://api.jina.ai/v1/rerank";
+const getDefaultJinaApiUrl = () => process.env.JINA_API_URL != null && process.env.JINA_API_URL !== "" ? process.env.JINA_API_URL : DEFAULT_JINA_API_URL;
+var BaseReranker = class {
+	apiKey;
+	logger;
+	constructor(logger) {
+		this.logger = logger || createDefaultLogger();
+	}
+	getDefaultRanking(documents, topK) {
+		return documents.slice(0, Math.min(topK, documents.length)).map((doc) => ({
+			text: doc,
+			score: 0
+		}));
+	}
+};
+var JinaReranker = class extends BaseReranker {
+	apiUrl;
+	constructor({ apiKey = process.env.JINA_API_KEY, apiUrl = getDefaultJinaApiUrl(), logger }) {
+		super(logger);
+		this.apiKey = apiKey;
+		this.apiUrl = apiUrl;
+	}
+	async rerank(query, documents, topK = 5) {
+		this.logger.debug(`Reranking ${documents.length} chunks with Jina using API URL: ${this.apiUrl}`);
+		try {
+			if (this.apiKey == null || this.apiKey === "") {
+				this.logger.warn("JINA_API_KEY is not set. Using default ranking.");
+				return this.getDefaultRanking(documents, topK);
+			}
+			const requestData = {
+				model: "jina-reranker-v2-base-multilingual",
+				query,
+				top_n: topK,
+				documents,
+				return_documents: true
+			};
+			const response = await axios.post(this.apiUrl, requestData, { headers: {
+				"Content-Type": "application/json",
+				Authorization: `Bearer ${this.apiKey}`
+			} });
+			this.logger.debug("Jina API Model:", response.data?.model);
+			this.logger.debug("Jina API Usage:", response.data?.usage);
+			if (response.data && response.data.results.length) return response.data.results.map((result) => {
+				const docIndex = result.index;
+				const score = result.relevance_score;
+				let text = "";
+				if (result.document != null) {
+					const doc = result.document;
+					if (typeof doc === "object" && "text" in doc) text = doc.text;
+					else if (typeof doc === "string") text = doc;
+				} else text = documents[docIndex];
+				return {
+					text,
+					score
+				};
+			});
+			else {
+				this.logger.warn("Unexpected response format from Jina API. Using default ranking.");
+				return this.getDefaultRanking(documents, topK);
+			}
+		} catch (error) {
+			this.logger.error("Error using Jina reranker", formatErrorForLog(error));
+			return this.getDefaultRanking(documents, topK);
+		}
+	}
+};
+var CohereReranker = class extends BaseReranker {
+	constructor({ apiKey = process.env.COHERE_API_KEY, logger }) {
+		super(logger);
+		this.apiKey = apiKey;
+	}
+	async rerank(query, documents, topK = 5) {
+		this.logger.debug(`Reranking ${documents.length} chunks with Cohere`);
+		try {
+			if (this.apiKey == null || this.apiKey === "") {
+				this.logger.warn("COHERE_API_KEY is not set. Using default ranking.");
+				return this.getDefaultRanking(documents, topK);
+			}
+			const requestData = {
+				model: "rerank-v3.5",
+				query,
+				top_n: topK,
+				documents
+			};
+			const response = await axios.post("https://api.cohere.com/v2/rerank", requestData, { headers: {
+				"Content-Type": "application/json",
+				Authorization: `Bearer ${this.apiKey}`
+			} });
+			this.logger.debug("Cohere API ID:", response.data?.id);
+			this.logger.debug("Cohere API Meta:", response.data?.meta);
+			if (response.data && response.data.results.length) return response.data.results.map((result) => {
+				const docIndex = result.index;
+				const score = result.relevance_score;
+				return {
+					text: documents[docIndex],
+					score
+				};
+			});
+			else {
+				this.logger.warn("Unexpected response format from Cohere API. Using default ranking.");
+				return this.getDefaultRanking(documents, topK);
+			}
+		} catch (error) {
+			this.logger.error("Error using Cohere reranker", formatErrorForLog(error));
+			return this.getDefaultRanking(documents, topK);
+		}
+	}
+};
+var InfinityReranker = class extends BaseReranker {
+	constructor(logger) {
+		super(logger);
+	}
+	async rerank(query, documents, topK = 5) {
+		this.logger.debug(`Reranking ${documents.length} chunks with Infinity (placeholder)`);
+		return this.getDefaultRanking(documents, topK);
+	}
+};
 /**
- * Creates the appropriate reranker based on type and configuration
- */
+* Creates the appropriate reranker based on type and configuration
+*/
 const createReranker = (config) => {
-    const { rerankerType, jinaApiKey, jinaApiUrl, cohereApiKey, logger } = config;
-    // Create a default logger if none is provided
-    const defaultLogger = logger || createDefaultLogger();
-    switch (rerankerType.toLowerCase()) {
-        case 'jina':
-            return new JinaReranker({ apiKey: jinaApiKey, apiUrl: jinaApiUrl, logger: defaultLogger });
-        case 'cohere':
-            return new CohereReranker({
-                apiKey: cohereApiKey,
-                logger: defaultLogger,
-            });
-        case 'infinity':
-            return new InfinityReranker(defaultLogger);
-        case 'none':
-            defaultLogger.debug('Skipping reranking as reranker is set to "none"');
-            return undefined;
-        default:
-            defaultLogger.warn(`Unknown reranker type: ${rerankerType}. Defaulting to InfinityReranker.`);
-            return new JinaReranker({ apiKey: jinaApiKey, apiUrl: jinaApiUrl, logger: defaultLogger });
-    }
+	const { rerankerType, jinaApiKey, jinaApiUrl, cohereApiKey, logger } = config;
+	const defaultLogger = logger || createDefaultLogger();
+	switch (rerankerType.toLowerCase()) {
+		case "jina": return new JinaReranker({
+			apiKey: jinaApiKey,
+			apiUrl: jinaApiUrl,
+			logger: defaultLogger
+		});
+		case "cohere": return new CohereReranker({
+			apiKey: cohereApiKey,
+			logger: defaultLogger
+		});
+		case "infinity": return new InfinityReranker(defaultLogger);
+		case "none":
+			defaultLogger.debug("Skipping reranking as reranker is set to \"none\"");
+			return;
+		default:
+			defaultLogger.warn(`Unknown reranker type: ${rerankerType}. Defaulting to InfinityReranker.`);
+			return new JinaReranker({
+				apiKey: jinaApiKey,
+				apiUrl: jinaApiUrl,
+				logger: defaultLogger
+			});
+	}
 };
-// Example usage:
-// const jinaReranker = new JinaReranker();
-// const cohereReranker = new CohereReranker();
-// const infinityReranker = new InfinityReranker();
+//#endregion
+export { createReranker };
 
-export { BaseReranker, CohereReranker, InfinityReranker, JinaReranker, createReranker };
-//# sourceMappingURL=rerankers.mjs.map
+//# sourceMappingURL=rerankers.mjs.map

package/dist/esm/tools/search/rerankers.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"rerankers.mjs","sources":["../../../../src/tools/search/rerankers.ts"],"sourcesContent":["import axios from 'axios';\nimport type * as t from './types';\nimport { createDefaultLogger, formatErrorForLog } from './utils';\n\nconst DEFAULT_JINA_API_URL = 'https://api.jina.ai/v1/rerank';\n\nconst getDefaultJinaApiUrl = (): string =>\n  process.env.JINA_API_URL != null && process.env.JINA_API_URL !== ''\n    ? process.env.JINA_API_URL\n    : DEFAULT_JINA_API_URL;\n\nexport abstract class BaseReranker {\n  protected apiKey: string | undefined;\n  protected logger: t.Logger;\n\n  constructor(logger?: t.Logger) {\n    // Each specific reranker will set its API key\n    this.logger = logger || createDefaultLogger();\n  }\n\n  abstract rerank(\n    query: string,\n    documents: string[],\n    topK?: number\n  ): Promise<t.Highlight[]>;\n\n  protected getDefaultRanking(\n    documents: string[],\n    topK: number\n  ): t.Highlight[] {\n    return documents\n      .slice(0, Math.min(topK, documents.length))\n      .map((doc) => ({ text: doc, score: 0 }));\n  }\n}\n\nexport class JinaReranker extends BaseReranker {\n  private apiUrl: string;\n\n  constructor({\n    apiKey = process.env.JINA_API_KEY,\n    apiUrl = getDefaultJinaApiUrl(),\n    logger,\n  }: {\n    apiKey?: string;\n    apiUrl?: string;\n    logger?: t.Logger;\n  }) {\n    super(logger);\n    this.apiKey = apiKey;\n    this.apiUrl = apiUrl;\n  }\n\n  async rerank(\n    query: string,\n    documents: string[],\n    topK: number = 5\n  ): Promise<t.Highlight[]> {\n    this.logger.debug(`Reranking ${documents.length} chunks with Jina using API URL: ${this.apiUrl}`);\n\n    try {\n      if (this.apiKey == null || this.apiKey === '') {\n        this.logger.warn('JINA_API_KEY is not set. Using default ranking.');\n        return this.getDefaultRanking(documents, topK);\n      }\n\n      const requestData = {\n        model: 'jina-reranker-v2-base-multilingual',\n        query: query,\n        top_n: topK,\n        documents: documents,\n        return_documents: true,\n      };\n\n      const response = await axios.post<t.JinaRerankerResponse | undefined>(\n        this.apiUrl,\n        requestData,\n        {\n          headers: {\n            'Content-Type': 'application/json',\n            Authorization: `Bearer ${this.apiKey}`,\n          },\n        }\n      );\n\n      this.logger.debug('Jina API Model:', response.data?.model);\n      this.logger.debug('Jina API Usage:', response.data?.usage);\n\n      if (response.data && response.data.results.length) {\n        return response.data.results.map((result) => {\n          const docIndex = result.index;\n          const score = result.relevance_score;\n          let text = '';\n\n          // If return_documents is true, the document field will be present\n          if (result.document != null) {\n            const doc = result.document;\n            if (typeof doc === 'object' && 'text' in doc) {\n              text = doc.text;\n            } else if (typeof doc === 'string') {\n              text = doc;\n            }\n          } else {\n            // Otherwise, use the index to get the document\n            text = documents[docIndex];\n          }\n\n          return { text, score };\n        });\n      } else {\n        this.logger.warn(\n          'Unexpected response format from Jina API. Using default ranking.'\n        );\n        return this.getDefaultRanking(documents, topK);\n      }\n    } catch (error) {\n      this.logger.error('Error using Jina reranker', formatErrorForLog(error));\n      // Fallback to default ranking on error\n      return this.getDefaultRanking(documents, topK);\n    }\n  }\n}\n\nexport class CohereReranker extends BaseReranker {\n  constructor({\n    apiKey = process.env.COHERE_API_KEY,\n    logger,\n  }: {\n    apiKey?: string;\n    logger?: t.Logger;\n  }) {\n    super(logger);\n    this.apiKey = apiKey;\n  }\n\n  async rerank(\n    query: string,\n    documents: string[],\n    topK: number = 5\n  ): Promise<t.Highlight[]> {\n    this.logger.debug(`Reranking ${documents.length} chunks with Cohere`);\n\n    try {\n      if (this.apiKey == null || this.apiKey === '') {\n        this.logger.warn('COHERE_API_KEY is not set. Using default ranking.');\n        return this.getDefaultRanking(documents, topK);\n      }\n\n      const requestData = {\n        model: 'rerank-v3.5',\n        query: query,\n        top_n: topK,\n        documents: documents,\n      };\n\n      const response = await axios.post<t.CohereRerankerResponse | undefined>(\n        'https://api.cohere.com/v2/rerank',\n        requestData,\n        {\n          headers: {\n            'Content-Type': 'application/json',\n            Authorization: `Bearer ${this.apiKey}`,\n          },\n        }\n      );\n\n      this.logger.debug('Cohere API ID:', response.data?.id);\n      this.logger.debug('Cohere API Meta:', response.data?.meta);\n\n      if (response.data && response.data.results.length) {\n        return response.data.results.map((result) => {\n          const docIndex = result.index;\n          const score = result.relevance_score;\n          const text = documents[docIndex];\n          return { text, score };\n        });\n      } else {\n        this.logger.warn(\n          'Unexpected response format from Cohere API. Using default ranking.'\n        );\n        return this.getDefaultRanking(documents, topK);\n      }\n    } catch (error) {\n      this.logger.error(\n        'Error using Cohere reranker',\n        formatErrorForLog(error)\n      );\n      // Fallback to default ranking on error\n      return this.getDefaultRanking(documents, topK);\n    }\n  }\n}\n\nexport class InfinityReranker extends BaseReranker {\n  constructor(logger?: t.Logger) {\n    super(logger);\n    // No API key needed for the placeholder implementation\n  }\n\n  async rerank(\n    query: string,\n    documents: string[],\n    topK: number = 5\n  ): Promise<t.Highlight[]> {\n    this.logger.debug(\n      `Reranking ${documents.length} chunks with Infinity (placeholder)`\n    );\n    // This would be replaced with actual Infinity reranker implementation\n    return this.getDefaultRanking(documents, topK);\n  }\n}\n\n/**\n * Creates the appropriate reranker based on type and configuration\n */\nexport const createReranker = (config: {\n  rerankerType: t.RerankerType;\n  jinaApiKey?: string;\n  jinaApiUrl?: string;\n  cohereApiKey?: string;\n  logger?: t.Logger;\n}): BaseReranker | undefined => {\n  const { rerankerType, jinaApiKey, jinaApiUrl, cohereApiKey, logger } = config;\n\n  // Create a default logger if none is provided\n  const defaultLogger = logger || createDefaultLogger();\n\n  switch (rerankerType.toLowerCase()) {\n  case 'jina':\n    return new JinaReranker({ apiKey: jinaApiKey, apiUrl: jinaApiUrl, logger: defaultLogger });\n  case 'cohere':\n    return new CohereReranker({\n      apiKey: cohereApiKey,\n      logger: defaultLogger,\n    });\n  case 'infinity':\n    return new InfinityReranker(defaultLogger);\n  case 'none':\n    defaultLogger.debug('Skipping reranking as reranker is set to \"none\"');\n    return undefined;\n  default:\n    defaultLogger.warn(\n      `Unknown reranker type: ${rerankerType}. Defaulting to InfinityReranker.`\n    );\n    return new JinaReranker({ apiKey: jinaApiKey, apiUrl: jinaApiUrl, logger: defaultLogger });\n  }\n};\n\n// Example usage:\n// const jinaReranker = new JinaReranker();\n// const cohereReranker = new CohereReranker();\n// const infinityReranker = new InfinityReranker();\n"],"names":[],"mappings":";;;AAIA,MAAM,oBAAoB,GAAG,+BAA+B;AAE5D,MAAM,oBAAoB,GAAG,MAC3B,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC,YAAY,KAAK;AAC/D,MAAE,OAAO,CAAC,GAAG,CAAC;MACZ,oBAAoB;MAEJ,YAAY,CAAA;AACtB,IAAA,MAAM;AACN,IAAA,MAAM;AAEhB,IAAA,WAAA,CAAY,MAAiB,EAAA;;AAE3B,QAAA,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,mBAAmB,EAAE;IAC/C;IAQU,iBAAiB,CACzB,SAAmB,EACnB,IAAY,EAAA;AAEZ,QAAA,OAAO;AACJ,aAAA,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC,MAAM,CAAC;AACzC,aAAA,GAAG,CAAC,CAAC,GAAG,MAAM,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;IAC5C;AACD;AAEK,MAAO,YAAa,SAAQ,YAAY,CAAA;AACpC,IAAA,MAAM;AAEd,IAAA,WAAA,CAAY,EACV,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,YAAY,EACjC,MAAM,GAAG,oBAAoB,EAAE,EAC/B,MAAM,GAKP,EAAA;QACC,KAAK,CAAC,MAAM,CAAC;AACb,QAAA,IAAI,CAAC,MAAM,GAAG,MAAM;AACpB,QAAA,IAAI,CAAC,MAAM,GAAG,MAAM;IACtB;IAEA,MAAM,MAAM,CACV,KAAa,EACb,SAAmB,EACnB,OAAe,CAAC,EAAA;AAEhB,QAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA,UAAA,EAAa,SAAS,CAAC,MAAM,oCAAoC,IAAI,CAAC,MAAM,CAAA,CAAE,CAAC;AAEjG,QAAA,IAAI;AACF,YAAA,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,EAAE,EAAE;AAC7C,gBAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iDAAiD,CAAC;gBACnE,OAAO,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,IAAI,CAAC;YAChD;AAEA,YAAA,MAAM,WAAW,GAAG;AAClB,gBAAA,KAAK,EAAE,oCAAoC;AAC3C,gBAAA,KAAK,EAAE,KAAK;AACZ,gBAAA,KAAK,EAAE,IAAI;AACX,gBAAA,SAAS,EAAE,SAAS;AACpB,gBAAA,gBAAgB,EAAE,IAAI;aACvB;AAED,YAAA,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,IAAI,CAAC,MAAM,EACX,WAAW,EACX;AACE,gBAAA,OAAO,EAAE;AACP,oBAAA,cAAc,EAAE,kBAAkB;AAClC,oBAAA,aAAa,EAAE,CAAA,OAAA,EAAU,IAAI,CAAC,MAAM,CAAA,CAAE;AACvC,iBAAA;AACF,aAAA,CACF;AAED,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iBAAiB,EAAE,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC;AAC1D,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,iBAAiB,EAAE,QAAQ,CAAC,IAAI,EAAE,KAAK,CAAC;AAE1D,YAAA,IAAI,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;gBACjD,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,KAAI;AAC1C,oBAAA,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK;AAC7B,oBAAA,MAAM,KAAK,GAAG,MAAM,CAAC,eAAe;oBACpC,IAAI,IAAI,GAAG,EAAE;;AAGb,oBAAA,IAAI,MAAM,CAAC,QAAQ,IAAI,IAAI,EAAE;AAC3B,wBAAA,MAAM,GAAG,GAAG,MAAM,CAAC,QAAQ;wBAC3B,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,MAAM,IAAI,GAAG,EAAE;AAC5C,4BAAA,IAAI,GAAG,GAAG,CAAC,IAAI;wBACjB;AAAO,6BAAA,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE;4BAClC,IAAI,GAAG,GAAG;wBACZ;oBACF;yBAAO;;AAEL,wBAAA,IAAI,GAAG,SAAS,CAAC,QAAQ,CAAC;oBAC5B;AAEA,oBAAA,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE;AACxB,gBAAA,CAAC,CAAC;YACJ;iBAAO;AACL,gBAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,kEAAkE,CACnE;gBACD,OAAO,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,IAAI,CAAC;YAChD;QACF;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,iBAAiB,CAAC,KAAK,CAAC,CAAC;;YAExE,OAAO,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,IAAI,CAAC;QAChD;IACF;AACD;AAEK,MAAO,cAAe,SAAQ,YAAY,CAAA;IAC9C,WAAA,CAAY,EACV,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,EACnC,MAAM,GAIP,EAAA;QACC,KAAK,CAAC,MAAM,CAAC;AACb,QAAA,IAAI,CAAC,MAAM,GAAG,MAAM;IACtB;IAEA,MAAM,MAAM,CACV,KAAa,EACb,SAAmB,EACnB,OAAe,CAAC,EAAA;QAEhB,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA,UAAA,EAAa,SAAS,CAAC,MAAM,CAAA,mBAAA,CAAqB,CAAC;AAErE,QAAA,IAAI;AACF,YAAA,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,IAAI,IAAI,CAAC,MAAM,KAAK,EAAE,EAAE;AAC7C,gBAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,mDAAmD,CAAC;gBACrE,OAAO,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,IAAI,CAAC;YAChD;AAEA,YAAA,MAAM,WAAW,GAAG;AAClB,gBAAA,KAAK,EAAE,aAAa;AACpB,gBAAA,KAAK,EAAE,KAAK;AACZ,gBAAA,KAAK,EAAE,IAAI;AACX,gBAAA,SAAS,EAAE,SAAS;aACrB;YAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,kCAAkC,EAClC,WAAW,EACX;AACE,gBAAA,OAAO,EAAE;AACP,oBAAA,cAAc,EAAE,kBAAkB;AAClC,oBAAA,aAAa,EAAE,CAAA,OAAA,EAAU,IAAI,CAAC,MAAM,CAAA,CAAE;AACvC,iBAAA;AACF,aAAA,CACF;AAED,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,gBAAgB,EAAE,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC;AACtD,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kBAAkB,EAAE,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC;AAE1D,YAAA,IAAI,QAAQ,CAAC,IAAI,IAAI,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;gBACjD,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,MAAM,KAAI;AAC1C,oBAAA,MAAM,QAAQ,GAAG,MAAM,CAAC,KAAK;AAC7B,oBAAA,MAAM,KAAK,GAAG,MAAM,CAAC,eAAe;AACpC,oBAAA,MAAM,IAAI,GAAG,SAAS,CAAC,QAAQ,CAAC;AAChC,oBAAA,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE;AACxB,gBAAA,CAAC,CAAC;YACJ;iBAAO;AACL,gBAAA,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,oEAAoE,CACrE;gBACD,OAAO,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,IAAI,CAAC;YAChD;QACF;QAAE,OAAO,KAAK,EAAE;AACd,YAAA,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,6BAA6B,EAC7B,iBAAiB,CAAC,KAAK,CAAC,CACzB;;YAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,IAAI,CAAC;QAChD;IACF;AACD;AAEK,MAAO,gBAAiB,SAAQ,YAAY,CAAA;AAChD,IAAA,WAAA,CAAY,MAAiB,EAAA;QAC3B,KAAK,CAAC,MAAM,CAAC;;IAEf;IAEA,MAAM,MAAM,CACV,KAAa,EACb,SAAmB,EACnB,OAAe,CAAC,EAAA;QAEhB,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,CAAA,UAAA,EAAa,SAAS,CAAC,MAAM,CAAA,mCAAA,CAAqC,CACnE;;QAED,OAAO,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,IAAI,CAAC;IAChD;AACD;AAED;;AAEG;AACI,MAAM,cAAc,GAAG,CAAC,MAM9B,KAA8B;AAC7B,IAAA,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,EAAE,GAAG,MAAM;;AAG7E,IAAA,MAAM,aAAa,GAAG,MAAM,IAAI,mBAAmB,EAAE;AAErD,IAAA,QAAQ,YAAY,CAAC,WAAW,EAAE;AAClC,QAAA,KAAK,MAAM;AACT,YAAA,OAAO,IAAI,YAAY,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;AAC5F,QAAA,KAAK,QAAQ;YACX,OAAO,IAAI,cAAc,CAAC;AACxB,gBAAA,MAAM,EAAE,YAAY;AACpB,gBAAA,MAAM,EAAE,aAAa;AACtB,aAAA,CAAC;AACJ,QAAA,KAAK,UAAU;AACb,YAAA,OAAO,IAAI,gBAAgB,CAAC,aAAa,CAAC;AAC5C,QAAA,KAAK,MAAM;AACT,YAAA,aAAa,CAAC,KAAK,CAAC,iDAAiD,CAAC;AACtE,YAAA,OAAO,SAAS;AAClB,QAAA;AACE,YAAA,aAAa,CAAC,IAAI,CAChB,0BAA0B,YAAY,CAAA,iCAAA,CAAmC,CAC1E;AACD,YAAA,OAAO,IAAI,YAAY,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC;;AAE9F;AAEA;AACA;AACA;AACA;;;;"}
+{"version":3,"file":"rerankers.mjs","names":[],"sources":["../../../../src/tools/search/rerankers.ts"],"sourcesContent":["import axios from 'axios';\nimport type * as t from './types';\nimport { createDefaultLogger, formatErrorForLog } from './utils';\n\nconst DEFAULT_JINA_API_URL = 'https://api.jina.ai/v1/rerank';\n\nconst getDefaultJinaApiUrl = (): string =>\n  process.env.JINA_API_URL != null && process.env.JINA_API_URL !== ''\n    ? process.env.JINA_API_URL\n    : DEFAULT_JINA_API_URL;\n\nexport abstract class BaseReranker {\n  protected apiKey: string | undefined;\n  protected logger: t.Logger;\n\n  constructor(logger?: t.Logger) {\n    // Each specific reranker will set its API key\n    this.logger = logger || createDefaultLogger();\n  }\n\n  abstract rerank(\n    query: string,\n    documents: string[],\n    topK?: number\n  ): Promise<t.Highlight[]>;\n\n  protected getDefaultRanking(\n    documents: string[],\n    topK: number\n  ): t.Highlight[] {\n    return documents\n      .slice(0, Math.min(topK, documents.length))\n      .map((doc) => ({ text: doc, score: 0 }));\n  }\n}\n\nexport class JinaReranker extends BaseReranker {\n  private apiUrl: string;\n\n  constructor({\n    apiKey = process.env.JINA_API_KEY,\n    apiUrl = getDefaultJinaApiUrl(),\n    logger,\n  }: {\n    apiKey?: string;\n    apiUrl?: string;\n    logger?: t.Logger;\n  }) {\n    super(logger);\n    this.apiKey = apiKey;\n    this.apiUrl = apiUrl;\n  }\n\n  async rerank(\n    query: string,\n    documents: string[],\n    topK: number = 5\n  ): Promise<t.Highlight[]> {\n    this.logger.debug(`Reranking ${documents.length} chunks with Jina using API URL: ${this.apiUrl}`);\n\n    try {\n      if (this.apiKey == null || this.apiKey === '') {\n        this.logger.warn('JINA_API_KEY is not set. Using default ranking.');\n        return this.getDefaultRanking(documents, topK);\n      }\n\n      const requestData = {\n        model: 'jina-reranker-v2-base-multilingual',\n        query: query,\n        top_n: topK,\n        documents: documents,\n        return_documents: true,\n      };\n\n      const response = await axios.post<t.JinaRerankerResponse | undefined>(\n        this.apiUrl,\n        requestData,\n        {\n          headers: {\n            'Content-Type': 'application/json',\n            Authorization: `Bearer ${this.apiKey}`,\n          },\n        }\n      );\n\n      this.logger.debug('Jina API Model:', response.data?.model);\n      this.logger.debug('Jina API Usage:', response.data?.usage);\n\n      if (response.data && response.data.results.length) {\n        return response.data.results.map((result) => {\n          const docIndex = result.index;\n          const score = result.relevance_score;\n          let text = '';\n\n          // If return_documents is true, the document field will be present\n          if (result.document != null) {\n            const doc = result.document;\n            if (typeof doc === 'object' && 'text' in doc) {\n              text = doc.text;\n            } else if (typeof doc === 'string') {\n              text = doc;\n            }\n          } else {\n            // Otherwise, use the index to get the document\n            text = documents[docIndex];\n          }\n\n          return { text, score };\n        });\n      } else {\n        this.logger.warn(\n          'Unexpected response format from Jina API. Using default ranking.'\n        );\n        return this.getDefaultRanking(documents, topK);\n      }\n    } catch (error) {\n      this.logger.error('Error using Jina reranker', formatErrorForLog(error));\n      // Fallback to default ranking on error\n      return this.getDefaultRanking(documents, topK);\n    }\n  }\n}\n\nexport class CohereReranker extends BaseReranker {\n  constructor({\n    apiKey = process.env.COHERE_API_KEY,\n    logger,\n  }: {\n    apiKey?: string;\n    logger?: t.Logger;\n  }) {\n    super(logger);\n    this.apiKey = apiKey;\n  }\n\n  async rerank(\n    query: string,\n    documents: string[],\n    topK: number = 5\n  ): Promise<t.Highlight[]> {\n    this.logger.debug(`Reranking ${documents.length} chunks with Cohere`);\n\n    try {\n      if (this.apiKey == null || this.apiKey === '') {\n        this.logger.warn('COHERE_API_KEY is not set. Using default ranking.');\n        return this.getDefaultRanking(documents, topK);\n      }\n\n      const requestData = {\n        model: 'rerank-v3.5',\n        query: query,\n        top_n: topK,\n        documents: documents,\n      };\n\n      const response = await axios.post<t.CohereRerankerResponse | undefined>(\n        'https://api.cohere.com/v2/rerank',\n        requestData,\n        {\n          headers: {\n            'Content-Type': 'application/json',\n            Authorization: `Bearer ${this.apiKey}`,\n          },\n        }\n      );\n\n      this.logger.debug('Cohere API ID:', response.data?.id);\n      this.logger.debug('Cohere API Meta:', response.data?.meta);\n\n      if (response.data && response.data.results.length) {\n        return response.data.results.map((result) => {\n          const docIndex = result.index;\n          const score = result.relevance_score;\n          const text = documents[docIndex];\n          return { text, score };\n        });\n      } else {\n        this.logger.warn(\n          'Unexpected response format from Cohere API. Using default ranking.'\n        );\n        return this.getDefaultRanking(documents, topK);\n      }\n    } catch (error) {\n      this.logger.error(\n        'Error using Cohere reranker',\n        formatErrorForLog(error)\n      );\n      // Fallback to default ranking on error\n      return this.getDefaultRanking(documents, topK);\n    }\n  }\n}\n\nexport class InfinityReranker extends BaseReranker {\n  constructor(logger?: t.Logger) {\n    super(logger);\n    // No API key needed for the placeholder implementation\n  }\n\n  async rerank(\n    query: string,\n    documents: string[],\n    topK: number = 5\n  ): Promise<t.Highlight[]> {\n    this.logger.debug(\n      `Reranking ${documents.length} chunks with Infinity (placeholder)`\n    );\n    // This would be replaced with actual Infinity reranker implementation\n    return this.getDefaultRanking(documents, topK);\n  }\n}\n\n/**\n * Creates the appropriate reranker based on type and configuration\n */\nexport const createReranker = (config: {\n  rerankerType: t.RerankerType;\n  jinaApiKey?: string;\n  jinaApiUrl?: string;\n  cohereApiKey?: string;\n  logger?: t.Logger;\n}): BaseReranker | undefined => {\n  const { rerankerType, jinaApiKey, jinaApiUrl, cohereApiKey, logger } = config;\n\n  // Create a default logger if none is provided\n  const defaultLogger = logger || createDefaultLogger();\n\n  switch (rerankerType.toLowerCase()) {\n  case 'jina':\n    return new JinaReranker({ apiKey: jinaApiKey, apiUrl: jinaApiUrl, logger: defaultLogger });\n  case 'cohere':\n    return new CohereReranker({\n      apiKey: cohereApiKey,\n      logger: defaultLogger,\n    });\n  case 'infinity':\n    return new InfinityReranker(defaultLogger);\n  case 'none':\n    defaultLogger.debug('Skipping reranking as reranker is set to \"none\"');\n    return undefined;\n  default:\n    defaultLogger.warn(\n      `Unknown reranker type: ${rerankerType}. Defaulting to InfinityReranker.`\n    );\n    return new JinaReranker({ apiKey: jinaApiKey, apiUrl: jinaApiUrl, logger: defaultLogger });\n  }\n};\n\n// Example usage:\n// const jinaReranker = new JinaReranker();\n// const cohereReranker = new CohereReranker();\n// const infinityReranker = new InfinityReranker();\n"],"mappings":";;;AAIA,MAAM,uBAAuB;AAE7B,MAAM,6BACJ,QAAQ,IAAI,gBAAgB,QAAQ,QAAQ,IAAI,iBAAiB,KAC7D,QAAQ,IAAI,eACZ;AAEN,IAAsB,eAAtB,MAAmC;CACjC;CACA;CAEA,YAAY,QAAmB;EAE7B,KAAK,SAAS,UAAU,oBAAoB;CAC9C;CAQA,kBACE,WACA,MACe;EACf,OAAO,UACJ,MAAM,GAAG,KAAK,IAAI,MAAM,UAAU,MAAM,CAAC,CAAC,CAC1C,KAAK,SAAS;GAAE,MAAM;GAAK,OAAO;EAAE,EAAE;CAC3C;AACF;AAEA,IAAa,eAAb,cAAkC,aAAa;CAC7C;CAEA,YAAY,EACV,SAAS,QAAQ,IAAI,cACrB,SAAS,qBAAqB,GAC9B,UAKC;EACD,MAAM,MAAM;EACZ,KAAK,SAAS;EACd,KAAK,SAAS;CAChB;CAEA,MAAM,OACJ,OACA,WACA,OAAe,GACS;EACxB,KAAK,OAAO,MAAM,aAAa,UAAU,OAAO,mCAAmC,KAAK,QAAQ;EAEhG,IAAI;GACF,IAAI,KAAK,UAAU,QAAQ,KAAK,WAAW,IAAI;IAC7C,KAAK,OAAO,KAAK,iDAAiD;IAClE,OAAO,KAAK,kBAAkB,WAAW,IAAI;GAC/C;GAEA,MAAM,cAAc;IAClB,OAAO;IACA;IACP,OAAO;IACI;IACX,kBAAkB;GACpB;GAEA,MAAM,WAAW,MAAM,MAAM,KAC3B,KAAK,QACL,aACA,EACE,SAAS;IACP,gBAAgB;IAChB,eAAe,UAAU,KAAK;GAChC,EACF,CACF;GAEA,KAAK,OAAO,MAAM,mBAAmB,SAAS,MAAM,KAAK;GACzD,KAAK,OAAO,MAAM,mBAAmB,SAAS,MAAM,KAAK;GAEzD,IAAI,SAAS,QAAQ,SAAS,KAAK,QAAQ,QACzC,OAAO,SAAS,KAAK,QAAQ,KAAK,WAAW;IAC3C,MAAM,WAAW,OAAO;IACxB,MAAM,QAAQ,OAAO;IACrB,IAAI,OAAO;IAGX,IAAI,OAAO,YAAY,MAAM;KAC3B,MAAM,MAAM,OAAO;KACnB,IAAI,OAAO,QAAQ,YAAY,UAAU,KACvC,OAAO,IAAI;UACN,IAAI,OAAO,QAAQ,UACxB,OAAO;IAEX,OAEE,OAAO,UAAU;IAGnB,OAAO;KAAE;KAAM;IAAM;GACvB,CAAC;QACI;IACL,KAAK,OAAO,KACV,kEACF;IACA,OAAO,KAAK,kBAAkB,WAAW,IAAI;GAC/C;EACF,SAAS,OAAO;GACd,KAAK,OAAO,MAAM,6BAA6B,kBAAkB,KAAK,CAAC;GAEvE,OAAO,KAAK,kBAAkB,WAAW,IAAI;EAC/C;CACF;AACF;AAEA,IAAa,iBAAb,cAAoC,aAAa;CAC/C,YAAY,EACV,SAAS,QAAQ,IAAI,gBACrB,UAIC;EACD,MAAM,MAAM;EACZ,KAAK,SAAS;CAChB;CAEA,MAAM,OACJ,OACA,WACA,OAAe,GACS;EACxB,KAAK,OAAO,MAAM,aAAa,UAAU,OAAO,oBAAoB;EAEpE,IAAI;GACF,IAAI,KAAK,UAAU,QAAQ,KAAK,WAAW,IAAI;IAC7C,KAAK,OAAO,KAAK,mDAAmD;IACpE,OAAO,KAAK,kBAAkB,WAAW,IAAI;GAC/C;GAEA,MAAM,cAAc;IAClB,OAAO;IACA;IACP,OAAO;IACI;GACb;GAEA,MAAM,WAAW,MAAM,MAAM,KAC3B,oCACA,aACA,EACE,SAAS;IACP,gBAAgB;IAChB,eAAe,UAAU,KAAK;GAChC,EACF,CACF;GAEA,KAAK,OAAO,MAAM,kBAAkB,SAAS,MAAM,EAAE;GACrD,KAAK,OAAO,MAAM,oBAAoB,SAAS,MAAM,IAAI;GAEzD,IAAI,SAAS,QAAQ,SAAS,KAAK,QAAQ,QACzC,OAAO,SAAS,KAAK,QAAQ,KAAK,WAAW;IAC3C,MAAM,WAAW,OAAO;IACxB,MAAM,QAAQ,OAAO;IAErB,OAAO;KAAE,MADI,UAAU;KACR;IAAM;GACvB,CAAC;QACI;IACL,KAAK,OAAO,KACV,oEACF;IACA,OAAO,KAAK,kBAAkB,WAAW,IAAI;GAC/C;EACF,SAAS,OAAO;GACd,KAAK,OAAO,MACV,+BACA,kBAAkB,KAAK,CACzB;GAEA,OAAO,KAAK,kBAAkB,WAAW,IAAI;EAC/C;CACF;AACF;AAEA,IAAa,mBAAb,cAAsC,aAAa;CACjD,YAAY,QAAmB;EAC7B,MAAM,MAAM;CAEd;CAEA,MAAM,OACJ,OACA,WACA,OAAe,GACS;EACxB,KAAK,OAAO,MACV,aAAa,UAAU,OAAO,oCAChC;EAEA,OAAO,KAAK,kBAAkB,WAAW,IAAI;CAC/C;AACF;;;;AAKA,MAAa,kBAAkB,WAMC;CAC9B,MAAM,EAAE,cAAc,YAAY,YAAY,cAAc,WAAW;CAGvE,MAAM,gBAAgB,UAAU,oBAAoB;CAEpD,QAAQ,aAAa,YAAY,GAAjC;EACA,KAAK,QACH,OAAO,IAAI,aAAa;GAAE,QAAQ;GAAY,QAAQ;GAAY,QAAQ;EAAc,CAAC;EAC3F,KAAK,UACH,OAAO,IAAI,eAAe;GACxB,QAAQ;GACR,QAAQ;EACV,CAAC;EACH,KAAK,YACH,OAAO,IAAI,iBAAiB,aAAa;EAC3C,KAAK;GACH,cAAc,MAAM,mDAAiD;GACrE;EACF;GACE,cAAc,KACZ,0BAA0B,aAAa,kCACzC;GACA,OAAO,IAAI,aAAa;IAAE,QAAQ;IAAY,QAAQ;IAAY,QAAQ;GAAc,CAAC;CAC3F;AACF"}

package/dist/esm/tools/search/schema.mjs

@@ -1,11 +1,12 @@
-var DATE_RANGE;
-(function (DATE_RANGE) {
-    DATE_RANGE["PAST_HOUR"] = "h";
-    DATE_RANGE["PAST_24_HOURS"] = "d";
-    DATE_RANGE["PAST_WEEK"] = "w";
-    DATE_RANGE["PAST_MONTH"] = "m";
-    DATE_RANGE["PAST_YEAR"] = "y";
-})(DATE_RANGE || (DATE_RANGE = {}));
+//#region src/tools/search/schema.ts
+let DATE_RANGE = /* @__PURE__ */ function(DATE_RANGE) {
+	DATE_RANGE["PAST_HOUR"] = "h";
+	DATE_RANGE["PAST_24_HOURS"] = "d";
+	DATE_RANGE["PAST_WEEK"] = "w";
+	DATE_RANGE["PAST_MONTH"] = "m";
+	DATE_RANGE["PAST_YEAR"] = "y";
+	return DATE_RANGE;
+}({});
 const DEFAULT_QUERY_DESCRIPTION = `
 GUIDELINES:
 - Start broad, then narrow: Begin with key concepts, then refine with specifics
@@ -34,44 +35,44 @@ Examples:
 - "in" for India
 `.trim();
 const querySchema = {
-    type: 'string',
-    description: DEFAULT_QUERY_DESCRIPTION,
+	type: "string",
+	description: DEFAULT_QUERY_DESCRIPTION
 };
 const dateSchema = {
-    type: 'string',
-    enum: Object.values(DATE_RANGE),
-    description: 'Date range for search results.',
+	type: "string",
+	enum: Object.values(DATE_RANGE),
+	description: "Date range for search results."
 };
 const countrySchema = {
-    type: 'string',
-    description: DEFAULT_COUNTRY_DESCRIPTION,
+	type: "string",
+	description: DEFAULT_COUNTRY_DESCRIPTION
 };
 const imagesSchema = {
-    type: 'boolean',
-    description: 'Whether to also run an image search.',
+	type: "boolean",
+	description: "Whether to also run an image search."
 };
 const videosSchema = {
-    type: 'boolean',
-    description: 'Whether to also run a video search.',
+	type: "boolean",
+	description: "Whether to also run a video search."
 };
 const newsSchema = {
-    type: 'boolean',
-    description: 'Whether to also run a news search.',
+	type: "boolean",
+	description: "Whether to also run a news search."
 };
 /** Combined web search tool schema with all properties */
 const WebSearchToolSchema = {
-    type: 'object',
-    properties: {
-        query: querySchema,
-        date: dateSchema,
-        country: countrySchema,
-        images: imagesSchema,
-        videos: videosSchema,
-        news: newsSchema,
-    },
-    required: ['query'],
+	type: "object",
+	properties: {
+		query: querySchema,
+		date: dateSchema,
+		country: countrySchema,
+		images: imagesSchema,
+		videos: videosSchema,
+		news: newsSchema
+	},
+	required: ["query"]
 };
-const WebSearchToolName = 'web_search';
+const WebSearchToolName = "web_search";
 const WebSearchToolDescription = `Real-time search. Results have required citation anchors.
 
 Note: Use ONCE per reply unless instructed otherwise.
@@ -95,10 +96,11 @@ Use anchor marker(s) immediately after the statement:
 
 **NEVER use markdown links, [1], or footnotes. CITE ONLY with anchors provided.**`;
 const WebSearchToolDefinition = {
-    name: WebSearchToolName,
-    description: WebSearchToolDescription,
-    schema: WebSearchToolSchema,
+	name: WebSearchToolName,
+	description: WebSearchToolDescription,
+	schema: WebSearchToolSchema
 };
-
+//#endregion
 export { DATE_RANGE, DEFAULT_COUNTRY_DESCRIPTION, DEFAULT_QUERY_DESCRIPTION, WebSearchToolDefinition, WebSearchToolDescription, WebSearchToolName, WebSearchToolSchema, countrySchema, dateSchema, imagesSchema, newsSchema, querySchema, videosSchema };
-//# sourceMappingURL=schema.mjs.map
+
+//# sourceMappingURL=schema.mjs.map

package/dist/esm/tools/search/schema.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"schema.mjs","sources":["../../../../src/tools/search/schema.ts"],"sourcesContent":["export enum DATE_RANGE {\n  PAST_HOUR = 'h',\n  PAST_24_HOURS = 'd',\n  PAST_WEEK = 'w',\n  PAST_MONTH = 'm',\n  PAST_YEAR = 'y',\n}\n\nexport const DEFAULT_QUERY_DESCRIPTION = `\nGUIDELINES:\n- Start broad, then narrow: Begin with key concepts, then refine with specifics\n- Think like sources: Use terminology experts would use in the field\n- Consider perspective: Frame queries from different viewpoints for better results\n- Quality over quantity: A precise 3-4 word query often beats lengthy sentences\n\nTECHNIQUES (combine for power searches):\n- EXACT PHRASES: Use quotes (\"climate change report\")\n- EXCLUDE TERMS: Use minus to remove unwanted results (-wikipedia)\n- SITE-SPECIFIC: Restrict to websites (site:edu research)\n- FILETYPE: Find specific documents (filetype:pdf study)\n- OR OPERATOR: Find alternatives (electric OR hybrid cars)\n- DATE RANGE: Recent information (data after:2020)\n- WILDCARDS: Use * for unknown terms (how to * bread)\n- SPECIFIC QUESTIONS: Use who/what/when/where/why/how\n- DOMAIN TERMS: Include technical terminology for specialized topics\n- CONCISE TERMS: Prioritize keywords over sentences\n`.trim();\n\nexport const DEFAULT_COUNTRY_DESCRIPTION =\n  `Country code to localize search results.\nUse standard 2-letter country codes: \"us\", \"uk\", \"ca\", \"de\", \"fr\", \"jp\", \"br\", etc.\nProvide this when the search should return results specific to a particular country.\nExamples:\n- \"us\" for United States (default)\n- \"de\" for Germany\n- \"in\" for India\n`.trim();\n\nexport const querySchema = {\n  type: 'string',\n  description: DEFAULT_QUERY_DESCRIPTION,\n} as const;\n\nexport const dateSchema = {\n  type: 'string',\n  enum: Object.values(DATE_RANGE),\n  description: 'Date range for search results.',\n} as const;\n\nexport const countrySchema = {\n  type: 'string',\n  description: DEFAULT_COUNTRY_DESCRIPTION,\n} as const;\n\nexport const imagesSchema = {\n  type: 'boolean',\n  description: 'Whether to also run an image search.',\n} as const;\n\nexport const videosSchema = {\n  type: 'boolean',\n  description: 'Whether to also run a video search.',\n} as const;\n\nexport const newsSchema = {\n  type: 'boolean',\n  description: 'Whether to also run a news search.',\n} as const;\n\n/** Combined web search tool schema with all properties */\nexport const WebSearchToolSchema = {\n  type: 'object',\n  properties: {\n    query: querySchema,\n    date: dateSchema,\n    country: countrySchema,\n    images: imagesSchema,\n    videos: videosSchema,\n    news: newsSchema,\n  },\n  required: ['query'],\n} as const;\n\nexport const WebSearchToolName = 'web_search';\n\nexport const WebSearchToolDescription = `Real-time search. Results have required citation anchors.\n\nNote: Use ONCE per reply unless instructed otherwise.\n\nAnchors:\n- \\\\ue202turnXtypeY\n- X = turn idx, type = 'search' | 'news' | 'image' | 'ref', Y = item idx\n\nSpecial Markers:\n- \\\\ue203...\\\\ue204 — highlight start/end of cited text (for Standalone or Group citations)\n- \\\\ue200...\\\\ue201 — group block (e.g. \\\\ue200\\\\ue202turn0search1\\\\ue202turn0news2\\\\ue201)\n\n**CITE EVERY NON-OBVIOUS FACT/QUOTE:**\nUse anchor marker(s) immediately after the statement:\n- Standalone: \"Pure functions produce same output. \\\\ue202turn0search0\"\n- Standalone (multiple): \"Today's News \\\\ue202turn0search0\\\\ue202turn0news0\"\n- Highlight: \"\\\\ue203Highlight text.\\\\ue204\\\\ue202turn0news1\"\n- Group: \"Sources. \\\\ue200\\\\ue202turn0search0\\\\ue202turn0news1\\\\ue201\"\n- Group Highlight: \"\\\\ue203Highlight for group.\\\\ue204 \\\\ue200\\\\ue202turn0search0\\\\ue202turn0news1\\\\ue201\"\n- Image: \"See photo \\\\ue202turn0image0.\"\n\n**NEVER use markdown links, [1], or footnotes. CITE ONLY with anchors provided.**`;\n\nexport const WebSearchToolDefinition = {\n  name: WebSearchToolName,\n  description: WebSearchToolDescription,\n  schema: WebSearchToolSchema,\n} as const;\n"],"names":[],"mappings":"IAAY;AAAZ,CAAA,UAAY,UAAU,EAAA;AACpB,IAAA,UAAA,CAAA,WAAA,CAAA,GAAA,GAAe;AACf,IAAA,UAAA,CAAA,eAAA,CAAA,GAAA,GAAmB;AACnB,IAAA,UAAA,CAAA,WAAA,CAAA,GAAA,GAAe;AACf,IAAA,UAAA,CAAA,YAAA,CAAA,GAAA,GAAgB;AAChB,IAAA,UAAA,CAAA,WAAA,CAAA,GAAA,GAAe;AACjB,CAAC,EANW,UAAU,KAAV,UAAU,GAAA,EAAA,CAAA,CAAA;AAQf,MAAM,yBAAyB,GAAG;;;;;;;;;;;;;;;;;;CAkBxC,CAAC,IAAI;AAEC,MAAM,2BAA2B,GACtC,CAAA;;;;;;;CAOD,CAAC,IAAI;AAEC,MAAM,WAAW,GAAG;AACzB,IAAA,IAAI,EAAE,QAAQ;AACd,IAAA,WAAW,EAAE,yBAAyB;;AAGjC,MAAM,UAAU,GAAG;AACxB,IAAA,IAAI,EAAE,QAAQ;AACd,IAAA,IAAI,EAAE,MAAM,CAAC,MAAM,CAAC,UAAU,CAAC;AAC/B,IAAA,WAAW,EAAE,gCAAgC;;AAGxC,MAAM,aAAa,GAAG;AAC3B,IAAA,IAAI,EAAE,QAAQ;AACd,IAAA,WAAW,EAAE,2BAA2B;;AAGnC,MAAM,YAAY,GAAG;AAC1B,IAAA,IAAI,EAAE,SAAS;AACf,IAAA,WAAW,EAAE,sCAAsC;;AAG9C,MAAM,YAAY,GAAG;AAC1B,IAAA,IAAI,EAAE,SAAS;AACf,IAAA,WAAW,EAAE,qCAAqC;;AAG7C,MAAM,UAAU,GAAG;AACxB,IAAA,IAAI,EAAE,SAAS;AACf,IAAA,WAAW,EAAE,oCAAoC;;AAGnD;AACO,MAAM,mBAAmB,GAAG;AACjC,IAAA,IAAI,EAAE,QAAQ;AACd,IAAA,UAAU,EAAE;AACV,QAAA,KAAK,EAAE,WAAW;AAClB,QAAA,IAAI,EAAE,UAAU;AAChB,QAAA,OAAO,EAAE,aAAa;AACtB,QAAA,MAAM,EAAE,YAAY;AACpB,QAAA,MAAM,EAAE,YAAY;AACpB,QAAA,IAAI,EAAE,UAAU;AACjB,KAAA;IACD,QAAQ,EAAE,CAAC,OAAO,CAAC;;AAGd,MAAM,iBAAiB,GAAG;AAE1B,MAAM,wBAAwB,GAAG,CAAA;;;;;;;;;;;;;;;;;;;;;;AAuBjC,MAAM,uBAAuB,GAAG;AACrC,IAAA,IAAI,EAAE,iBAAiB;AACvB,IAAA,WAAW,EAAE,wBAAwB;AACrC,IAAA,MAAM,EAAE,mBAAmB;;;;;"}
+{"version":3,"file":"schema.mjs","names":[],"sources":["../../../../src/tools/search/schema.ts"],"sourcesContent":["export enum DATE_RANGE {\n  PAST_HOUR = 'h',\n  PAST_24_HOURS = 'd',\n  PAST_WEEK = 'w',\n  PAST_MONTH = 'm',\n  PAST_YEAR = 'y',\n}\n\nexport const DEFAULT_QUERY_DESCRIPTION = `\nGUIDELINES:\n- Start broad, then narrow: Begin with key concepts, then refine with specifics\n- Think like sources: Use terminology experts would use in the field\n- Consider perspective: Frame queries from different viewpoints for better results\n- Quality over quantity: A precise 3-4 word query often beats lengthy sentences\n\nTECHNIQUES (combine for power searches):\n- EXACT PHRASES: Use quotes (\"climate change report\")\n- EXCLUDE TERMS: Use minus to remove unwanted results (-wikipedia)\n- SITE-SPECIFIC: Restrict to websites (site:edu research)\n- FILETYPE: Find specific documents (filetype:pdf study)\n- OR OPERATOR: Find alternatives (electric OR hybrid cars)\n- DATE RANGE: Recent information (data after:2020)\n- WILDCARDS: Use * for unknown terms (how to * bread)\n- SPECIFIC QUESTIONS: Use who/what/when/where/why/how\n- DOMAIN TERMS: Include technical terminology for specialized topics\n- CONCISE TERMS: Prioritize keywords over sentences\n`.trim();\n\nexport const DEFAULT_COUNTRY_DESCRIPTION =\n  `Country code to localize search results.\nUse standard 2-letter country codes: \"us\", \"uk\", \"ca\", \"de\", \"fr\", \"jp\", \"br\", etc.\nProvide this when the search should return results specific to a particular country.\nExamples:\n- \"us\" for United States (default)\n- \"de\" for Germany\n- \"in\" for India\n`.trim();\n\nexport const querySchema = {\n  type: 'string',\n  description: DEFAULT_QUERY_DESCRIPTION,\n} as const;\n\nexport const dateSchema = {\n  type: 'string',\n  enum: Object.values(DATE_RANGE),\n  description: 'Date range for search results.',\n} as const;\n\nexport const countrySchema = {\n  type: 'string',\n  description: DEFAULT_COUNTRY_DESCRIPTION,\n} as const;\n\nexport const imagesSchema = {\n  type: 'boolean',\n  description: 'Whether to also run an image search.',\n} as const;\n\nexport const videosSchema = {\n  type: 'boolean',\n  description: 'Whether to also run a video search.',\n} as const;\n\nexport const newsSchema = {\n  type: 'boolean',\n  description: 'Whether to also run a news search.',\n} as const;\n\n/** Combined web search tool schema with all properties */\nexport const WebSearchToolSchema = {\n  type: 'object',\n  properties: {\n    query: querySchema,\n    date: dateSchema,\n    country: countrySchema,\n    images: imagesSchema,\n    videos: videosSchema,\n    news: newsSchema,\n  },\n  required: ['query'],\n} as const;\n\nexport const WebSearchToolName = 'web_search';\n\nexport const WebSearchToolDescription = `Real-time search. Results have required citation anchors.\n\nNote: Use ONCE per reply unless instructed otherwise.\n\nAnchors:\n- \\\\ue202turnXtypeY\n- X = turn idx, type = 'search' | 'news' | 'image' | 'ref', Y = item idx\n\nSpecial Markers:\n- \\\\ue203...\\\\ue204 — highlight start/end of cited text (for Standalone or Group citations)\n- \\\\ue200...\\\\ue201 — group block (e.g. \\\\ue200\\\\ue202turn0search1\\\\ue202turn0news2\\\\ue201)\n\n**CITE EVERY NON-OBVIOUS FACT/QUOTE:**\nUse anchor marker(s) immediately after the statement:\n- Standalone: \"Pure functions produce same output. \\\\ue202turn0search0\"\n- Standalone (multiple): \"Today's News \\\\ue202turn0search0\\\\ue202turn0news0\"\n- Highlight: \"\\\\ue203Highlight text.\\\\ue204\\\\ue202turn0news1\"\n- Group: \"Sources. \\\\ue200\\\\ue202turn0search0\\\\ue202turn0news1\\\\ue201\"\n- Group Highlight: \"\\\\ue203Highlight for group.\\\\ue204 \\\\ue200\\\\ue202turn0search0\\\\ue202turn0news1\\\\ue201\"\n- Image: \"See photo \\\\ue202turn0image0.\"\n\n**NEVER use markdown links, [1], or footnotes. CITE ONLY with anchors provided.**`;\n\nexport const WebSearchToolDefinition = {\n  name: WebSearchToolName,\n  description: WebSearchToolDescription,\n  schema: WebSearchToolSchema,\n} as const;\n"],"mappings":";AAAA,IAAY,aAAL,yBAAA,YAAA;CACL,WAAA,eAAA;CACA,WAAA,mBAAA;CACA,WAAA,eAAA;CACA,WAAA,gBAAA;CACA,WAAA,eAAA;;AACF,EAAA,CAAA,CAAA;AAEA,MAAa,4BAA4B;;;;;;;;;;;;;;;;;;EAkBvC,KAAK;AAEP,MAAa,8BACX;;;;;;;EAOA,KAAK;AAEP,MAAa,cAAc;CACzB,MAAM;CACN,aAAa;AACf;AAEA,MAAa,aAAa;CACxB,MAAM;CACN,MAAM,OAAO,OAAO,UAAU;CAC9B,aAAa;AACf;AAEA,MAAa,gBAAgB;CAC3B,MAAM;CACN,aAAa;AACf;AAEA,MAAa,eAAe;CAC1B,MAAM;CACN,aAAa;AACf;AAEA,MAAa,eAAe;CAC1B,MAAM;CACN,aAAa;AACf;AAEA,MAAa,aAAa;CACxB,MAAM;CACN,aAAa;AACf;;AAGA,MAAa,sBAAsB;CACjC,MAAM;CACN,YAAY;EACV,OAAO;EACP,MAAM;EACN,SAAS;EACT,QAAQ;EACR,QAAQ;EACR,MAAM;CACR;CACA,UAAU,CAAC,OAAO;AACpB;AAEA,MAAa,oBAAoB;AAEjC,MAAa,2BAA2B;;;;;;;;;;;;;;;;;;;;;;AAuBxC,MAAa,0BAA0B;CACrC,MAAM;CACN,aAAa;CACb,QAAQ;AACV"}