feat(api): OpenAPI spec update via Stainless API

togethercomputer · May 20, 2024 · 404545c · 404545c
1 parent 8cc7d02
commit 404545c
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/.stats.yml b/.stats.yml
@@ -1,2 +1,2 @@
 configured_endpoints: 15
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/togetherai%2FTogetherAI-628b11c5046d63f1f609888e5c537b2891f39ba27c74b267736fb6694713dfe4.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/togetherai%2FTogetherAI-ab60dcafb04015e34b38c2991f76f78f35eeb0860fe02ee4033c4161c297b3bc.yml
diff --git a/src/resources/chat/completions.ts b/src/resources/chat/completions.ts
@@ -140,7 +140,7 @@ export interface CompletionCreateParamsBase {
    * The `logit_bias` parameter allows us to adjust the likelihood of specific tokens
    * appearing in the generated output.
    */
-  logit_bias?: Record<string, string>;
+  logit_bias?: Record<string, unknown>;
 
   /**
    * Determines the number of most likely tokens to return at each token position log

diff --git a/src/resources/completions.ts b/src/resources/completions.ts
@@ -135,7 +135,7 @@ export interface CompletionCreateParamsBase {
    * The `logit_bias` parameter allows us to adjust the likelihood of specific tokens
    * appearing in the generated output.
    */
-  logit_bias?: Record<string, string>;
+  logit_bias?: Record<string, unknown>;
 
   /**
    * Determines the number of most likely tokens to return at each token position log

diff --git a/tests/api-resources/chat/completions.test.ts b/tests/api-resources/chat/completions.test.ts
@@ -37,7 +37,7 @@ describe('resource completions', () => {
       model: 'mistralai/Mixtral-8x7B-Instruct-v0.1',
       echo: true,
       frequency_penalty: 0,
-      logit_bias: { '105': 'string', '1024': 'string' },
+      logit_bias: { '105': 21.4, '1024': -10.5 },
       logprobs: 0,
       max_tokens: 0,
       min_p: 0,

diff --git a/tests/api-resources/completions.test.ts b/tests/api-resources/completions.test.ts
@@ -29,7 +29,7 @@ describe('resource completions', () => {
       prompt: '<s>[INST] What is the capital of France? [/INST]',
       echo: true,
       frequency_penalty: 0,
-      logit_bias: { '105': 'string', '1024': 'string' },
+      logit_bias: { '105': 21.4, '1024': -10.5 },
       logprobs: 0,
       max_tokens: 0,
       min_p: 0,