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

Added Luma connector

Browse source
This commit is contained in:
samkul-swe 2025-09-28 14:59:10 -07:00
parent 8bbc8dba4e
commit 9d2b808e66
27 changed files with 1757 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -413,6 +413,41 @@ async def fetch_documents_by_ids(
else:
url = ""
elif doc_type == "LUMA_CONNECTOR":
# Extract Luma-specific metadata
event_id = metadata.get("event_id", "")
event_name = metadata.get("event_name", "Untitled Event")
event_url = metadata.get("event_url", "")
start_time = metadata.get("start_time", "")
location_name = metadata.get("location_name", "")
meeting_url = metadata.get("meeting_url", "")
title = f"Luma: {event_name}"
if start_time:
# Format the start time for display
try:
if "T" in start_time:
from datetime import datetime
start_dt = datetime.fromisoformat(
start_time.replace("Z", "+00:00")
)
formatted_time = start_dt.strftime("%Y-%m-%d %H:%M")
title += f" ({formatted_time})"
except Exception:
pass
description = (
doc.content[:100] + "..."
if len(doc.content) > 100
else doc.content
)
if location_name:
description += f" | Venue: {location_name}"
elif meeting_url:
description += f" | Online Event"
url = event_url if event_url else ""
elif doc_type == "EXTENSION":
# Extract Extension-specific metadata
webpage_title = metadata.get("VisitedWebPageTitle", doc.title)
@ -487,6 +522,7 @@ async def fetch_documents_by_ids(
"CONFLUENCE_CONNECTOR": "Confluence (Selected)",
"CLICKUP_CONNECTOR": "ClickUp (Selected)",
"AIRTABLE_CONNECTOR": "Airtable (Selected)",
"LUMA_CONNECTOR": "Luma Events (Selected)",
}
source_object = {
@ -1197,6 +1233,33 @@ async def fetch_relevant_documents(
}
)
elif connector == "LUMA_CONNECTOR":
(
source_object,
luma_chunks,
) = await connector_service.search_luma(
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(luma_chunks)
# Stream found document count
if streaming_service and writer:
writer(
{
"yield_value": streaming_service.format_terminal_info_delta(
f"🎯 Found {len(luma_chunks)} Luma events related to your query"
)
}
)
except Exception as e:
logging.error("Error in search_airtable: %s", traceback.format_exc())
error_message = f"Error searching connector {connector}: {e!s}"