khronoz commited on
Commit
8500091
·
1 Parent(s): 82d0321

Fix: query api not working

Browse files
backend/backend/app/api/routers/query.py CHANGED
@@ -28,8 +28,8 @@ class _ChatData(BaseModel):
28
  messages: List[_Message]
29
 
30
 
31
- @r.get("")
32
- async def search(
33
  request: Request,
34
  # Note: To support clients sending a JSON object using content-type "text/plain",
35
  # we need to use Depends(json_to_model(_ChatData)) here
@@ -42,18 +42,18 @@ async def search(
42
  status_code=status.HTTP_400_BAD_REQUEST,
43
  detail="No query provided",
44
  )
45
- query = data.messages.pop()
46
- if query.role != MessageRole.USER:
47
  raise HTTPException(
48
  status_code=status.HTTP_400_BAD_REQUEST,
49
  detail="Last message must be from user",
50
  )
51
  logger = logging.getLogger("uvicorn")
52
- logger.info(f"Query: {query}")
53
 
54
  # Query index
55
- query_engine = index.as_query_engine(streaming=True, similarity_top_k=1)
56
- response = query_engine.query(query)
57
 
58
  # stream response
59
  async def event_generator():
 
28
  messages: List[_Message]
29
 
30
 
31
+ @r.post("")
32
+ async def query(
33
  request: Request,
34
  # Note: To support clients sending a JSON object using content-type "text/plain",
35
  # we need to use Depends(json_to_model(_ChatData)) here
 
42
  status_code=status.HTTP_400_BAD_REQUEST,
43
  detail="No query provided",
44
  )
45
+ lastMessage = data.messages.pop()
46
+ if lastMessage.role != MessageRole.USER:
47
  raise HTTPException(
48
  status_code=status.HTTP_400_BAD_REQUEST,
49
  detail="Last message must be from user",
50
  )
51
  logger = logging.getLogger("uvicorn")
52
+ logger.info(f"Query: {lastMessage}")
53
 
54
  # Query index
55
+ query_engine = index.as_query_engine(streaming=True, similarity_top_k=5)
56
+ response = query_engine.query(lastMessage.content)
57
 
58
  # stream response
59
  async def event_generator():