huanx520 commited on
Commit
b98e00b
·
1 Parent(s): 83b36db

模型统一为 deepseek-flash / deepseek-pro,去除别名映射

Browse files
Files changed (2) hide show
  1. deepseek_browser.py +1 -1
  2. main.py +7 -15
deepseek_browser.py CHANGED
@@ -132,7 +132,7 @@ class DeepSeekBrowser:
132
 
133
  async def switch_model(self, model: str):
134
  try:
135
- if 'reasoner' in model or 'thinking' in model:
136
  thinking_btn = self.page.locator('button:has-text("深度思考"), div:has-text("深度思考")').first
137
  if await thinking_btn.count() > 0:
138
  await thinking_btn.click()
 
132
 
133
  async def switch_model(self, model: str):
134
  try:
135
+ if 'reasoner' in model or 'thinking' in model or 'pro' in model:
136
  thinking_btn = self.page.locator('button:has-text("深度思考"), div:has-text("深度思考")').first
137
  if await thinking_btn.count() > 0:
138
  await thinking_btn.click()
main.py CHANGED
@@ -63,10 +63,8 @@ async def list_models(authorization: str = Header(...)):
63
 
64
  return {
65
  "data": [
66
- {"id": "deepseek-chat", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
67
- {"id": "deepseek-reasoner", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
68
- {"id": "deepseek-v4-flash", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
69
- {"id": "deepseek-v4-pro", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
70
  ],
71
  "object": "list",
72
  }
@@ -77,10 +75,8 @@ async def get_model(model_id: str, authorization: str = Header(...)):
77
  verify_api_key(authorization)
78
 
79
  models = {
80
- "deepseek-chat": {"id": "deepseek-chat", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
81
- "deepseek-reasoner": {"id": "deepseek-reasoner", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
82
- "deepseek-v4-flash": {"id": "deepseek-v4-flash", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
83
- "deepseek-v4-pro": {"id": "deepseek-v4-pro", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
84
  }
85
 
86
  if model_id in models:
@@ -101,9 +97,7 @@ async def chat_completions(
101
 
102
  prompt = request.messages[-1].content
103
 
104
- # 模型别名: chat→flash, reasoner→pro
105
- MODEL_ALIASES = {"deepseek-chat": "deepseek-v4-flash", "deepseek-reasoner": "deepseek-v4-pro"}
106
- model = MODEL_ALIASES.get(request.model, request.model)
107
 
108
  account = await manager.acquire()
109
 
@@ -374,10 +368,8 @@ textarea::placeholder{color:var(--dim)}
374
  <div class="panel-body">
375
  <div class="row">
376
  <select id="model" style="flex:1">
377
- <option value="deepseek-v4-flash">deepseek-v4-flash</option>
378
- <option value="deepseek-v4-pro">deepseek-v4-pro</option>
379
- <option value="deepseek-chat">deepseek-chat</option>
380
- <option value="deepseek-reasoner">deepseek-reasoner</option>
381
  </select>
382
  <label style="font-size:11px;color:var(--dim);display:flex;align-items:center;gap:4px;white-space:nowrap">
383
  <input type="checkbox" id="stream" checked> 流式
 
63
 
64
  return {
65
  "data": [
66
+ {"id": "deepseek-flash", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
67
+ {"id": "deepseek-pro", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
 
 
68
  ],
69
  "object": "list",
70
  }
 
75
  verify_api_key(authorization)
76
 
77
  models = {
78
+ "deepseek-flash": {"id": "deepseek-flash", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
79
+ "deepseek-pro": {"id": "deepseek-pro", "object": "model", "created": int(time.time()), "owned_by": "deepseek"},
 
 
80
  }
81
 
82
  if model_id in models:
 
97
 
98
  prompt = request.messages[-1].content
99
 
100
+ model = request.model
 
 
101
 
102
  account = await manager.acquire()
103
 
 
368
  <div class="panel-body">
369
  <div class="row">
370
  <select id="model" style="flex:1">
371
+ <option value="deepseek-flash">deepseek-flash</option>
372
+ <option value="deepseek-pro">deepseek-pro</option>
 
 
373
  </select>
374
  <label style="font-size:11px;color:var(--dim);display:flex;align-items:center;gap:4px;white-space:nowrap">
375
  <input type="checkbox" id="stream" checked> 流式