apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-04-25 16:56:22 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'MODSetter:main' into main

Browse source
This commit is contained in:
Ritik Prajapat 2025-05-12 09:36:47 +05:30 • committed by GitHub
commit c1acd9cfaf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 132 additions and 65 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,4 +1,4 @@
from typing import AsyncGenerator, List, Union
from typing import Any, AsyncGenerator, List, Union
from uuid import UUID
from app.agents.researcher.graph import graph as researcher_graph
@ -13,7 +13,8 @@ async def stream_connector_search_results(
search_space_id: int,
session: AsyncSession,
research_mode: str,
selected_connectors: List[str]
selected_connectors: List[str],
langchain_chat_history: List[Any]
) -> AsyncGenerator[str, None]:
"""
Stream connector search results to the client
@ -53,7 +54,8 @@ async def stream_connector_search_results(
# Initialize state with database session and streaming service
initial_state = State(
db_session=session,
streaming_service=streaming_service
streaming_service=streaming_service,
chat_history=langchain_chat_history
)
# Run the graph directly