mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-31 19:45:15 +02:00
feat(web): enhance chat context and mention handling with connector support
This commit is contained in:
parent
701ae800b4
commit
a41b16b73e
15 changed files with 773 additions and 449 deletions
|
|
@ -1771,6 +1771,11 @@ async def handle_new_chat(
|
|||
if request.mentioned_documents
|
||||
else None
|
||||
)
|
||||
mentioned_connectors_payload = (
|
||||
[doc.model_dump() for doc in request.mentioned_connectors]
|
||||
if request.mentioned_connectors
|
||||
else None
|
||||
)
|
||||
|
||||
return StreamingResponse(
|
||||
stream_new_chat(
|
||||
|
|
@ -1782,6 +1787,8 @@ async def handle_new_chat(
|
|||
mentioned_document_ids=request.mentioned_document_ids,
|
||||
mentioned_surfsense_doc_ids=request.mentioned_surfsense_doc_ids,
|
||||
mentioned_folder_ids=request.mentioned_folder_ids,
|
||||
mentioned_connector_ids=request.mentioned_connector_ids,
|
||||
mentioned_connectors=mentioned_connectors_payload,
|
||||
mentioned_documents=mentioned_documents_payload,
|
||||
needs_history_bootstrap=thread.needs_history_bootstrap,
|
||||
thread_visibility=thread.visibility,
|
||||
|
|
@ -2258,6 +2265,11 @@ async def regenerate_response(
|
|||
if request.mentioned_documents
|
||||
else None
|
||||
)
|
||||
mentioned_connectors_payload = (
|
||||
[doc.model_dump() for doc in request.mentioned_connectors]
|
||||
if request.mentioned_connectors
|
||||
else None
|
||||
)
|
||||
try:
|
||||
async for chunk in stream_new_chat(
|
||||
user_query=str(user_query_to_use),
|
||||
|
|
@ -2268,6 +2280,8 @@ async def regenerate_response(
|
|||
mentioned_document_ids=request.mentioned_document_ids,
|
||||
mentioned_surfsense_doc_ids=request.mentioned_surfsense_doc_ids,
|
||||
mentioned_folder_ids=request.mentioned_folder_ids,
|
||||
mentioned_connector_ids=request.mentioned_connector_ids,
|
||||
mentioned_connectors=mentioned_connectors_payload,
|
||||
mentioned_documents=mentioned_documents_payload,
|
||||
checkpoint_id=target_checkpoint_id,
|
||||
needs_history_bootstrap=thread.needs_history_bootstrap,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue