Guru-25 commited on
Commit
9285cb5
·
verified ·
1 Parent(s): dbf0853

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +8 -4
config.yaml CHANGED
@@ -7,9 +7,9 @@ model_list:
7
  litellm_params:
8
  model: github/gpt-4o-mini
9
  api_key: "os.environ/GITHUB_API_KEY"
10
- - model_name: gemini-1.5-pro-002
11
  litellm_params:
12
- model: gemini/gemini-1.5-pro-002
13
  api_key: "os.environ/GEMINI_API_KEY"
14
  safety_settings:
15
  - category: HARM_CATEGORY_HARASSMENT
@@ -20,9 +20,11 @@ model_list:
20
  threshold: BLOCK_NONE
21
  - category: HARM_CATEGORY_DANGEROUS_CONTENT
22
  threshold: BLOCK_NONE
23
- - model_name: gemini-1.5-flash-002
 
 
24
  litellm_params:
25
- model: gemini/gemini-1.5-flash-002
26
  api_key: "os.environ/GEMINI_API_KEY"
27
  safety_settings:
28
  - category: HARM_CATEGORY_HARASSMENT
@@ -33,6 +35,8 @@ model_list:
33
  threshold: BLOCK_NONE
34
  - category: HARM_CATEGORY_DANGEROUS_CONTENT
35
  threshold: BLOCK_NONE
 
 
36
  - model_name: text-embedding-004
37
  litellm_params:
38
  model: gemini/text-embedding-004
 
7
  litellm_params:
8
  model: github/gpt-4o-mini
9
  api_key: "os.environ/GITHUB_API_KEY"
10
+ - model_name: gemini-1.5-pro-latest
11
  litellm_params:
12
+ model: gemini/gemini-1.5-pro-latest
13
  api_key: "os.environ/GEMINI_API_KEY"
14
  safety_settings:
15
  - category: HARM_CATEGORY_HARASSMENT
 
20
  threshold: BLOCK_NONE
21
  - category: HARM_CATEGORY_DANGEROUS_CONTENT
22
  threshold: BLOCK_NONE
23
+ - category: HARM_CATEGORY_CIVIC_INTEGRITY
24
+ threshold: BLOCK_NONE
25
+ - model_name: gemini-1.5-flash-latest
26
  litellm_params:
27
+ model: gemini/gemini-1.5-flash-latest
28
  api_key: "os.environ/GEMINI_API_KEY"
29
  safety_settings:
30
  - category: HARM_CATEGORY_HARASSMENT
 
35
  threshold: BLOCK_NONE
36
  - category: HARM_CATEGORY_DANGEROUS_CONTENT
37
  threshold: BLOCK_NONE
38
+ - category: HARM_CATEGORY_CIVIC_INTEGRITY
39
+ threshold: BLOCK_NONE
40
  - model_name: text-embedding-004
41
  litellm_params:
42
  model: gemini/text-embedding-004