From 8ae976082f7b6ccf9a2aa14cca3df0cd6e4a38e8 Mon Sep 17 00:00:00 2001
From: Vali98 <al244098@gmail.com>
Date: Sun, 11 Aug 2024 19:39:31 +0800
Subject: [PATCH] feat: added max_tokens field to Cohere API

---
 app/constants/APIState/CohereAPI.ts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/app/constants/APIState/CohereAPI.ts b/app/constants/APIState/CohereAPI.ts
index 756e85d..769ca47 100644
--- a/app/constants/APIState/CohereAPI.ts
+++ b/app/constants/APIState/CohereAPI.ts
@@ -8,6 +8,7 @@ import { APIBase, APISampler } from './BaseAPI'
 class CohereAPI extends APIBase {
     samplers: APISampler[] = [
         { externalName: 'max_context_length', samplerID: SamplerID.CONTEXT_LENGTH },
+        { externalName: 'max_tokens', samplerID: SamplerID.GENERATED_LENGTH },
         { externalName: 'stream', samplerID: SamplerID.STREAMING },
         { externalName: 'temperature', samplerID: SamplerID.TEMPERATURE },
         { externalName: 'p', samplerID: SamplerID.TOP_P },