activate FakeGpt (#69)

This commit is contained in:
Dmitry Afanasyev
2023-12-19 22:49:41 +03:00
committed by GitHub
parent a8276167d8
commit 31499a3861
5 changed files with 249 additions and 245 deletions

View File

@@ -59,6 +59,7 @@ class ChatGptModelsEnum(StrEnum):
gpt_3_5_turbo_stream_chatGptAi = "gpt-3.5-turbo-stream-chatGptAi"
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"
@classmethod
def values(cls) -> set[str]:
@@ -70,14 +71,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 "llama2":
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