ParthSadaria commited on
Commit
20b41cb
·
verified ·
1 Parent(s): 7d8010a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +12 -1
main.py CHANGED
@@ -106,7 +106,18 @@ secret_api_endpoint_2 = os.getenv('SECRET_API_ENDPOINT_2')
106
  secret_api_endpoint_3 = os.getenv('SECRET_API_ENDPOINT_3') # New endpoint for searchgpt
107
  mistral_api = "https://api.mistral.ai"
108
  mistral_key = os.getenv('MISTRAL_KEY')
109
- mistral_models = ['mistral-saba-latest','mistral-small-latest','pixtral-large-latest']
 
 
 
 
 
 
 
 
 
 
 
110
  image_endpoint = os.getenv("IMAGE_ENDPOINT")
111
  ENDPOINT_ORIGIN = os.getenv('ENDPOINT_ORIGIN')
112
 
 
106
  secret_api_endpoint_3 = os.getenv('SECRET_API_ENDPOINT_3') # New endpoint for searchgpt
107
  mistral_api = "https://api.mistral.ai"
108
  mistral_key = os.getenv('MISTRAL_KEY')
109
+ mistral_models = [
110
+ "mistral-large-latest",
111
+ "pixtral-large-latest",
112
+ "mistral-moderation-latest",
113
+ "ministral-3b-latest",
114
+ "ministral-8b-latest",
115
+ "open-mistral-nemo",
116
+ "mistral-small-latest",
117
+ "mistral-saba-latest",
118
+ "codestral-latest"
119
+ ]
120
+
121
  image_endpoint = os.getenv("IMAGE_ENDPOINT")
122
  ENDPOINT_ORIGIN = os.getenv('ENDPOINT_ORIGIN')
123