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",
|