sofhiaazzhr commited on
Commit
9d82622
·
1 Parent(s): 5db64d1

[NOTICKET] fixed single source into multiple sources

Browse files
Files changed (1) hide show
  1. src/api/v1/chat.py +3 -2
src/api/v1/chat.py CHANGED
@@ -63,7 +63,7 @@ def _extract_sources(results: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
63
  for result in results:
64
  if "document_id" in result["metadata"]["data"]:
65
  meta = result["metadata"]
66
- key = (meta.get("document_id"), meta.get("page_label"))
67
  if key not in seen:
68
  seen.add(key)
69
  sources.append({
@@ -73,7 +73,7 @@ def _extract_sources(results: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
73
  })
74
  else:
75
  meta = result["metadata"]
76
- key = (meta.get("table_name"), meta.get("column_name"))
77
  if key not in seen:
78
  seen.add(key)
79
  table_name = meta.get("data", {}).get("table_name")
@@ -192,6 +192,7 @@ async def chat_stream(request: ChatRequest, db: AsyncSession = Depends(get_db)):
192
  query=search_query,
193
  user_id=request.user_id,
194
  db=db,
 
195
  )
196
  else:
197
  raw_results = await retrieval_task
 
63
  for result in results:
64
  if "document_id" in result["metadata"]["data"]:
65
  meta = result["metadata"]
66
+ key = (meta.get("data", {}).get("document_id"), meta.get("data", {}).get("page_label"))
67
  if key not in seen:
68
  seen.add(key)
69
  sources.append({
 
73
  })
74
  else:
75
  meta = result["metadata"]
76
+ key = (meta.get("data", {}).get("table_name"), meta.get("data", {}).get("column_name"))
77
  if key not in seen:
78
  seen.add(key)
79
  table_name = meta.get("data", {}).get("table_name")
 
192
  query=search_query,
193
  user_id=request.user_id,
194
  db=db,
195
+ source_hint=intent_result.get("source_hint", "both"),
196
  )
197
  else:
198
  raw_results = await retrieval_task