Function rewriteQuery

src/tools/query-rewriter.ts:95–112

(action: SearchAction, tracker?: TokenTracker)

const TOOL_NAME = 'queryRewriter';

export async function rewriteQuery(action: SearchAction, tracker?: TokenTracker): Promise<{ queries: string[] }> {
  try {
    const generator = new ObjectGeneratorSafe(tracker);
    const prompt = getPrompt(action);

    const result = await generator.generateObject({
      model: TOOL_NAME,
      schema: responseSchema,
      prompt,
    });

    console.log(TOOL_NAME, result.object.queries);
    return { queries: result.object.queries };
  } catch (error) {
    console.error(`Error in ${TOOL_NAME}`, error);
    throw error;
  }
}