diff --git a/fern/apis/fai/definition/chat.yml b/fern/apis/fai/definition/chat.yml
index 34be34919..4ff478da8 100644
--- a/fern/apis/fai/definition/chat.yml
+++ b/fern/apis/fai/definition/chat.yml
@@ -19,9 +19,15 @@ service:
         name: ChatCompletionRequest
         body:
           properties:
-            model: optional<string>
-            system_prompt: optional<string>
-            messages: list<ChatTurn>
+            model:
+              type: LanguageModel
+              docs: The model to use for the chat completion.
+            system_prompt:
+              type: optional<string>
+              docs: The system prompt to use for the chat completion.
+            messages: 
+              type: list<ChatTurn>
+              docs: The messages to use for the chat completion.
       response: ChatCompletionResponse
       errors:
         - commons.BadRequestError
@@ -29,9 +35,16 @@ service:
 
 types:
   ChatCompletionResponse:
-    type: list<ChatTurn>
+    properties:
+      turns: list<ChatTurn>
+      citations: list<string>
 
   ChatTurn:
     properties:
       role: string
       content: string
+
+  LanguageModel:
+    enum:
+      - claude-4-sonnet-20250514
+      - command-a-03-2025