bugfixing #1
This commit is contained in:
parent
afd28d934e
commit
2a6c56ed23
12 changed files with 300 additions and 269 deletions
|
|
@ -1,4 +1,3 @@
|
|||
// utils/enrich-word.ts
|
||||
import { ENRICHMENT_SYSTEM_PROMPT } from "../config/prompt.js";
|
||||
import { createAdapter } from "./llm-adapters/factory.js";
|
||||
import { BATCH_CONFIG } from "../config/batch.js";
|
||||
|
|
@ -26,8 +25,6 @@ interface LlmResponse {
|
|||
totalTokens: number;
|
||||
promptTimeMs: number;
|
||||
completionTimeMs: number;
|
||||
promptTokensPerSecond: number;
|
||||
completionTokensPerSecond: number;
|
||||
}
|
||||
|
||||
export interface EnrichmentResult {
|
||||
|
|
@ -38,13 +35,11 @@ export interface EnrichmentResult {
|
|||
totalTokens: number;
|
||||
promptTimeMs: number;
|
||||
completionTimeMs: number;
|
||||
promptTokensPerSecond: number;
|
||||
completionTokensPerSecond: number;
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* Calls the local LLM with the enrichment prompt.
|
||||
* Calls the LLM with the enrichment prompt.
|
||||
* Returns the response content and timing metrics.
|
||||
*/
|
||||
async function callLlm(words: string[]): Promise<LlmResponse> {
|
||||
|
|
@ -157,8 +152,6 @@ export async function enrichWord(
|
|||
totalTokens: llmResponse.totalTokens,
|
||||
promptTimeMs: llmResponse.promptTimeMs,
|
||||
completionTimeMs: llmResponse.completionTimeMs,
|
||||
promptTokensPerSecond: llmResponse.promptTokensPerSecond,
|
||||
completionTokensPerSecond: llmResponse.completionTokensPerSecond,
|
||||
},
|
||||
};
|
||||
}
|
||||
|
|
@ -225,14 +218,6 @@ export async function enrichWordWithRetry(
|
|||
completionTimeMs:
|
||||
leftResult.metrics.completionTimeMs +
|
||||
rightResult.metrics.completionTimeMs,
|
||||
promptTokensPerSecond:
|
||||
(leftResult.metrics.promptTokensPerSecond +
|
||||
rightResult.metrics.promptTokensPerSecond) /
|
||||
2,
|
||||
completionTokensPerSecond:
|
||||
(leftResult.metrics.completionTokensPerSecond +
|
||||
rightResult.metrics.completionTokensPerSecond) /
|
||||
2,
|
||||
};
|
||||
|
||||
return { results: merged, metrics: mergedMetrics };
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue