Update config.yaml
Browse files- config.yaml +4 -4
config.yaml
CHANGED
@@ -41,10 +41,10 @@ model_list:
|
|
41 |
threshold: BLOCK_NONE
|
42 |
- category: HARM_CATEGORY_CIVIC_INTEGRITY
|
43 |
threshold: BLOCK_NONE
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
|
49 |
litellm_settings:
|
50 |
set_verbose: False
|
|
|
41 |
threshold: BLOCK_NONE
|
42 |
- category: HARM_CATEGORY_CIVIC_INTEGRITY
|
43 |
threshold: BLOCK_NONE
|
44 |
+
- model_name: text-embedding-004
|
45 |
+
litellm_params:
|
46 |
+
model: gemini/text-embedding-004
|
47 |
+
api_key: "os.environ/GEMINI_API_KEY"
|
48 |
|
49 |
litellm_settings:
|
50 |
set_verbose: False
|