ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-25 06:14:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: get tokens from api usage (#4671)

Browse Source
This commit is contained in:
Archer
2025-04-27 01:13:38 +08:00
committed by GitHub
parent 0720bbe4da
commit 1465999c46
26 changed files with 270 additions and 223 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
packages/global/core/ai/constants.ts
View File

@@ -1,4 +1,13 @@
import { i18nT } from '../../../web/i18n/utils';
import type { CompletionUsage } from './type';
export const getLLMDefaultUsage = (): CompletionUsage => {
return {
prompt_tokens: 0,
completion_tokens: 0,
total_tokens: 0
};
};
export enum ChatCompletionRequestMessageRoleEnum {
'System' = 'system',

2
packages/global/core/ai/type.d.ts vendored
View File

@@ -10,6 +10,7 @@ import type {
} from 'openai/resources';
import { ChatMessageTypeEnum } from './constants';
import { WorkflowInteractiveResponseType } from '../workflow/template/system/interactive/type';
import { Stream } from 'openai/streaming';
export * from 'openai/resources';
// Extension of ChatCompletionMessageParam, Add file url type
@@ -84,6 +85,7 @@ export type CompletionFinishReason =
export default openai;
export * from 'openai';
export type { Stream };
// Other
export type PromptTemplateItem = {