diff --git a/src/backend/drivers/ai-chat/providers/moonshot/MoonshotProvider.ts b/src/backend/drivers/ai-chat/providers/moonshot/MoonshotProvider.ts index 079a74601..df886c1c0 100644 --- a/src/backend/drivers/ai-chat/providers/moonshot/MoonshotProvider.ts +++ b/src/backend/drivers/ai-chat/providers/moonshot/MoonshotProvider.ts @@ -21,12 +21,12 @@ import { OpenAI } from 'openai'; import { ChatCompletionCreateParams } from 'openai/resources/index.js'; import { Context } from '../../../../core/context.js'; import type { MeteringService } from '../../../../services/metering/MeteringService.js'; -import * as OpenAIUtil from '../../utils/OpenAIUtil.js'; import type { + IChatCompleteResult, IChatProvider, ICompleteArguments, - IChatCompleteResult, } from '../../types.js'; +import * as OpenAIUtil from '../../utils/OpenAIUtil.js'; import { inlineHttpImageUrls } from './imageHandling.js'; import { MOONSHOT_MODELS } from './models.js'; @@ -68,6 +68,7 @@ export class MoonshotProvider implements IChatProvider { stream, model, tools, + max_tokens, }: ICompleteArguments): Promise { const actor = Context.get('actor'); const availableModels = this.models(); @@ -89,7 +90,7 @@ export class MoonshotProvider implements IChatProvider { messages, model: modelUsed.id, ...(tools ? { tools } : {}), - max_tokens: 1000, + max_tokens, stream, ...(stream ? {