yym68686 commited on
Commit
f27ccad
·
1 Parent(s): 3790bd8

update models

Browse files
Files changed (1) hide show
  1. main.py +10 -7
main.py CHANGED
@@ -135,16 +135,19 @@ async def request_model(request: RequestModel, token: str = Depends(verify_api_k
135
  def get_all_models():
136
  config = load_config()
137
  all_models = []
 
138
 
139
  for provider in config:
140
  for model in provider['model']:
141
- model_info = {
142
- "id": model,
143
- "object": "model",
144
- "created": 1720524448858,
145
- "owned_by": provider['provider']
146
- }
147
- all_models.append(model_info)
 
 
148
 
149
  return all_models
150
 
 
135
  def get_all_models():
136
  config = load_config()
137
  all_models = []
138
+ unique_models = set()
139
 
140
  for provider in config:
141
  for model in provider['model']:
142
+ if model not in unique_models:
143
+ unique_models.add(model)
144
+ model_info = {
145
+ "id": model,
146
+ "object": "model",
147
+ "created": 1720524448858,
148
+ "owned_by": provider['provider']
149
+ }
150
+ all_models.append(model_info)
151
 
152
  return all_models
153