vrr/SurfSense
2
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2025-09-13 23:59:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

feat: discord knowledge retrieval

Browse source
This commit is contained in:
Muhamad Aji Wibisono 2025-06-02 18:43:32 +07:00
parent 158976e802
commit 1d67a87b82
5 changed files with 124 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

17
surfsense_backend/app/agents/researcher/nodes.py
View file

@ -400,6 +400,23 @@ async def fetch_relevant_documents(
if streaming_service and writer:
streaming_service.only_update_terminal(f"🔗 Found {len(linkup_chunks)} Linkup results related to your query")
writer({"yeild_value": streaming_service._format_annotations()})
elif connector == "DISCORD_CONNECTOR":
source_object, discord_chunks = await connector_service.search_discord(
user_query=reformulated_query,
user_id=user_id,
search_space_id=search_space_id,
top_k=top_k,
search_mode=search_mode
)
# Add to sources and raw documents
if source_object:
all_sources.append(source_object)
all_raw_documents.extend(discord_chunks)
# Stream found document count
if streaming_service and writer:
streaming_service.only_update_terminal(f"🗨️ Found {len(discord_chunks)} Discord messages related to your query")
writer({"yeild_value": streaming_service._format_annotations()})
except Exception as e: