Guru-25 commited on
Commit
3fa1649
1 Parent(s): ee97eb6

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +46 -0
config.yaml CHANGED
@@ -1,4 +1,50 @@
1
  model_list:
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  - model_name: guru-llama
3
  litellm_params:
4
  model: github/meta-llama-3.1-70b-instruct
 
1
  model_list:
2
+ - model_name: gpt-4o
3
+ litellm_params:
4
+ model: github/gpt-4o
5
+ api_key: "os.environ/GITHUB_API_KEY"
6
+ - model_name: gpt-4o-mini
7
+ litellm_params:
8
+ model: github/gpt-4o-mini
9
+ api_key: "os.environ/GITHUB_API_KEY"
10
+ - model_name: mistral-large-2407
11
+ litellm_params:
12
+ model: github/mistral-large-2407
13
+ api_key: "os.environ/GITHUB_API_KEY"
14
+ - model_name: gemini-1.5-pro-latest
15
+ litellm_params:
16
+ model: gemini/gemini-1.5-pro-latest
17
+ api_key: "os.environ/GEMINI_API_KEY"
18
+ safety_settings:
19
+ - category: HARM_CATEGORY_HARASSMENT
20
+ threshold: BLOCK_NONE
21
+ - category: HARM_CATEGORY_HATE_SPEECH
22
+ threshold: BLOCK_NONE
23
+ - category: HARM_CATEGORY_SEXUALLY_EXPLICIT
24
+ threshold: BLOCK_NONE
25
+ - category: HARM_CATEGORY_DANGEROUS_CONTENT
26
+ threshold: BLOCK_NONE
27
+ - category: HARM_CATEGORY_CIVIC_INTEGRITY
28
+ threshold: BLOCK_NONE
29
+ - model_name: gemini-1.5-flash-latest
30
+ litellm_params:
31
+ model: gemini/gemini-1.5-flash-latest
32
+ api_key: "os.environ/GEMINI_API_KEY"
33
+ safety_settings:
34
+ - category: HARM_CATEGORY_HARASSMENT
35
+ threshold: BLOCK_NONE
36
+ - category: HARM_CATEGORY_HATE_SPEECH
37
+ threshold: BLOCK_NONE
38
+ - category: HARM_CATEGORY_SEXUALLY_EXPLICIT
39
+ threshold: BLOCK_NONE
40
+ - category: HARM_CATEGORY_DANGEROUS_CONTENT
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
  - model_name: guru-llama
49
  litellm_params:
50
  model: github/meta-llama-3.1-70b-instruct