Merge pull request #54 from samselikoff/add-enums-for-model-props

Add enums for model property
togethercomputer · Oct 30, 2024 · 9dd8113 · 9dd8113
2 parents 11ef1ca + 8d29fc9
commit 9dd8113
Showing 1 changed file with 54 additions and 5 deletions.
diff --git a/openapi.yaml b/openapi.yaml
@@ -251,8 +251,18 @@ paths:
                   example: cat floating in space, cinematic
                 model:
                   type: string
-                  description: The model to use for image generation.
+                  description: >
+                    The model to use for image generation.<br>
+                    <br>
+                    [See all of Together AI's image models](https://docs.together.ai/docs/serverless-models#image-models)
                   example: stabilityai/stable-diffusion-xl-base-1.0
+                  anyOf:
+                    - type: string
+                      enum:
+                        - black-forest-labs/FLUX.1-schnell-Free
+                        - black-forest-labs/FLUX.1-schnell
+                        - black-forest-labs/FLUX.1.1-pro
+                    - type: string
                 steps:
                   type: integer
                   default: 20
@@ -603,8 +613,17 @@ components:
       properties:
         model:
           type: string
-          description: The model to be used for the rerank request.
+          description: >
+            The model to be used for the rerank request.<br>
+            <br>
+            [See all of Together AI's rerank models](https://docs.together.ai/docs/serverless-models#rerank-models)
           example: Salesforce/Llama-Rank-V1
+          anyOf:
+            - type: string
+              enum:
+                - Salesforce/Llama-Rank-v1
+            - type: string
+
         query:
           type: string
           description: The search query to be used for ranking.
@@ -830,8 +849,19 @@ components:
           example: <s>[INST] What is the capital of France? [/INST]
         model:
           type: string
-          description: The name of the model to query.
+          description: >
+            The name of the model to query.<br>
+            <br>
+            [See all of Together AI's chat models](https://docs.together.ai/docs/serverless-models#chat-models)
           example: mistralai/Mixtral-8x7B-Instruct-v0.1
+          anyOf:
+            - type: string
+              enum:
+                - meta-llama/Llama-2-70b-hf
+                - mistralai/Mistral-7B-v0.1
+                - mistralai/Mixtral-8x7B-v0.1
+                - Meta-Llama/Llama-Guard-7b
+            - type: string
         max_tokens:
           type: integer
           description: The maximum number of tokens to generate.
@@ -876,6 +906,11 @@ components:
           type: string
           description: The name of the moderation model used to validate tokens. Choose from the available moderation models found [here](https://docs.together.ai/docs/inference-models#moderation-models).
           example: 'safety_model_name'
+          anyOf:
+            - type: string
+              enum:
+                - Meta-Llama/Llama-Guard-7b
+            - type: string
         min_p:
           type: number
           description: A number between 0 and 1 that can be used as an alternative to top-p and top-k.
@@ -1071,7 +1106,10 @@ components:
               - role
               - content
         model:
-          description: The name of the model to query.
+          description: >
+            The name of the model to query.<br>
+            <br>
+            [See all of Together AI's chat models](https://docs.together.ai/docs/serverless-models#chat-models)
           example: meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo
           anyOf:
             - type: string
@@ -1442,8 +1480,19 @@ components:
       properties:
         model:
           type: string
-          description: The name of the embedding model to use.
+          description: >
+            The name of the embedding model to use.<br>
+            <br>
+            [See all of Together AI's embedding models](https://docs.together.ai/docs/serverless-models#embedding-models)
           example: togethercomputer/m2-bert-80M-8k-retrieval
+          anyOf:
+            - type: string
+              enum:
+                - WhereIsAI/UAE-Large-V1
+                - BAAI/bge-large-en-v1.5
+                - BAAI/bge-base-en-v1.5
+                - togethercomputer/m2-bert-80M-8k-retrieval
+            - type: string
         input:
           oneOf:
             - type: string