apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-06-06 20:15:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat: Document Selector in Chat.

Browse source 
- Still need improvements but lets use it first.
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2025-06-04 21:46:50 -07:00
parent e8a19c496b
commit d7bb31f894
12 changed files with 599 additions and 67 deletions
Download patch file Download diff file Expand all files Collapse all files

6
surfsense_backend/app/tasks/stream_connector_search_results.py
View file

@ -17,7 +17,8 @@ async def stream_connector_search_results(
research_mode: str,
selected_connectors: List[str],
langchain_chat_history: List[Any],
search_mode_str: str
search_mode_str: str,
document_ids_to_add_in_context: List[int]
) -> AsyncGenerator[str, None]:
"""
Stream connector search results to the client
@ -62,7 +63,8 @@ async def stream_connector_search_results(
"user_id": user_id_str,
"search_space_id": search_space_id,
"search_mode": search_mode,
"research_mode": research_mode
"research_mode": research_mode,
"document_ids_to_add_in_context": document_ids_to_add_in_context
}
}
# Initialize state with database session and streaming service