feat(api): api update (#79)

Author: stainless-app[bot]

.stats.yml

@@ -1,2 +1,2 @@
 configured_endpoints: 16
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/togetherai%2FTogetherAI-dac0349a130b23065cbd771f383ef7eb7fa99ba35cae0da6be2f55f82d2f6d3a.yml
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/togetherai%2FTogetherAI-3950390ec89ef1e5aec351d8e841d511ebc0a3ccec5bbf0d4c73157d1f9d7c1d.yml

src/resources/chat/completions.ts

@@ -250,6 +250,8 @@ export interface CompletionCreateParamsBase {
 
   /**
    * The name of the model to query.
+   *
+   * [See all of Together AI's chat models](https://docs.together.ai/docs/serverless-models#chat-models)
    */
   model:
     | 'Qwen/Qwen2.5-72B-Instruct-Turbo'

src/resources/completions.ts

@@ -123,8 +123,15 @@ export type CompletionCreateParams = CompletionCreateParamsNonStreaming | Comple
 export interface CompletionCreateParamsBase {
   /**
    * The name of the model to query.
+   *
+   * [See all of Together AI's chat models](https://docs.together.ai/docs/serverless-models#chat-models)
    */
-  model: string;
+  model:
+    | 'meta-llama/Llama-2-70b-hf'
+    | 'mistralai/Mistral-7B-v0.1'
+    | 'mistralai/Mixtral-8x7B-v0.1'
+    | 'Meta-Llama/Llama-Guard-7b'
+    | (string & {});
 
   /**
    * A string providing context for the model to complete.
@@ -186,7 +193,7 @@ export interface CompletionCreateParamsBase {
    * available moderation models found
    * [here](https://docs.together.ai/docs/inference-models#moderation-models).
    */
-  safety_model?: string;
+  safety_model?: 'Meta-Llama/Llama-Guard-7b' | (string & {});
 
   /**
    * Seed value for reproducibility.

src/resources/embeddings.ts

@@ -39,8 +39,15 @@ export interface EmbeddingCreateParams {
 
   /**
    * The name of the embedding model to use.
+   *
+   * [See all of Together AI's embedding models](https://docs.together.ai/docs/serverless-models#embedding-models)
    */
-  model: string;
+  model:
+    | 'WhereIsAI/UAE-Large-V1'
+    | 'BAAI/bge-large-en-v1.5'
+    | 'BAAI/bge-base-en-v1.5'
+    | 'togethercomputer/m2-bert-80M-8k-retrieval'
+    | (string & {});
 }
 
 export namespace Embeddings {

src/resources/images.ts

@@ -34,8 +34,14 @@ export namespace ImageFile {
 export interface ImageCreateParams {
   /**
    * The model to use for image generation.
+   *
+   * [See all of Together AI's image models](https://docs.together.ai/docs/serverless-models#image-models)
    */
-  model: string;
+  model:
+    | 'black-forest-labs/FLUX.1-schnell-Free'
+    | 'black-forest-labs/FLUX.1-schnell'
+    | 'black-forest-labs/FLUX.1.1-pro'
+    | (string & {});
 
   /**
    * A description of the desired images. Maximum length varies by model.

src/resources/top-level.ts

@@ -48,8 +48,10 @@ export interface RerankParams {
 
   /**
    * The model to be used for the rerank request.
+   *
+   * [See all of Together AI's rerank models](https://docs.together.ai/docs/serverless-models#rerank-models)
    */
-  model: string;
+  model: 'Salesforce/Llama-Rank-v1' | (string & {});
 
   /**
    * The search query to be used for ranking.

tests/api-resources/embeddings.test.ts

@@ -12,7 +12,7 @@ describe('resource embeddings', () => {
   test('create: only required params', async () => {
     const responsePromise = client.embeddings.create({
       input: 'Our solar system orbits the Milky Way galaxy at about 515,000 mph',
-      model: 'togethercomputer/m2-bert-80M-8k-retrieval',
+      model: 'WhereIsAI/UAE-Large-V1',
     });
     const rawResponse = await responsePromise.asResponse();
     expect(rawResponse).toBeInstanceOf(Response);
@@ -26,7 +26,7 @@ describe('resource embeddings', () => {
   test('create: required and optional params', async () => {
     const response = await client.embeddings.create({
       input: 'Our solar system orbits the Milky Way galaxy at about 515,000 mph',
-      model: 'togethercomputer/m2-bert-80M-8k-retrieval',
+      model: 'WhereIsAI/UAE-Large-V1',
     });
   });
 });