diff --git a/packages/sdk/js/src/v2/gen/types.gen.ts b/packages/sdk/js/src/v2/gen/types.gen.ts
index cb1606e3f..10483d71e 100644
--- a/packages/sdk/js/src/v2/gen/types.gen.ts
+++ b/packages/sdk/js/src/v2/gen/types.gen.ts
@@ -1379,7 +1379,7 @@ export type PermissionConfig =
   | PermissionActionConfig
 
 export type AgentConfig = {
-  model?: string
+  model?: Model
   /**
    * Default model variant for this agent (applies only when using the agent's configured model).
    */
@@ -1421,6 +1421,7 @@ export type AgentConfig = {
   permission?: PermissionConfig
   [key: string]:
     | unknown
+    | Model
     | string
     | number
     | {
@@ -1650,7 +1651,7 @@ export type Config = {
       template: string
       description?: string
       agent?: string
-      model?: string
+      model?: Model
       subtask?: boolean
     }
   }
@@ -1688,14 +1689,8 @@ export type Config = {
    * When set, ONLY these providers will be enabled. All other providers will be ignored
    */
   enabled_providers?: Array<string>
-  /**
-   * Model to use in the format of provider/model, eg anthropic/claude-2
-   */
-  model?: string
-  /**
-   * Small model to use for tasks like title generation in the format of provider/model
-   */
-  small_model?: string
+  model?: Model
+  small_model?: Model
   /**
    * Default agent to use when none is specified. Must be a primary agent. Falls back to 'build' if not set or if the specified agent is invalid.
    */
diff --git a/packages/sdk/openapi.json b/packages/sdk/openapi.json
index 1425a6e9c..1033c1f13 100644
--- a/packages/sdk/openapi.json
+++ b/packages/sdk/openapi.json
@@ -9028,7 +9028,7 @@
         "type": "object",
         "properties": {
           "model": {
-            "type": "string"
+            "$ref": "#/components/schemas/Model"
           },
           "variant": {
             "description": "Default model variant for this agent (applies only when using the agent's configured model).",
@@ -9528,7 +9528,7 @@
                   "type": "string"
                 },
                 "model": {
-                  "type": "string"
+                  "$ref": "#/components/schemas/Model"
                 },
                 "subtask": {
                   "type": "boolean"
@@ -9606,12 +9606,10 @@
             }
           },
           "model": {
-            "description": "Model to use in the format of provider/model, eg anthropic/claude-2",
-            "type": "string"
+            "$ref": "#/components/schemas/Model"
           },
           "small_model": {
-            "description": "Small model to use for tasks like title generation in the format of provider/model",
-            "type": "string"
+            "$ref": "#/components/schemas/Model"
           },
           "default_agent": {
             "description": "Default agent to use when none is specified. Must be a primary agent. Falls back to 'build' if not set or if the specified agent is invalid.",