fix: xata clone config --mode=ai (#2629)

Synced from monorepo@1da65b45fc9190b0ae85a8d80fb43e2891fdeab0

Author: divyenduz

.sync-commit

@@ -1 +1 @@
-fd1fcfb9be6f0863b3a59624e3c64c3e86243b59
+1da65b45fc9190b0ae85a8d80fb43e2891fdeab0

src/commands/ai/sql.ts

@@ -16,6 +16,7 @@ type Flags = {
   project?: string;
   branch?: string;
   database?: string;
+  model?: string;
   yes: boolean;
 };
 
@@ -113,7 +114,7 @@ export async function implementation(this: LocalContext, flags: Flags) {
 
     const handleGenerateSQL = async (query: string, currentSQL: string): Promise<string> => {
       invariant(env.ANTHROPIC_API_KEY, 'ANTHROPIC_API_KEY is required for AI features');
-      return await generateSQL(env.ANTHROPIC_API_KEY, query, formattedSchema, currentSQL);
+      return await generateSQL(env.ANTHROPIC_API_KEY, query, formattedSchema, currentSQL, { model: flags.model });
     };
 
     const ui = new TerminalUI({
@@ -161,6 +162,12 @@ export const GenerateSQLCommand = buildCommand({
         parse: String,
         optional: true
       },
+      model: {
+        kind: 'parsed',
+        brief: 'Anthropic model override for AI SQL generation',
+        parse: String,
+        optional: true
+      },
       yes: {
         kind: 'boolean',
         brief: 'Do not ask for confirmation, assume yes.',

src/commands/clone/config.ts

@@ -60,6 +60,7 @@ type Flags = {
   project?: string;
   branch?: string;
   prompt?: string;
+  model?: string;
 };
 
 export async function implementation(this: LocalContext, flags: Flags) {
@@ -147,7 +148,8 @@ export async function implementation(this: LocalContext, flags: Flags) {
           env.ANTHROPIC_API_KEY,
           effectivePrompt,
           formattedSchema,
-          previousConfigYaml
+          previousConfigYaml,
+          { model: flags.model }
         );
       } catch (err) {
         this.process.stderr.write(
@@ -265,6 +267,12 @@ export const CloneConfigCommand = buildCommand({
         brief: 'Instructions for AI mode (e.g., which columns to anonymize, specific transformers to use)',
         parse: String,
         optional: true
+      },
+      model: {
+        kind: 'parsed',
+        brief: 'Anthropic model override for AI mode',
+        parse: String,
+        optional: true
       }
     }
   },