diff --git a/src/config.ts b/src/config.ts
index 8bdbeba..7ad9da0 100644
--- a/src/config.ts
+++ b/src/config.ts
@@ -43,7 +43,7 @@ export const aiConfig: AIConfig = {
   providers: {
     gemini: {
       type: 'gemini',
-      model: 'gemini-pro',
+      model: 'gemini-1.5-pro',
       temperature: 0
     },
     openai: {
diff --git a/src/utils/schema.ts b/src/utils/schema.ts
index 6a50b35..6b79616 100644
--- a/src/utils/schema.ts
+++ b/src/utils/schema.ts
@@ -110,14 +110,8 @@ export function getProviderSchema(provider: ProviderType, schema: z.ZodSchema):
     case 'gemini':
       return convertToGeminiSchema(schema);
     case 'openai':
-    case 'ollama': {
-      const functionSchema = convertToOpenAIFunctionSchema(schema);
-      return {
-        type: 'object',
-        properties: functionSchema.properties,
-        required: functionSchema.required
-      };
-    }
+    case 'ollama':
+      return convertToOpenAIFunctionSchema(schema);
     default:
       throw new Error(`Unsupported provider: ${provider}`);
   }