AlekhyaC2005 commited on
Commit
dc77970
·
1 Parent(s): 46fb1fc

change the user id to optional and with deafault value

Browse files
Files changed (1) hide show
  1. app/routes/chat.py +8 -4
app/routes/chat.py CHANGED
@@ -1,3 +1,5 @@
 
 
1
  from fastapi import APIRouter
2
  from fastapi.responses import StreamingResponse
3
  from pydantic import BaseModel
@@ -8,8 +10,10 @@ from app.utils.streaming import stream_response
8
  router = APIRouter()
9
 
10
  class ChatRequest(BaseModel):
11
- user_id: str
12
- message: str
 
 
13
 
14
  @router.post("/chat")
15
 
@@ -17,14 +21,14 @@ async def chat(request: ChatRequest):
17
 
18
  response = await generate_response(
19
  request.user_id,
20
- request.message
21
  )
22
 
23
  return StreamingResponse(
24
  stream_response(
25
  response,
26
  request.user_id,
27
- request.message
28
  ),
29
  media_type="text/plain"
30
  )
 
1
+ from typing import Optional
2
+
3
  from fastapi import APIRouter
4
  from fastapi.responses import StreamingResponse
5
  from pydantic import BaseModel
 
10
  router = APIRouter()
11
 
12
  class ChatRequest(BaseModel):
13
+
14
+ user_query: str
15
+
16
+ user_id: Optional[str] = "alekhya"
17
 
18
  @router.post("/chat")
19
 
 
21
 
22
  response = await generate_response(
23
  request.user_id,
24
+ request.user_query
25
  )
26
 
27
  return StreamingResponse(
28
  stream_response(
29
  response,
30
  request.user_id,
31
+ request.user_query
32
  ),
33
  media_type="text/plain"
34
  )