Spaces:
Running
Running
:zap: [Enhance] Update available models description
Browse files- apis/chat_api.py +5 -4
apis/chat_api.py
CHANGED
@@ -24,18 +24,19 @@ class ChatAPIApp:
|
|
24 |
self.setup_routes()
|
25 |
|
26 |
def get_available_models(self):
|
|
|
27 |
self.available_models = [
|
28 |
{
|
29 |
"id": "mixtral-8x7b",
|
30 |
-
"description": "[Mixtral-8x7B-Instruct-v0.1]: https://huggingface.co/mistralai/Mixtral-8x7B-Instruct-v0.1",
|
31 |
},
|
32 |
{
|
33 |
"id": "mistral-7b",
|
34 |
-
"description": "[Mistral-7B-Instruct-v0.2]: https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2",
|
35 |
},
|
36 |
{
|
37 |
"id": "openchat-3.5",
|
38 |
-
"description": "[
|
39 |
},
|
40 |
]
|
41 |
return self.available_models
|
@@ -64,7 +65,7 @@ class ChatAPIApp:
|
|
64 |
description="(float) Temperature",
|
65 |
)
|
66 |
max_tokens: int = Field(
|
67 |
-
default=
|
68 |
description="(int) Max tokens",
|
69 |
)
|
70 |
stream: bool = Field(
|
|
|
24 |
self.setup_routes()
|
25 |
|
26 |
def get_available_models(self):
|
27 |
+
# ANCHOR[id=available-models]: Available models
|
28 |
self.available_models = [
|
29 |
{
|
30 |
"id": "mixtral-8x7b",
|
31 |
+
"description": "[mistralai/Mixtral-8x7B-Instruct-v0.1]: https://huggingface.co/mistralai/Mixtral-8x7B-Instruct-v0.1",
|
32 |
},
|
33 |
{
|
34 |
"id": "mistral-7b",
|
35 |
+
"description": "[mistralai/Mistral-7B-Instruct-v0.2]: https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2",
|
36 |
},
|
37 |
{
|
38 |
"id": "openchat-3.5",
|
39 |
+
"description": "[openchat/openchat-3.5-1210]: https://huggingface.co/openchat/openchat-3.5-1210",
|
40 |
},
|
41 |
]
|
42 |
return self.available_models
|
|
|
65 |
description="(float) Temperature",
|
66 |
)
|
67 |
max_tokens: int = Field(
|
68 |
+
default=4096,
|
69 |
description="(int) Max tokens",
|
70 |
)
|
71 |
stream: bool = Field(
|