Update config.yaml

config.yaml

@@ -11,10 +11,6 @@ model_list:
     litellm_params:
       model: github/mistral-large-2407
       api_key: "os.environ/GITHUB_API_KEY"
-  - model_name: text-embedding-3-large
-    litellm_params:
-      model: github/text-embedding-3-large
-      api_key: "os.environ/GITHUB_API_KEY"
   - model_name: gemini-1.5-pro-latest
     litellm_params:
       model: gemini/gemini-1.5-pro-latest
@@ -45,6 +41,10 @@ model_list:
           threshold: BLOCK_NONE
         - category: HARM_CATEGORY_CIVIC_INTEGRITY
           threshold: BLOCK_NONE
+    - model_name: text-embedding-004
+      litellm_params:
+        model: gemini/text-embedding-004
+        api_key: "os.environ/GEMINI_API_KEY"
 
 litellm_settings:
   set_verbose: False