PersonalPublic/gpt_chat_bot
1
0
Fork 0
mirror of https://github.com/Balshgit/gpt_chat_bot.git synced 2025-12-15 16:10:39 +03:00
Code Issues Packages Projects Releases Wiki Activity

add provider aura (#70)

Browse Source
This commit is contained in:
Dmitry Afanasyev
2023-12-21 23:11:36 +03:00
committed by GitHub
parent 31499a3861
commit 29204f1592
4 changed files with 68 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
bot_microservice/constants.py
View File

@@ -60,6 +60,7 @@ class ChatGptModelsEnum(StrEnum):
gpt_3_5_turbo_stream_GeekGpt = "gpt-3.5-turbo-stream-GeekGpt"
gpt_3_5_turbo_stream_gptforlove = "gpt-3.5-turbo-stream-gptforlove"
gpt_3_5_turbo_stream_fakeGpt = "gpt-3.5-turbo-stream-fakeGpt"
gpt_3_5_turbo_stream_aura = "gpt-3.5-turbo-stream-aura"
@classmethod
def values(cls) -> set[str]:
@@ -71,14 +72,14 @@ class ChatGptModelsEnum(StrEnum):
for model in ChatGptModelsEnum.values():
priority = 0
match model:
case "gpt-3-stream-binjie":
priority = 3
case "gpt-3.5-turbo-stream-yqcloud":
priority = 3
case "gpt-3.5-turbo-stream-GeekGpt":
priority = 2
case "gpt-3.5-turbo-stream-fakeGpt":
priority = 2
case "gpt-3-stream-binjie":
priority = 1
case "gpt-3.5-turbo-stream-yqcloud":
priority = 1
fields = {"model": model, "priority": priority}
models.append(fields)
return models