Update config.yaml

config.yaml

@@ -7,9 +7,9 @@ model_list:
     litellm_params:
       model: github/gpt-4o-mini
       api_key: "os.environ/GITHUB_API_KEY"
-  - model_name: gemini-1.5-pro-002
+  - model_name: gemini-1.5-pro
     litellm_params:
-      model: gemini/gemini-1.5-pro-002
+      model: gemini/gemini-1.5-pro
       api_key: "os.environ/GEMINI_API_KEY"
       safety_settings:
         - category: HARM_CATEGORY_HARASSMENT
@@ -20,9 +20,9 @@ model_list:
           threshold: BLOCK_NONE
         - category: HARM_CATEGORY_DANGEROUS_CONTENT
           threshold: BLOCK_NONE
-  - model_name: gemini-1.5-flash-002
+  - model_name: gemini-1.5-flash
     litellm_params:
-      model: gemini/gemini-1.5-flash-002
+      model: gemini/gemini-1.5-flash
       api_key: "os.environ/GEMINI_API_KEY"
       safety_settings:
         - category: HARM_CATEGORY_HARASSMENT