Hansimov commited on
Commit
bb9b91b
1 Parent(s): 84cebc2

:gem: [Feature] ChatAPI: Support /v1 prefix

Browse files
Files changed (1) hide show
  1. apis/chat_api.py +20 -20
apis/chat_api.py CHANGED
@@ -5,7 +5,6 @@ from sse_starlette.sse import EventSourceResponse
5
  from conversations import (
6
  ConversationConnector,
7
  ConversationCreator,
8
- ConversationSession,
9
  MessageComposer,
10
  )
11
 
@@ -136,25 +135,26 @@ class ChatAPIApp:
136
  )
137
 
138
  def setup_routes(self):
139
- self.app.get(
140
- "/models",
141
- summary="Get available models",
142
- )(self.get_available_models)
143
-
144
- self.app.post(
145
- "/create",
146
- summary="Create a conversation session",
147
- )(self.create_conversation_session)
148
-
149
- self.app.post(
150
- "/chat",
151
- summary="Chat in conversation session",
152
- )(self.chat)
153
-
154
- self.app.post(
155
- "/chat/completions",
156
- summary="Chat completions in conversation session",
157
- )(self.chat_completions)
 
158
 
159
 
160
  app = ChatAPIApp().app
 
5
  from conversations import (
6
  ConversationConnector,
7
  ConversationCreator,
 
8
  MessageComposer,
9
  )
10
 
 
135
  )
136
 
137
  def setup_routes(self):
138
+ for prefix in ["", "/v1"]:
139
+ self.app.get(
140
+ prefix + "/models",
141
+ summary="Get available models",
142
+ )(self.get_available_models)
143
+
144
+ self.app.post(
145
+ prefix + "/create",
146
+ summary="Create a conversation session",
147
+ )(self.create_conversation_session)
148
+
149
+ self.app.post(
150
+ prefix + "/chat",
151
+ summary="Chat in conversation session",
152
+ )(self.chat)
153
+
154
+ self.app.post(
155
+ prefix + "/chat/completions",
156
+ summary="Chat completions in conversation session",
157
+ )(self.chat_completions)
158
 
159
 
160
  app = ChatAPIApp().app