Commit ·
9d82622
1
Parent(s): 5db64d1
[NOTICKET] fixed single source into multiple sources
Browse files- 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
|