feat: added max_tokens field to Cohere API

Vali-98 · Aug 11, 2024 · 8ae9760 · 8ae9760
1 parent 5807bf5
commit 8ae9760
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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 },