Update apis/chat_api.py
Browse files- apis/chat_api.py +5 -0
apis/chat_api.py
CHANGED
|
@@ -243,6 +243,11 @@ class ChatAPIApp:
|
|
| 243 |
include_in_schema=include_in_schema,
|
| 244 |
)(self.chat_completions)
|
| 245 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 246 |
self.app.post(
|
| 247 |
prefix + "/embedding", # Use the specific prefix for this route
|
| 248 |
summary="Generate embeddings for the given texts",
|
|
|
|
| 243 |
include_in_schema=include_in_schema,
|
| 244 |
)(self.chat_completions)
|
| 245 |
|
| 246 |
+
if prefix in ["v1"]:
|
| 247 |
+
include_in_schema = True
|
| 248 |
+
else:
|
| 249 |
+
include_in_schema = False
|
| 250 |
+
|
| 251 |
self.app.post(
|
| 252 |
prefix + "/embedding", # Use the specific prefix for this route
|
| 253 |
summary="Generate embeddings for the given texts",
|