chore: first commit

This commit is contained in:
Han Xiao 2025-02-02 23:41:59 +08:00
parent 48f8565291
commit 1039a3f827
2 changed files with 3 additions and 24 deletions

View File

@ -656,10 +656,10 @@ const genAI = new GoogleGenerativeAI(GEMINI_API_KEY);
export async function main() {
const question = process.argv[2] || "";
const { result: finalStep } = await getResponse(question) as { result: AnswerAction; context: TrackerContext };
const { result: finalStep, context: tracker } = await getResponse(question) as { result: AnswerAction; context: TrackerContext };
console.log('Final Answer:', finalStep.answer);
const tracker = new TokenTracker();
tracker.printSummary();
tracker.tokenTracker.printSummary();
}
if (require.main === module) {

View File

@ -1,21 +0,0 @@
import { SearchResult, QueryResult, StepData } from '../types';
export function buildURLMap(data: StepData[]): Record<string, SearchResult['url'], QueryResult['query']> {
const urlMap: Record<string, string> = {};
data.forEach(step => {
if (step.result && Array.isArray(step.result)) {
step.result.forEach(queryResult => {
if (queryResult.results && Array.isArray(queryResult.results)) {
queryResult.results.forEach(result => {
if (!urlMap[result.url]) {
urlMap[result.url] = `${result.title}`;
}
});
}
});
}
});
return urlMap;
}