Add o1 and o1-2024-12-17 model support (#2090)

* Add o1 and o1-2024-12-17 model support

* patch api key selection

---------

Co-authored-by: 蒋翌琪 <jiangyiqi99@jiangyiqideMacBook-Pro.local>
Co-authored-by: binary-husky <qingxu.fu@outlook.com>
这个提交包含在:
YIQI JIANG
2024-12-19 22:32:57 +08:00
提交者 GitHub
父节点 8413fb15ba
当前提交 f60a12f8b4
共有 4 个文件被更改,包括 28 次插入3 次删除

查看文件

@@ -79,7 +79,7 @@ def select_api_key(keys, llm_model):
key_list = keys.split(',')
if llm_model.startswith('gpt-') or llm_model.startswith('chatgpt-') or \
llm_model.startswith('one-api-') or llm_model.startswith('o1-'):
llm_model.startswith('one-api-') or llm_model == 'o1' or llm_model.startswith('o1-'):
for k in key_list:
if is_openai_api_key(k): avail_key_list.append(k)