ka1kuk commited on
Commit
d14782a
·
verified ·
1 Parent(s): 6b447ee

Update apis/chat_api.py

Browse files
Files changed (1) hide show
  1. 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",