diff --git a/packages/client/src/pages/import/import.constants.ts b/packages/client/src/pages/import/import.constants.ts
index 8ca7c8ac32..9877f15d07 100644
--- a/packages/client/src/pages/import/import.constants.ts
+++ b/packages/client/src/pages/import/import.constants.ts
@@ -1844,7 +1844,7 @@ export const CONNECTION_OPTIONS = {
                     ],
                 },
                 {
-                    name: 'Gemini 1.5 pro',
+                    name: 'Gemini',
                     disable: false,
                     icon: VERTEX,
                     fields: [
@@ -1891,9 +1891,19 @@ export const CONNECTION_OPTIONS = {
                             label: 'Model',
                             defaultValue: 'gemini-1.5-pro-002',
                             options: {
-                                component: 'text-field',
+                                component: 'select',
+                                options: [
+                                    {
+                                        display: 'gemini-1.5-pro-002',
+                                        value: 'gemini-1.5-pro-002',
+                                    },
+                                    {
+                                        display: 'gemini-2.0-flash-001',
+                                        value: 'gemini-2.0-flash-001',
+                                    },
+                                ],
                             },
-                            disabled: true,
+                            disabled: false,
                             rules: { required: true },
                         },
                         {