slockers commited on
Commit
7195864
·
1 Parent(s): 6481e74

added openai embedding, cohere, mistral models

Browse files
Files changed (1) hide show
  1. config.yaml +63 -4
config.yaml CHANGED
@@ -86,7 +86,7 @@ model_list:
86
  litellm_params:
87
  model: groq/deepseek-r1-distill-llama-70b
88
  api_key: os.environ/GROQ_API_KEY2
89
-
90
  # --------------openai--------------------
91
 
92
  - model_name: op1/gpt-4o-mini
@@ -94,10 +94,69 @@ model_list:
94
  model: openai/gpt-4o-mini
95
  api_key: os.environ/OPENAI_API_KEY1
96
 
97
- - model_name: op2/gpt-4o-mini
98
  litellm_params:
99
- model: openai/gpt-4o-mini
100
- api_key: os.environ/OPENAI_API_KEY2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
101
 
102
  # --------------huggingface--------------------
103
 
 
86
  litellm_params:
87
  model: groq/deepseek-r1-distill-llama-70b
88
  api_key: os.environ/GROQ_API_KEY2
89
+
90
  # --------------openai--------------------
91
 
92
  - model_name: op1/gpt-4o-mini
 
94
  model: openai/gpt-4o-mini
95
  api_key: os.environ/OPENAI_API_KEY1
96
 
97
+ - model_name: op1/text-embedding-3-large
98
  litellm_params:
99
+ model: openai/text-embedding-3-large
100
+ api_key: os.environ/OPENAI_API_KEY1
101
+
102
+ - model_name: op1/text-embedding-3-small
103
+ litellm_params:
104
+ model: openai/text-embedding-3-small
105
+ api_key: os.environ/OPENAI_API_KEY1
106
+
107
+ # - model_name: op2/gpt-4o-mini
108
+ # litellm_params:
109
+ # model: openai/gpt-4o-mini
110
+ # api_key: os.environ/OPENAI_API_KEY2
111
+
112
+ # --------------cohere--------------------
113
+
114
+ - model_name: ch/command-r-plus
115
+ litellm_params:
116
+ model: cohere/command-r-plus-08-2024
117
+ api_key: os.environ/COHERE_API_KEY
118
+
119
+ - model_name: ch/embed-english
120
+ litellm_params:
121
+ model: cohere/embed-english-v3.0
122
+ api_key: os.environ/COHERE_API_KEY
123
+
124
+ - model_name: ch/embed-english-light
125
+ litellm_params:
126
+ model: cohere/embed-english-light-v3.0
127
+ api_key: os.environ/COHERE_API_KEY
128
+
129
+ - model_name: ch/embed-multilingual
130
+ litellm_params:
131
+ model: cohere/embed-multilingual-v3.0
132
+ api_key: os.environ/COHERE_API_KEY
133
+
134
+ - model_name: ch/embed-multilingual-light
135
+ litellm_params:
136
+ model: cohere/embed-multilingual-light-v3.0
137
+ api_key: os.environ/COHERE_API_KEY
138
+
139
+ - model_name: ch/rerank-english
140
+ litellm_params:
141
+ model: cohere/rerank-english-v3.0
142
+ api_key: os.environ/COHERE_API_KEY
143
+
144
+ # --------------mistral--------------------
145
+
146
+ - model_name: ms/mistral-large-latest
147
+ litellm_params:
148
+ model: mistral/mistral-large-latest
149
+ api_key: os.environ/MISTRAL_API_KEY
150
+
151
+ - model_name: ms/mistral-small-latest
152
+ litellm_params:
153
+ model: mistral/mistral-small-latest
154
+ api_key: os.environ/MISTRAL_API_KEY
155
+
156
+ - model_name: ms/mistral-embed
157
+ litellm_params:
158
+ model: mistral/mistral-embed
159
+ api_key: os.environ/MISTRAL_API_KEY
160
 
161
  # --------------huggingface--------------------
162