github/node-DeepResearch
1
0
Fork 0
mirror of https://github.com/jina-ai/node-DeepResearch.git synced 2026-03-22 15:39:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: categorize search tokens as prompt tokens

Browse Source 
Co-Authored-By: sha.zhou@jina.ai <sha.zhou@jina.ai>
This commit is contained in:
Devin AI
2025-02-11 10:34:59 +00:00
parent 67347d7a40
commit 2c249c18ba
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/tools/jina-search.ts
View File

@@ -1,6 +1,6 @@
import https from 'https';
import { TokenTracker } from "../utils/token-tracker";
import { SearchResponse } from '../types';
import { SearchResponse, TOKEN_CATEGORIES } from '../types';
import { JINA_API_KEY } from "../config";
export function search(query: string, tracker?: TokenTracker): Promise<{ response: SearchResponse, tokens: number }> {
@@ -63,7 +63,7 @@ export function search(query: string, tracker?: TokenTracker): Promise<{ respons
console.log('Total URLs:', response.data.length);
if (tracker) {
tracker.trackUsage('search', totalTokens);
tracker.trackUsage('search', totalTokens, TOKEN_CATEGORIES.PROMPT);
}
resolve({ response, tokens: totalTokens });