@graphext/cuery 0.4.0 → 0.5.2

package/script/src/tools/sentiment.js

@@ -1,11 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.extractAspectBasedSentiments = extractAspectBasedSentiments;
-exports.extractABSForBrandBatch = extractABSForBrandBatch;
-const async_js_1 = require("../async.js");
-const openai_js_1 = require("../openai.js");
+exports.ABSentimentsSchema = exports.ABSentimentSchema = exports.SentimentExtractor = void 0;
+const tool_js_1 = require("../tool.js");
 const sentiment_schema_js_1 = require("../schemas/sentiment.schema.js");
-const utils_js_1 = require("../utils.js");
+const utils_js_1 = require("../helpers/utils.js");
 const ABS_PROMPT_SYSTEM = (0, utils_js_1.dedent)(`
 You're an expert in Aspect-Based Sentiment Analysis. Your task involves identifying specific
 entities mentioned in a text (e.g. a person, product, service, or experience) and determining the
@@ -41,32 +39,41 @@ Return the entities and their sentiments with reasons from the following text se
 {text}
 `);
 /**
- * Extracts aspect-based sentiments from a text.
+ * A tool that extracts aspect-based sentiments from text.
  */
-async function extractAspectBasedSentiments(text, instructions = '', model = 'gpt-4.1-mini') {
-    if (text == null || text.trim() === '') {
-        return [];
+class SentimentExtractor extends tool_js_1.Tool {
+    systemPrompt;
+    constructor(config = {}, modelConfig) {
+        super(modelConfig);
+        const { instructions = '', brand = null } = config;
+        const brandInstructions = brand
+            ? (0, utils_js_1.dedent)(`
+				When analyzing the text, pay special attention to any mentions of the brand "${brand.shortName}"
+				or its products/services (${brand.portfolio}). Ensure that any sentiments expressed toward this
+				brand or its offerings are accurately captured in your output. Respond in language code ${brand.language}.
+			`)
+            : '';
+        const combinedInstructions = [instructions, brandInstructions].filter(Boolean).join('\n\n');
+        this.systemPrompt = ABS_PROMPT_SYSTEM.replace('{instructions}', combinedInstructions);
     }
-    const promptSystem = ABS_PROMPT_SYSTEM.replace('{instructions}', instructions);
-    const promptUser = ABS_PROMPT_USER.replace('{text}', text);
-    const conversation = [
-        { role: 'system', content: promptSystem },
-        { role: 'user', content: promptUser }
-    ];
-    const { parsed } = await (0, openai_js_1.askOpenAISafe)(conversation, model, sentiment_schema_js_1.ABSentimentsSchema);
-    if (!parsed) {
-        throw new Error('Failed to parse response from OpenAI');
+    schema() {
+        return sentiment_schema_js_1.ABSentimentsSchema;
+    }
+    prompt(text) {
+        const userPrompt = ABS_PROMPT_USER.replace('{text}', text ?? '');
+        return [
+            { role: 'system', content: this.systemPrompt },
+            { role: 'user', content: userPrompt }
+        ];
+    }
+    isEmpty(text) {
+        return text == null || text.trim() === '';
+    }
+    extractResult(parsed) {
+        return parsed.aspects;
     }
-    return parsed.aspects;
-}
-/**
- * Classifies multiple data records concurrently while preserving order.
- */
-async function extractABSForBrandBatch(texts, brand = null, model = 'gpt-4.1-mini', maxConcurrency = 100) {
-    const instructions = brand ? (0, utils_js_1.dedent)(`
-    When analyzing the text, pay special attention to any mentions of the brand "${brand.shortName}"
-    or its products/services (${brand.portfolio}). Ensure that any sentiments expressed toward this
-    brand or its offerings are accurately captured in your output. Respond in language code ${brand.language}.
-    `) : '';
-    return (0, async_js_1.mapParallel)(texts, maxConcurrency, (text) => extractAspectBasedSentiments(text, instructions, model));
 }
+exports.SentimentExtractor = SentimentExtractor;
+var sentiment_schema_js_2 = require("../schemas/sentiment.schema.js");
+Object.defineProperty(exports, "ABSentimentSchema", { enumerable: true, get: function () { return sentiment_schema_js_2.ABSentimentSchema; } });
+Object.defineProperty(exports, "ABSentimentsSchema", { enumerable: true, get: function () { return sentiment_schema_js_2.ABSentimentsSchema; } });

package/script/src/tools/sources.d.ts

@@ -1,7 +1,7 @@
-import { type FlaggedBrand } from '../schemas/brand.schema.js';
+import type { FlaggedBrand } from '../schemas/brand.schema.js';
 import type { CategorizedSource, EnrichedSource, Source } from '../schemas/sources.schema.js';
 import { type TopicLabel } from './topics.js';
-import type { AIParams } from '../openai.js';
+import type { ProviderParams } from '../llm.js';
 import type { Entity } from './entities.js';
 export type { Source, EnrichedSource, CategorizedSource } from '../schemas/sources.schema.js';
 /**
@@ -27,13 +27,13 @@ export declare function collectURLs(sourceLists: Array<Array<Source>>, domains?:
 /**
  * Classifies an array of URLs into categories based on WEB_TAXONOMY.
  */
-export declare function classifyURLs(urls: Array<string>, model?: string, modelParams?: AIParams, maxConcurrency?: number): Promise<Record<string, TopicLabel | null>>;
+export declare function classifyURLs(urls: Array<string>, model?: string, modelParams?: ProviderParams, maxConcurrency?: number): Promise<Record<string, TopicLabel | null>>;
 /**
  * Creates a category mapper for sources by collecting unique URLs/domains and classifying them.
  */
-export declare function makeCategoryMapper(sourceLists: Array<Array<Source>>, domains?: boolean, model?: string, modelParams?: AIParams, maxConcurrency?: number): Promise<Record<string, TopicLabel | null>>;
+export declare function makeCategoryMapper(sourceLists: Array<Array<Source>>, domains?: boolean, model?: string, modelParams?: ProviderParams, maxConcurrency?: number): Promise<Record<string, TopicLabel | null>>;
 /**
  * Categorizes sources by assigning topic and subtopic to each EnrichedSource in-place(!).
  */
-export declare function categorizeSources(sourceLists: Array<Array<EnrichedSource>>, domains?: boolean, model?: string, modelParams?: AIParams, maxConcurrency?: number): Promise<Array<Array<CategorizedSource>>>;
+export declare function categorizeSources(sourceLists: Array<Array<EnrichedSource>>, domains?: boolean, model?: string, modelParams?: ProviderParams, maxConcurrency?: number): Promise<Array<Array<CategorizedSource>>>;
 //# sourceMappingURL=sources.d.ts.map

package/script/src/tools/sources.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"sources.d.ts","sourceRoot":"","sources":["../../../src/src/tools/sources.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,KAAK,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,8BAA8B,CAAC;AAC9F,OAAO,EAA4C,KAAK,UAAU,EAAE,MAAM,aAAa,CAAC;AACxF,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AAC7C,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAE5C,YAAY,EAAE,MAAM,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAE9F;;GAEG;AACH,wBAAgB,YAAY,CAC3B,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,KAAK,CAAC,YAAY,CAAC,EAC3B,QAAQ,GAAE,KAAK,CAAC,MAAM,CAAM,GAC1B,cAAc,CAoEhB;AAED,wBAAsB,aAAa,CAClC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,EACpC,MAAM,EAAE,KAAK,CAAC,YAAY,CAAC,EAC3B,QAAQ,GAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAM,GACjC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAIvC;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CACtC,OAAO,EAAE,KAAK,CAAC,cAAc,CAAC,GAC5B;IAAE,eAAe,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAAC,YAAY,EAAE,KAAK,CAAC,MAAM,CAAC,CAAA;CAAE,CA8DjE;AAED,wBAAgB,qBAAqB,CACpC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,GACnC;IAAE,eAAe,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAAC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAA;CAAE,CAc/E;AAED;;GAEG;AACH,wBAAgB,WAAW,CAC1B,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EACjC,OAAO,GAAE,OAAe,GACtB,KAAK,CAAC,MAAM,CAAC,CAgBf;AAiED;;GAEG;AACH,wBAAsB,YAAY,CACjC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,EACnB,KAAK,GAAE,MAAkB,EACzB,WAAW,GAAE,QAA4C,EACzD,cAAc,GAAE,MAAY,GAC1B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI,CAAC,CAAC,CAa5C;AAED;;GAEG;AACH,wBAAsB,kBAAkB,CACvC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EACjC,OAAO,GAAE,OAAe,EACxB,KAAK,GAAE,MAAkB,EACzB,WAAW,GAAE,QAA4C,EACzD,cAAc,GAAE,MAAY,GAC1B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI,CAAC,CAAC,CAI5C;AAED;;GAEG;AACH,wBAAsB,iBAAiB,CACtC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,EACzC,OAAO,GAAE,OAAe,EACxB,KAAK,GAAE,MAAkB,EACzB,WAAW,GAAE,QAA4C,EACzD,cAAc,GAAE,MAAY,GAC1B,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAiB1C"}
+{"version":3,"file":"sources.d.ts","sourceRoot":"","sources":["../../../src/src/tools/sources.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC/D,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,EAAE,MAAM,8BAA8B,CAAC;AAC9F,OAAO,EAA2B,KAAK,UAAU,EAAE,MAAM,aAAa,CAAC;AACvE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAChD,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAE5C,YAAY,EAAE,MAAM,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AAE9F;;GAEG;AACH,wBAAgB,YAAY,CAC3B,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,KAAK,CAAC,YAAY,CAAC,EAC3B,QAAQ,GAAE,KAAK,CAAC,MAAM,CAAM,GAC1B,cAAc,CAoEhB;AAED,wBAAsB,aAAa,CAClC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,EACpC,MAAM,EAAE,KAAK,CAAC,YAAY,CAAC,EAC3B,QAAQ,GAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAM,GACjC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,CAAC,CAIvC;AAED;;GAEG;AACH,wBAAgB,uBAAuB,CACtC,OAAO,EAAE,KAAK,CAAC,cAAc,CAAC,GAC5B;IAAE,eAAe,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IAAC,YAAY,EAAE,KAAK,CAAC,MAAM,CAAC,CAAA;CAAE,CA8DjE;AAED,wBAAgB,qBAAqB,CACpC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,GACnC;IAAE,eAAe,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAAC,YAAY,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAA;CAAE,CAc/E;AAED;;GAEG;AACH,wBAAgB,WAAW,CAC1B,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EACjC,OAAO,GAAE,OAAe,GACtB,KAAK,CAAC,MAAM,CAAC,CAgBf;AA+DD;;GAEG;AACH,wBAAsB,YAAY,CACjC,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,EACnB,KAAK,GAAE,MAAkB,EACzB,WAAW,GAAE,cAAkD,EAC/D,cAAc,GAAE,MAAY,GAC1B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI,CAAC,CAAC,CAc5C;AAED;;GAEG;AACH,wBAAsB,kBAAkB,CACvC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,EACjC,OAAO,GAAE,OAAe,EACxB,KAAK,GAAE,MAAkB,EACzB,WAAW,GAAE,cAAkD,EAC/D,cAAc,GAAE,MAAY,GAC1B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,UAAU,GAAG,IAAI,CAAC,CAAC,CAI5C;AAED;;GAEG;AACH,wBAAsB,iBAAiB,CACtC,WAAW,EAAE,KAAK,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC,EACzC,OAAO,GAAE,OAAe,EACxB,KAAK,GAAE,MAAkB,EACzB,WAAW,GAAE,cAAkD,EAC/D,cAAc,GAAE,MAAY,GAC1B,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC,CAAC,CAiB1C"}

package/script/src/tools/sources.js

@@ -8,7 +8,6 @@ exports.collectURLs = collectURLs;
 exports.classifyURLs = classifyURLs;
 exports.makeCategoryMapper = makeCategoryMapper;
 exports.categorizeSources = categorizeSources;
-const async_js_1 = require("../async.js");
 const brands_js_1 = require("./brands.js");
 const topics_js_1 = require("./topics.js");
 /**
@@ -232,16 +231,16 @@ const WEB_TAXONOMY = {
         'Personal / Portfolio Site'
     ]
 };
-const WEB_LABEL_SCHEMA = (0, topics_js_1.createLabelSchema)({ topics: (0, topics_js_1.toTopics)(WEB_TAXONOMY) });
-const WEB_TAXONOMY_SERIALIZED = JSON.stringify(WEB_TAXONOMY, null, 2);
+const WEB_TAXONOMY_TOPICS = (0, topics_js_1.toTopics)(WEB_TAXONOMY);
 /**
  * Classifies an array of URLs into categories based on WEB_TAXONOMY.
  */
 async function classifyURLs(urls, model = 'gpt-5.1', modelParams = { reasoning: { effort: 'none' } }, maxConcurrency = 100) {
-    const labels = await (0, async_js_1.mapParallel)(urls, maxConcurrency, url => (0, topics_js_1.assignTopic)(url, WEB_TAXONOMY_SERIALIZED, WEB_LABEL_SCHEMA, model, modelParams));
+    const assigner = new topics_js_1.TopicAssigner({ taxonomy: WEB_TAXONOMY_TOPICS }, { model, modelParams, maxConcurrency });
+    const results = await assigner.batch(urls);
     const urlToCategory = {};
     urls.forEach((url, index) => {
-        urlToCategory[url] = labels[index];
+        urlToCategory[url] = results.toArray()[index];
     });
     return urlToCategory;
 }

package/script/src/tools/topics.d.ts

@@ -1,5 +1,6 @@
 import { z } from '../../deps/jsr.io/@zod/zod/4.3.6/src/index.js';
-import { type AIParams } from '../openai.js';
+import { Tool, type ModelConfig } from '../tool.js';
+import type { Message } from '../llm.js';
 import { type TopicType, type TaxonomyType, type TopicLabel } from '../schemas/topics.schema.js';
 export type { TopicType, TaxonomyType, TopicLabel };
 /**
@@ -48,27 +49,54 @@ export declare function createLabelSchema(taxonomy: TaxonomyType): z.ZodObject<{
         [x: string]: string;
     }>;
 }, z.core.$strip>;
-/**
- * Assigns a topic and subtopic to a single text using an LLM call.
- */
-export declare function assignTopic(text: string | null, taxonomy: TaxonomyType | Array<TopicType> | string, labelSchema: z.ZodType<TopicLabel>, model?: string, modelParams?: AIParams): Promise<TopicLabel | null>;
-/**
- * Assigns topics to multiple texts concurrently while preserving order.
- */
-export declare function assignTopics(texts: Array<string | null>, taxonomy: TaxonomyType | Array<TopicType>, model?: string, modelParams?: AIParams, maxConcurrency?: number): Promise<Array<TopicLabel | null>>;
-export interface TopicExtractionOptions {
-    records: Array<Record<string, unknown>>;
+export interface TopicExtractorConfig {
+    /** Maximum number of top-level topics (default: 10) */
     nTopics?: number;
+    /** Maximum number of subtopics per topic (default: 5) */
     nSubtopics?: number;
+    /** Additional instructions */
     instructions?: string;
+    /** Maximum number of records to sample (default: 500) */
     maxSamples?: number;
-    model?: string;
-    modelParams?: AIParams;
-    maxRetries?: number;
+    /** Language for topics (default: same as records) */
     language?: string;
 }
 /**
- * Extracts a topic hierarchy from an array of records using an LLM.
+ * A tool that extracts a topic taxonomy from a set of records.
  */
-export declare function extractTopics({ records, nTopics, nSubtopics, instructions, maxSamples, model, modelParams, maxRetries, language }: TopicExtractionOptions): Promise<TaxonomyType>;
+export declare class TopicExtractor extends Tool<Array<Record<string, unknown>>, TaxonomyType, TaxonomyType> {
+    private readonly maxSamples;
+    private readonly promptTemplate;
+    constructor(config: TopicExtractorConfig | undefined, modelConfig: ModelConfig);
+    protected schema(): z.ZodObject<{
+        topics: z.ZodArray<z.ZodObject<{
+            topic: z.ZodString;
+            subtopics: z.ZodArray<z.ZodString>;
+        }, z.core.$strip>>;
+    }, z.core.$strip>;
+    protected prompt(records: Array<Record<string, unknown>>): string;
+    protected isEmpty(records: Array<Record<string, unknown>>): boolean;
+    /**
+     * Not supported. TopicExtractor is an aggregation operation that extracts
+     * a single taxonomy from many records. Use invoke() instead.
+     */
+    batch(): never;
+}
+/**
+ * Configuration for the TopicAssigner tool.
+ */
+export interface TopicAssignerConfig {
+    taxonomy: TaxonomyType | Array<TopicType>;
+}
+/**
+ * A tool that assigns topic and subtopic labels to text based on a taxonomy.
+ */
+export declare class TopicAssigner extends Tool<string | null, TopicLabel, TopicLabel> {
+    private readonly labelSchema;
+    private readonly systemPrompt;
+    constructor(config: TopicAssignerConfig, modelConfig: ModelConfig);
+    protected schema(): z.ZodType<TopicLabel, unknown, z.core.$ZodTypeInternals<TopicLabel, unknown>>;
+    protected prompt(text: string | null): Message[];
+    protected isEmpty(text: string | null): boolean;
+}
 //# sourceMappingURL=topics.d.ts.map

package/script/src/tools/topics.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"topics.d.ts","sourceRoot":"","sources":["../../../src/src/tools/topics.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,+CAA+C,CAAC;AAElE,OAAO,EAAsC,KAAK,QAAQ,EAAE,MAAM,cAAc,CAAC;AAEjF,OAAO,EAEN,KAAK,SAAS,EACd,KAAK,YAAY,EACjB,KAAK,UAAU,EACf,MAAM,6BAA6B,CAAC;AAIrC,YAAY,EAAE,SAAS,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC;AAuEpD;;;GAGG;AACH,eAAO,MAAM,KAAK;;;iBAShB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,YAAY;;;;;iBAEvB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,QAAQ;;;;;iBAEnB,CAAC;AAEH;;;GAGG;AACH,wBAAgB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,KAAK,CAAC;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,CAAA;CAAE,CAAC,CAKpH;AACD;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,YAAY;;;;;;;kBAmBvD;AA+CD;;GAEG;AACH,wBAAsB,WAAW,CAChC,IAAI,EAAE,MAAM,GAAG,IAAI,EACnB,QAAQ,EAAE,YAAY,GAAG,KAAK,CAAC,SAAS,CAAC,GAAG,MAAM,EAClD,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,EAClC,KAAK,GAAE,MAAkB,EACzB,WAAW,GAAE,QAA4C,GACvD,OAAO,CAAC,UAAU,GAAG,IAAI,CAAC,CA2B5B;AAED;;GAEG;AACH,wBAAgB,YAAY,CAC3B,KAAK,EAAE,KAAK,CAAC,MAAM,GAAG,IAAI,CAAC,EAC3B,QAAQ,EAAE,YAAY,GAAG,KAAK,CAAC,SAAS,CAAC,EACzC,KAAK,GAAE,MAAkB,EACzB,WAAW,GAAE,QAA4C,EACzD,cAAc,GAAE,MAAY,GAC1B,OAAO,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC,CAgBnC;AAED,MAAM,WAAW,sBAAsB;IACtC,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC;IACxC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,QAAQ,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,wBAAsB,aAAa,CAAC,EACnC,OAAO,EACP,OAAY,EACZ,UAAc,EACd,YAAiB,EACjB,UAAgB,EAChB,KAAiB,EACjB,WAAgB,EAChB,UAAc,EACd,QAA6C,EAC7C,EAAE,sBAAsB,GAAG,OAAO,CAAC,YAAY,CAAC,CAsChD"}
+{"version":3,"file":"topics.d.ts","sourceRoot":"","sources":["../../../src/src/tools/topics.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,+CAA+C,CAAC;AAClE,OAAO,EAAE,IAAI,EAAE,KAAK,WAAW,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,EAEN,KAAK,SAAS,EACd,KAAK,YAAY,EACjB,KAAK,UAAU,EACf,MAAM,6BAA6B,CAAC;AAIrC,YAAY,EAAE,SAAS,EAAE,YAAY,EAAE,UAAU,EAAE,CAAC;AA+EpD;;;GAGG;AACH,eAAO,MAAM,KAAK;;;iBAShB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,YAAY;;;;;iBAEvB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,QAAQ;;;;;iBAEnB,CAAC;AAEH;;;GAGG;AACH,wBAAgB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,GAAG,KAAK,CAAC;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,KAAK,CAAC,MAAM,CAAC,CAAA;CAAE,CAAC,CAKpH;AAED;;;GAGG;AACH,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,YAAY;;;;;;;kBAmBvD;AA+BD,MAAM,WAAW,oBAAoB;IACpC,uDAAuD;IACvD,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,yDAAyD;IACzD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,8BAA8B;IAC9B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,yDAAyD;IACzD,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,qDAAqD;IACrD,QAAQ,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;GAEG;AACH,qBAAa,cAAe,SAAQ,IAAI,CACvC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,EAC9B,YAAY,EACZ,YAAY,CACZ;IACA,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAS;gBAE5B,MAAM,EAAE,oBAAoB,YAAK,EAAE,WAAW,EAAE,WAAW;cAkBpD,MAAM;;;;;;IAIzB,SAAS,CAAC,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;cAQrC,OAAO,CAAC,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,GAAG,OAAO;IAI5E;;;OAGG;IACM,KAAK,IAAI,KAAK;CAOvB;AAyBD;;GAEG;AACH,MAAM,WAAW,mBAAmB;IACnC,QAAQ,EAAE,YAAY,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC;CAC1C;AAED;;GAEG;AACH,qBAAa,aAAc,SAAQ,IAAI,CAAC,MAAM,GAAG,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;IAC7E,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAwB;IACpD,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;gBAE1B,MAAM,EAAE,mBAAmB,EAAE,WAAW,EAAE,WAAW;cAiB9C,MAAM;IAIzB,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO,EAAE;cAQ7B,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO;CAGxD"}

package/script/src/tools/topics.js

@@ -1,17 +1,16 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Taxonomy = exports.TaxonomyBase = exports.Topic = void 0;
+exports.TopicAssigner = exports.TopicExtractor = exports.Taxonomy = exports.TaxonomyBase = exports.Topic = void 0;
 exports.toTopics = toTopics;
 exports.createLabelSchema = createLabelSchema;
-exports.assignTopic = assignTopic;
-exports.assignTopics = assignTopics;
-exports.extractTopics = extractTopics;
 /* eslint no-console: ["warn", { allow: ["log", "warn", "error"] }] */
 const index_js_1 = require("../../deps/jsr.io/@zod/zod/4.3.6/src/index.js");
-const async_js_1 = require("../async.js");
-const openai_js_1 = require("../openai.js");
+const tool_js_1 = require("../tool.js");
 const topics_schema_js_1 = require("../schemas/topics.schema.js");
-const utils_js_1 = require("../utils.js");
+const utils_js_1 = require("../helpers/utils.js");
+// =============================================================================
+// Helper Functions
+// =============================================================================
 /**
  * Levenshtein distance implementation for string similarity validation
  */
@@ -69,6 +68,9 @@ function validateSubtopics(topic, subtopics) {
     }
     return errors;
 }
+// =============================================================================
+// Schemas
+// =============================================================================
 /**
  * Topic schema with subtopic validation
  * Validates that subtopics are sufficiently distinct from each other and from the parent topic.
@@ -128,6 +130,9 @@ function createLabelSchema(taxonomy) {
         }
     });
 }
+// =============================================================================
+// TopicExtractor
+// =============================================================================
 const TOPICS_PROMPT = (0, utils_js_1.dedent)(`
 # Instructions
 
@@ -152,6 +157,49 @@ The taxonomy should follow the MECE framework (Mutually Exclusive, Collectively
 
 {records}
 `);
+/**
+ * A tool that extracts a topic taxonomy from a set of records.
+ */
+class TopicExtractor extends tool_js_1.Tool {
+    maxSamples;
+    promptTemplate;
+    constructor(config = {}, modelConfig) {
+        super(modelConfig);
+        const { nTopics = 10, nSubtopics = 5, instructions = '', maxSamples = 500, language = 'The same language as the records' } = config;
+        this.maxSamples = maxSamples;
+        this.promptTemplate = TOPICS_PROMPT
+            .replace('{n_topics}', String(nTopics))
+            .replace('{n_subtopics}', String(nSubtopics))
+            .replace('{instructions}', instructions)
+            .replace('{language}', language);
+    }
+    schema() {
+        return exports.Taxonomy;
+    }
+    prompt(records) {
+        const sampledRecords = records.length > this.maxSamples
+            ? records.slice(0, this.maxSamples)
+            : records;
+        const formattedRecords = (0, utils_js_1.formatRecordsAttrWise)(sampledRecords);
+        return this.promptTemplate.replace('{records}', formattedRecords);
+    }
+    isEmpty(records) {
+        return !records || records.length === 0;
+    }
+    /**
+     * Not supported. TopicExtractor is an aggregation operation that extracts
+     * a single taxonomy from many records. Use invoke() instead.
+     */
+    batch() {
+        throw new Error('TopicExtractor.batch() is not supported. ' +
+            'This tool extracts a single taxonomy from many records (aggregation). ' +
+            'Use invoke() with all records instead.');
+    }
+}
+exports.TopicExtractor = TopicExtractor;
+// =============================================================================
+// TopicAssigner
+// =============================================================================
 const LABEL_PROMPT_SYSTEM = (0, utils_js_1.dedent)(`
 You're task is to use the following hierarchy of topics and subtopics (in json format),
 to assign the correct topic and subtopic to each text in the input.
@@ -170,74 +218,34 @@ Assign the correct topic and subtopic to the following text.
 {text}
 `);
 /**
- * Assigns a topic and subtopic to a single text using an LLM call.
+ * A tool that assigns topic and subtopic labels to text based on a taxonomy.
  */
-async function assignTopic(text, taxonomy, labelSchema, model = 'gpt-5.1', modelParams = { reasoning: { effort: 'none' } }) {
-    if (text == null || text.trim() === '') {
-        return null;
-    }
-    const systemPrompt = LABEL_PROMPT_SYSTEM.replace('{taxonomy}', typeof taxonomy === 'string' ? taxonomy : JSON.stringify(taxonomy, null, 2));
-    const userPrompt = LABEL_PROMPT_USER.replace('{text}', text);
-    const prompts = [
-        { role: 'system', content: systemPrompt },
-        { role: 'user', content: userPrompt }
-    ];
-    try {
-        const { parsed, error } = await (0, openai_js_1.askOpenAISafe)(prompts, model, labelSchema, modelParams);
-        if (error != null || parsed == null) {
-            return null;
-        }
-        return parsed;
+class TopicAssigner extends tool_js_1.Tool {
+    labelSchema;
+    systemPrompt;
+    constructor(config, modelConfig) {
+        super(modelConfig);
+        const { taxonomy } = config;
+        // Normalize taxonomy to TaxonomyType
+        const normalizedTaxonomy = Array.isArray(taxonomy)
+            ? { topics: taxonomy }
+            : taxonomy;
+        // Build schema and system prompt once in constructor
+        this.labelSchema = createLabelSchema(normalizedTaxonomy);
+        this.systemPrompt = LABEL_PROMPT_SYSTEM.replace('{taxonomy}', JSON.stringify(normalizedTaxonomy, null, 2));
     }
-    catch (error) {
-        console.warn(`Failed to assign topic for text "${text.substring(0, 50)}...":`, error);
-        return null;
+    schema() {
+        return this.labelSchema;
     }
-}
-/**
- * Assigns topics to multiple texts concurrently while preserving order.
- */
-function assignTopics(texts, taxonomy, model = 'gpt-5.1', modelParams = { reasoning: { effort: 'none' } }, maxConcurrency = 100) {
-    // Precompute normalized taxonomy and label schema once for whole batch
-    const normalizedTaxonomy = Array.isArray(taxonomy)
-        ? { topics: taxonomy }
-        : taxonomy;
-    const labelSchema = createLabelSchema(normalizedTaxonomy);
-    const serializedTaxonomy = JSON.stringify(normalizedTaxonomy, null, 2);
-    return (0, async_js_1.mapParallel)(texts, maxConcurrency, text => assignTopic(text, serializedTaxonomy, labelSchema, model, modelParams));
-}
-/**
- * Extracts a topic hierarchy from an array of records using an LLM.
- */
-async function extractTopics({ records, nTopics = 10, nSubtopics = 5, instructions = '', maxSamples = 500, model = 'gpt-4.1', modelParams = {}, maxRetries = 8, language = 'The same language as the records' }) {
-    if (!records || records.length === 0) {
-        return { topics: [] };
-    }
-    const sampledRecords = records.length > maxSamples
-        ? records.slice(0, maxSamples)
-        : records;
-    const formattedRecords = (0, utils_js_1.formatRecordsAttrWise)(sampledRecords);
-    const prompt = TOPICS_PROMPT
-        .replace('{n_topics}', String(nTopics))
-        .replace('{n_subtopics}', String(nSubtopics))
-        .replace('{instructions}', instructions)
-        .replace('{records}', formattedRecords)
-        .replace('{language}', language);
-    const { parsed, output_text, error } = await (0, openai_js_1.askOpenAISafe)(prompt, model, exports.Taxonomy, modelParams, maxRetries, 'return');
-    if (error != null) {
-        if (output_text == null) {
-            throw new Error('Failed to get response from OpenAI');
-        }
-        try {
-            const taxonomy = JSON.parse(output_text);
-            return exports.TaxonomyBase.parse(taxonomy);
-        }
-        catch (parseError) {
-            throw new Error(`Failed to parse response even without validation: ${parseError instanceof Error ? parseError.message : String(parseError)}`);
-        }
+    prompt(text) {
+        const userPrompt = LABEL_PROMPT_USER.replace('{text}', text ?? '');
+        return [
+            { role: 'system', content: this.systemPrompt },
+            { role: 'user', content: userPrompt }
+        ];
     }
-    if (parsed == null) {
-        throw new Error('Failed to parse response from OpenAI');
+    isEmpty(text) {
+        return text == null || text.trim() === '';
     }
-    return parsed;
 }
+exports.TopicAssigner = TopicAssigner;

package/script/src/tools/translate.d.ts

@@ -1,44 +1,35 @@
+import { Tool, type ModelConfig } from '../tool.js';
 import type { BrandContext } from '../schemas/brand.schema.js';
-export interface translateParams {
-    keyword: string;
+export interface KeywordTranslatorConfig {
+    /** Brand context for disambiguation */
+    brandContext: BrandContext;
+    /** Language for the output prompt (default: 'en') */
     language?: string;
-    model?: string;
+    /** Additional instructions */
     instructions?: string;
-    brandContext: BrandContext;
 }
 /**
- * Translates a single Google search keyword into an equivalent LLM prompt.
+ * A tool that converts Google search keywords into natural language LLM prompts.
+ * Uses raw text mode (no schema) since the output is plain text.
  */
-export declare function translate({ keyword, language, model, brandContext, instructions }: translateParams): Promise<string>;
-/**
- * Translates multiple Google search keywords to LLM prompts concurrently while preserving order.
- */
-export interface translateBatchParams {
-    keywords: Array<string>;
-    language?: string;
-    model?: string;
-    maxConcurrency?: number;
-    instructions?: string;
-    brandContext: BrandContext;
+export declare class KeywordTranslator extends Tool<string | null, string, string> {
+    private readonly promptTemplate;
+    constructor(config: KeywordTranslatorConfig, modelConfig: ModelConfig);
+    protected prompt(keyword: string | null): string;
+    protected isEmpty(keyword: string | null): boolean;
 }
-export declare function translateBatch(params: translateBatchParams): Promise<Array<string>>;
-export interface reverseTranslateParams {
-    prompt: string;
+export interface PromptToKeywordConfig {
+    /** Language for the output keyword (default: 'en') */
     language?: string;
-    model?: string;
 }
 /**
- * Converts a natural language prompt into an equivalent Google search keyword.
+ * A tool that converts natural language prompts into Google search keywords.
+ * Uses raw text mode (no schema) since the output is plain text.
  */
-export declare function reverseTranslate({ prompt, language, model }: reverseTranslateParams): Promise<string>;
-export interface reverseTranslateBatchParams {
-    prompts: Array<string>;
-    language?: string;
-    model?: string;
-    maxConcurrency?: number;
+export declare class PromptToKeyword extends Tool<string | null, string, string> {
+    private readonly promptTemplate;
+    constructor(config: PromptToKeywordConfig | undefined, modelConfig: ModelConfig);
+    protected prompt(text: string | null): string;
+    protected isEmpty(text: string | null): boolean;
 }
-/**
- * Converts multiple natural language prompts to Google search keywords concurrently while preserving order.
- */
-export declare function reverseTranslateBatch(params: reverseTranslateBatchParams): Promise<Array<string>>;
 //# sourceMappingURL=translate.d.ts.map

package/script/src/tools/translate.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"translate.d.ts","sourceRoot":"","sources":["../../../src/src/tools/translate.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AA+C/D,MAAM,WAAW,eAAe;IAC/B,OAAO,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,YAAY,CAAC;CAC3B;AAED;;GAEG;AACH,wBAAsB,SAAS,CAC9B,EAAE,OAAO,EAAE,QAAe,EAAE,KAAoB,EAAE,YAAY,EAAE,YAAiB,EAAE,EAAE,eAAe,GAClG,OAAO,CAAC,MAAM,CAAC,CAcjB;AAED;;GAEG;AAEH,MAAM,WAAW,oBAAoB;IACpC,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,EAAE,YAAY,CAAC;CAC3B;AAED,wBAAsB,cAAc,CACnC,MAAM,EAAE,oBAAoB,GAC1B,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAQxB;AA+BD,MAAM,WAAW,sBAAsB;IACtC,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;GAEG;AACH,wBAAsB,gBAAgB,CACrC,EAAE,MAAM,EAAE,QAAe,EAAE,KAAsB,EAAE,EAAE,sBAAsB,GACzE,OAAO,CAAC,MAAM,CAAC,CAWjB;AAED,MAAM,WAAW,2BAA2B;IAC3C,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC;IACvB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,cAAc,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;GAEG;AACH,wBAAsB,qBAAqB,CAC1C,MAAM,EAAE,2BAA2B,GACjC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAQxB"}
+{"version":3,"file":"translate.d.ts","sourceRoot":"","sources":["../../../src/src/tools/translate.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,KAAK,WAAW,EAAE,MAAM,YAAY,CAAC;AAEpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AA+C/D,MAAM,WAAW,uBAAuB;IACvC,uCAAuC;IACvC,YAAY,EAAE,YAAY,CAAC;IAC3B,qDAAqD;IACrD,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,8BAA8B;IAC9B,YAAY,CAAC,EAAE,MAAM,CAAC;CACtB;AAED;;;GAGG;AACH,qBAAa,iBAAkB,SAAQ,IAAI,CAAC,MAAM,GAAG,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC;IACzE,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAS;gBAE5B,MAAM,EAAE,uBAAuB,EAAE,WAAW,EAAE,WAAW;IAarE,SAAS,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;cAIpB,OAAO,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO;CAG3D;AAmCD,MAAM,WAAW,qBAAqB;IACrC,sDAAsD;IACtD,QAAQ,CAAC,EAAE,MAAM,CAAC;CAClB;AAED;;;GAGG;AACH,qBAAa,eAAgB,SAAQ,IAAI,CAAC,MAAM,GAAG,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC;IACvE,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAS;gBAE5B,MAAM,EAAE,qBAAqB,YAAK,EAAE,WAAW,EAAE,WAAW;IASxE,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI;cAIjB,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,GAAG,OAAO;CAGxD"}

package/script/src/tools/translate.js

@@ -1,12 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.translate = translate;
-exports.translateBatch = translateBatch;
-exports.reverseTranslate = reverseTranslate;
-exports.reverseTranslateBatch = reverseTranslateBatch;
-const async_js_1 = require("../async.js");
-const openai_js_1 = require("../openai.js");
-const utils_js_1 = require("../utils.js");
+exports.PromptToKeyword = exports.KeywordTranslator = void 0;
+const tool_js_1 = require("../tool.js");
+const utils_js_1 = require("../helpers/utils.js");
 const TRANSLATE_PROMPT = (0, utils_js_1.dedent)(`
 You are an expert in understanding search intent and conversational AI interactions.
 
@@ -39,7 +35,7 @@ You will receive brand context (sector, country). Follow these rules strictly:
 - WRONG: "Tell me about geography" (ignores sector context)
 - CORRECT: "What is GEO in AI visibility and search optimization?" (uses sector to disambiguate the acronym)
 
-{additional_instructions}
+{instructions}
 
 # Keyword
 
@@ -52,25 +48,32 @@ You will receive brand context (sector, country). Follow these rules strictly:
 Return only the converted prompt, nothing else.
 `);
 /**
- * Translates a single Google search keyword into an equivalent LLM prompt.
+ * A tool that converts Google search keywords into natural language LLM prompts.
+ * Uses raw text mode (no schema) since the output is plain text.
  */
-async function translate({ keyword, language = 'en', model = 'gpt-5-mini', brandContext, instructions = '' }) {
-    const prompt = TRANSLATE_PROMPT
-        .replace('{keyword}', keyword)
-        .replace('{language}', language)
-        .replace('{sector}', brandContext.sector)
-        .replace('{country}', brandContext.country)
-        .replace('{additional_instructions}', instructions);
-    const { parsed } = await (0, openai_js_1.askOpenAISafe)(prompt, model);
-    if (!parsed) {
-        throw new Error('Failed to parse translation from OpenAI response');
+class KeywordTranslator extends tool_js_1.Tool {
+    promptTemplate;
+    constructor(config, modelConfig) {
+        super(modelConfig);
+        const { brandContext, language = 'en', instructions = '' } = config;
+        this.promptTemplate = TRANSLATE_PROMPT
+            .replace('{language}', language)
+            .replace('{sector}', brandContext.sector)
+            .replace('{country}', brandContext.country)
+            .replace('{instructions}', instructions);
+    }
+    // No schema() override - uses default null for raw text mode
+    prompt(keyword) {
+        return this.promptTemplate.replace('{keyword}', keyword ?? '');
+    }
+    isEmpty(keyword) {
+        return keyword == null || keyword.trim() === '';
     }
-    return parsed;
-}
-async function translateBatch(params) {
-    const { keywords, language = 'en', model = 'gpt-4.1-mini', maxConcurrency = 100, brandContext, instructions = '' } = params;
-    return (0, async_js_1.mapParallel)(keywords, maxConcurrency, kwd => translate({ keyword: kwd, language, model, brandContext, instructions }));
 }
+exports.KeywordTranslator = KeywordTranslator;
+// =============================================================================
+// PromptToKeyword (LLM prompt → keyword)
+// =============================================================================
 const REVERSE_TRANSLATE_PROMPT = (0, utils_js_1.dedent)(`
 You are an expert in understanding search intent.
 
@@ -100,22 +103,22 @@ Otherwise, Google keyword planner will not accept it.
 Return only the keyword, nothing else.
 `);
 /**
- * Converts a natural language prompt into an equivalent Google search keyword.
+ * A tool that converts natural language prompts into Google search keywords.
+ * Uses raw text mode (no schema) since the output is plain text.
  */
-async function reverseTranslate({ prompt, language = 'en', model = 'gpt-4.1-mini' }) {
-    const requestPrompt = REVERSE_TRANSLATE_PROMPT
-        .replace('{prompt}', prompt)
-        .replace('{language}', language);
-    const { parsed } = await (0, openai_js_1.askOpenAISafe)(requestPrompt, model);
-    if (!parsed) {
-        throw new Error('Failed to parse reverse translation from OpenAI response');
+class PromptToKeyword extends tool_js_1.Tool {
+    promptTemplate;
+    constructor(config = {}, modelConfig) {
+        super(modelConfig);
+        const { language = 'en' } = config;
+        this.promptTemplate = REVERSE_TRANSLATE_PROMPT.replace('{language}', language);
+    }
+    // No schema() override - uses default null for raw text mode
+    prompt(text) {
+        return this.promptTemplate.replace('{prompt}', text ?? '');
+    }
+    isEmpty(text) {
+        return text == null || text.trim() === '';
     }
-    return parsed;
-}
-/**
- * Converts multiple natural language prompts to Google search keywords concurrently while preserving order.
- */
-async function reverseTranslateBatch(params) {
-    const { prompts, language = 'en', model = 'gpt-5-mini', maxConcurrency = 100 } = params;
-    return (0, async_js_1.mapParallel)(prompts, maxConcurrency, p => reverseTranslate({ prompt: p, language, model }));
 }
+exports.PromptToKeyword = PromptToKeyword;