bugfixing #3
This commit is contained in:
parent
1c4dcdd108
commit
e89e3b7a70
6 changed files with 137 additions and 30 deletions
|
|
@ -26,8 +26,9 @@ export class GeminiAdapter implements LlmAdapter {
|
|||
promptTokens: number;
|
||||
completionTokens: number;
|
||||
totalTokens: number;
|
||||
promptTimeMs: number;
|
||||
completionTimeMs: number;
|
||||
promptTimeMs: number | null;
|
||||
completionTimeMs: number | null;
|
||||
totalTimeMs: number;
|
||||
}> {
|
||||
const url = `https://generativelanguage.googleapis.com/v1beta/models/${this.model}:generateContent?key=${this.apiKey}`;
|
||||
|
||||
|
|
@ -72,8 +73,9 @@ export class GeminiAdapter implements LlmAdapter {
|
|||
promptTokens,
|
||||
completionTokens,
|
||||
totalTokens: json.usageMetadata.totalTokenCount,
|
||||
promptTimeMs: totalTimeMs * 0.3,
|
||||
completionTimeMs: totalTimeMs * 0.7,
|
||||
promptTimeMs: null,
|
||||
completionTimeMs: null,
|
||||
totalTimeMs,
|
||||
};
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue