mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-23 19:05:16 +02:00
add Luma list events, read event, create event tools
This commit is contained in:
parent
49f8d1abd4
commit
ba8e3133b9
5 changed files with 355 additions and 0 deletions
15
surfsense_backend/app/agents/new_chat/tools/luma/__init__.py
Normal file
15
surfsense_backend/app/agents/new_chat/tools/luma/__init__.py
Normal file
|
|
@ -0,0 +1,15 @@
|
||||||
|
from app.agents.new_chat.tools.luma.create_event import (
|
||||||
|
create_create_luma_event_tool,
|
||||||
|
)
|
||||||
|
from app.agents.new_chat.tools.luma.list_events import (
|
||||||
|
create_list_luma_events_tool,
|
||||||
|
)
|
||||||
|
from app.agents.new_chat.tools.luma.read_event import (
|
||||||
|
create_read_luma_event_tool,
|
||||||
|
)
|
||||||
|
|
||||||
|
__all__ = [
|
||||||
|
"create_create_luma_event_tool",
|
||||||
|
"create_list_luma_events_tool",
|
||||||
|
"create_read_luma_event_tool",
|
||||||
|
]
|
||||||
42
surfsense_backend/app/agents/new_chat/tools/luma/_auth.py
Normal file
42
surfsense_backend/app/agents/new_chat/tools/luma/_auth.py
Normal file
|
|
@ -0,0 +1,42 @@
|
||||||
|
"""Shared auth helper for Luma agent tools."""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
from sqlalchemy.future import select
|
||||||
|
|
||||||
|
from app.db import SearchSourceConnector, SearchSourceConnectorType
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
LUMA_API = "https://public-api.luma.com/v1"
|
||||||
|
|
||||||
|
|
||||||
|
async def get_luma_connector(
|
||||||
|
db_session: AsyncSession,
|
||||||
|
search_space_id: int,
|
||||||
|
user_id: str,
|
||||||
|
) -> SearchSourceConnector | None:
|
||||||
|
result = await db_session.execute(
|
||||||
|
select(SearchSourceConnector).filter(
|
||||||
|
SearchSourceConnector.search_space_id == search_space_id,
|
||||||
|
SearchSourceConnector.user_id == user_id,
|
||||||
|
SearchSourceConnector.connector_type == SearchSourceConnectorType.LUMA_CONNECTOR,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
return result.scalars().first()
|
||||||
|
|
||||||
|
|
||||||
|
def get_api_key(connector: SearchSourceConnector) -> str:
|
||||||
|
"""Extract the API key from connector config (handles both key names)."""
|
||||||
|
key = connector.config.get("api_key") or connector.config.get("LUMA_API_KEY")
|
||||||
|
if not key:
|
||||||
|
raise ValueError("Luma API key not found in connector config.")
|
||||||
|
return key
|
||||||
|
|
||||||
|
|
||||||
|
def luma_headers(api_key: str) -> dict[str, str]:
|
||||||
|
return {
|
||||||
|
"Content-Type": "application/json",
|
||||||
|
"x-luma-api-key": api_key,
|
||||||
|
}
|
||||||
116
surfsense_backend/app/agents/new_chat/tools/luma/create_event.py
Normal file
116
surfsense_backend/app/agents/new_chat/tools/luma/create_event.py
Normal file
|
|
@ -0,0 +1,116 @@
|
||||||
|
import logging
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
import httpx
|
||||||
|
from langchain_core.tools import tool
|
||||||
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
|
from app.agents.new_chat.tools.hitl import request_approval
|
||||||
|
|
||||||
|
from ._auth import LUMA_API, get_api_key, get_luma_connector, luma_headers
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
def create_create_luma_event_tool(
|
||||||
|
db_session: AsyncSession | None = None,
|
||||||
|
search_space_id: int | None = None,
|
||||||
|
user_id: str | None = None,
|
||||||
|
):
|
||||||
|
@tool
|
||||||
|
async def create_luma_event(
|
||||||
|
name: str,
|
||||||
|
start_at: str,
|
||||||
|
end_at: str,
|
||||||
|
description: str | None = None,
|
||||||
|
timezone: str = "UTC",
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
"""Create a new event on Luma.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
name: The event title.
|
||||||
|
start_at: Start time in ISO 8601 format (e.g. "2026-05-01T18:00:00").
|
||||||
|
end_at: End time in ISO 8601 format (e.g. "2026-05-01T20:00:00").
|
||||||
|
description: Optional event description (markdown supported).
|
||||||
|
timezone: Timezone string (default "UTC", e.g. "America/New_York").
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary with status, event_id on success.
|
||||||
|
|
||||||
|
IMPORTANT:
|
||||||
|
- If status is "rejected", the user explicitly declined. Do NOT retry.
|
||||||
|
"""
|
||||||
|
if db_session is None or search_space_id is None or user_id is None:
|
||||||
|
return {"status": "error", "message": "Luma tool not properly configured."}
|
||||||
|
|
||||||
|
try:
|
||||||
|
connector = await get_luma_connector(db_session, search_space_id, user_id)
|
||||||
|
if not connector:
|
||||||
|
return {"status": "error", "message": "No Luma connector found."}
|
||||||
|
|
||||||
|
result = request_approval(
|
||||||
|
action_type="luma_create_event",
|
||||||
|
tool_name="create_luma_event",
|
||||||
|
params={
|
||||||
|
"name": name,
|
||||||
|
"start_at": start_at,
|
||||||
|
"end_at": end_at,
|
||||||
|
"description": description,
|
||||||
|
"timezone": timezone,
|
||||||
|
},
|
||||||
|
context={"connector_id": connector.id},
|
||||||
|
)
|
||||||
|
|
||||||
|
if result.rejected:
|
||||||
|
return {"status": "rejected", "message": "User declined. Event was not created."}
|
||||||
|
|
||||||
|
final_name = result.params.get("name", name)
|
||||||
|
final_start = result.params.get("start_at", start_at)
|
||||||
|
final_end = result.params.get("end_at", end_at)
|
||||||
|
final_desc = result.params.get("description", description)
|
||||||
|
final_tz = result.params.get("timezone", timezone)
|
||||||
|
|
||||||
|
api_key = get_api_key(connector)
|
||||||
|
headers = luma_headers(api_key)
|
||||||
|
|
||||||
|
body: dict[str, Any] = {
|
||||||
|
"name": final_name,
|
||||||
|
"start_at": final_start,
|
||||||
|
"end_at": final_end,
|
||||||
|
"timezone": final_tz,
|
||||||
|
}
|
||||||
|
if final_desc:
|
||||||
|
body["description_md"] = final_desc
|
||||||
|
|
||||||
|
async with httpx.AsyncClient(timeout=20.0) as client:
|
||||||
|
resp = await client.post(
|
||||||
|
f"{LUMA_API}/event/create",
|
||||||
|
headers=headers,
|
||||||
|
json=body,
|
||||||
|
)
|
||||||
|
|
||||||
|
if resp.status_code == 401:
|
||||||
|
return {"status": "auth_error", "message": "Luma API key is invalid.", "connector_type": "luma"}
|
||||||
|
if resp.status_code == 403:
|
||||||
|
return {"status": "error", "message": "Luma Plus subscription required to create events via API."}
|
||||||
|
if resp.status_code not in (200, 201):
|
||||||
|
return {"status": "error", "message": f"Luma API error: {resp.status_code} — {resp.text[:200]}"}
|
||||||
|
|
||||||
|
data = resp.json()
|
||||||
|
event_id = data.get("api_id") or data.get("event", {}).get("api_id")
|
||||||
|
|
||||||
|
return {
|
||||||
|
"status": "success",
|
||||||
|
"event_id": event_id,
|
||||||
|
"message": f"Event '{final_name}' created on Luma.",
|
||||||
|
}
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
from langgraph.errors import GraphInterrupt
|
||||||
|
|
||||||
|
if isinstance(e, GraphInterrupt):
|
||||||
|
raise
|
||||||
|
logger.error("Error creating Luma event: %s", e, exc_info=True)
|
||||||
|
return {"status": "error", "message": "Failed to create Luma event."}
|
||||||
|
|
||||||
|
return create_luma_event
|
||||||
100
surfsense_backend/app/agents/new_chat/tools/luma/list_events.py
Normal file
100
surfsense_backend/app/agents/new_chat/tools/luma/list_events.py
Normal file
|
|
@ -0,0 +1,100 @@
|
||||||
|
import logging
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
import httpx
|
||||||
|
from langchain_core.tools import tool
|
||||||
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
|
from ._auth import LUMA_API, get_api_key, get_luma_connector, luma_headers
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
def create_list_luma_events_tool(
|
||||||
|
db_session: AsyncSession | None = None,
|
||||||
|
search_space_id: int | None = None,
|
||||||
|
user_id: str | None = None,
|
||||||
|
):
|
||||||
|
@tool
|
||||||
|
async def list_luma_events(
|
||||||
|
max_results: int = 25,
|
||||||
|
) -> dict[str, Any]:
|
||||||
|
"""List upcoming and recent Luma events.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
max_results: Maximum events to return (default 25, max 50).
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary with status and a list of events including
|
||||||
|
event_id, name, start_at, end_at, location, url.
|
||||||
|
"""
|
||||||
|
if db_session is None or search_space_id is None or user_id is None:
|
||||||
|
return {"status": "error", "message": "Luma tool not properly configured."}
|
||||||
|
|
||||||
|
max_results = min(max_results, 50)
|
||||||
|
|
||||||
|
try:
|
||||||
|
connector = await get_luma_connector(db_session, search_space_id, user_id)
|
||||||
|
if not connector:
|
||||||
|
return {"status": "error", "message": "No Luma connector found."}
|
||||||
|
|
||||||
|
api_key = get_api_key(connector)
|
||||||
|
headers = luma_headers(api_key)
|
||||||
|
|
||||||
|
all_entries: list[dict] = []
|
||||||
|
cursor = None
|
||||||
|
|
||||||
|
async with httpx.AsyncClient(timeout=20.0) as client:
|
||||||
|
while len(all_entries) < max_results:
|
||||||
|
params: dict[str, Any] = {"limit": min(100, max_results - len(all_entries))}
|
||||||
|
if cursor:
|
||||||
|
params["cursor"] = cursor
|
||||||
|
|
||||||
|
resp = await client.get(
|
||||||
|
f"{LUMA_API}/calendar/list-events",
|
||||||
|
headers=headers,
|
||||||
|
params=params,
|
||||||
|
)
|
||||||
|
|
||||||
|
if resp.status_code == 401:
|
||||||
|
return {"status": "auth_error", "message": "Luma API key is invalid.", "connector_type": "luma"}
|
||||||
|
if resp.status_code != 200:
|
||||||
|
return {"status": "error", "message": f"Luma API error: {resp.status_code}"}
|
||||||
|
|
||||||
|
data = resp.json()
|
||||||
|
entries = data.get("entries", [])
|
||||||
|
if not entries:
|
||||||
|
break
|
||||||
|
all_entries.extend(entries)
|
||||||
|
|
||||||
|
next_cursor = data.get("next_cursor")
|
||||||
|
if not next_cursor:
|
||||||
|
break
|
||||||
|
cursor = next_cursor
|
||||||
|
|
||||||
|
events = []
|
||||||
|
for entry in all_entries[:max_results]:
|
||||||
|
ev = entry.get("event", {})
|
||||||
|
geo = ev.get("geo_info", {})
|
||||||
|
events.append({
|
||||||
|
"event_id": entry.get("api_id"),
|
||||||
|
"name": ev.get("name", "Untitled"),
|
||||||
|
"start_at": ev.get("start_at", ""),
|
||||||
|
"end_at": ev.get("end_at", ""),
|
||||||
|
"timezone": ev.get("timezone", ""),
|
||||||
|
"location": geo.get("name", ""),
|
||||||
|
"url": ev.get("url", ""),
|
||||||
|
"visibility": ev.get("visibility", ""),
|
||||||
|
})
|
||||||
|
|
||||||
|
return {"status": "success", "events": events, "total": len(events)}
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
from langgraph.errors import GraphInterrupt
|
||||||
|
|
||||||
|
if isinstance(e, GraphInterrupt):
|
||||||
|
raise
|
||||||
|
logger.error("Error listing Luma events: %s", e, exc_info=True)
|
||||||
|
return {"status": "error", "message": "Failed to list Luma events."}
|
||||||
|
|
||||||
|
return list_luma_events
|
||||||
|
|
@ -0,0 +1,82 @@
|
||||||
|
import logging
|
||||||
|
from typing import Any
|
||||||
|
|
||||||
|
import httpx
|
||||||
|
from langchain_core.tools import tool
|
||||||
|
from sqlalchemy.ext.asyncio import AsyncSession
|
||||||
|
|
||||||
|
from ._auth import LUMA_API, get_api_key, get_luma_connector, luma_headers
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
def create_read_luma_event_tool(
|
||||||
|
db_session: AsyncSession | None = None,
|
||||||
|
search_space_id: int | None = None,
|
||||||
|
user_id: str | None = None,
|
||||||
|
):
|
||||||
|
@tool
|
||||||
|
async def read_luma_event(event_id: str) -> dict[str, Any]:
|
||||||
|
"""Read detailed information about a specific Luma event.
|
||||||
|
|
||||||
|
Args:
|
||||||
|
event_id: The Luma event API ID (from list_luma_events).
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
Dictionary with status and full event details including
|
||||||
|
description, attendees count, meeting URL.
|
||||||
|
"""
|
||||||
|
if db_session is None or search_space_id is None or user_id is None:
|
||||||
|
return {"status": "error", "message": "Luma tool not properly configured."}
|
||||||
|
|
||||||
|
try:
|
||||||
|
connector = await get_luma_connector(db_session, search_space_id, user_id)
|
||||||
|
if not connector:
|
||||||
|
return {"status": "error", "message": "No Luma connector found."}
|
||||||
|
|
||||||
|
api_key = get_api_key(connector)
|
||||||
|
headers = luma_headers(api_key)
|
||||||
|
|
||||||
|
async with httpx.AsyncClient(timeout=15.0) as client:
|
||||||
|
resp = await client.get(
|
||||||
|
f"{LUMA_API}/events/{event_id}",
|
||||||
|
headers=headers,
|
||||||
|
)
|
||||||
|
|
||||||
|
if resp.status_code == 401:
|
||||||
|
return {"status": "auth_error", "message": "Luma API key is invalid.", "connector_type": "luma"}
|
||||||
|
if resp.status_code == 404:
|
||||||
|
return {"status": "not_found", "message": f"Event '{event_id}' not found."}
|
||||||
|
if resp.status_code != 200:
|
||||||
|
return {"status": "error", "message": f"Luma API error: {resp.status_code}"}
|
||||||
|
|
||||||
|
data = resp.json()
|
||||||
|
ev = data.get("event", data)
|
||||||
|
geo = ev.get("geo_info", {})
|
||||||
|
|
||||||
|
event_detail = {
|
||||||
|
"event_id": event_id,
|
||||||
|
"name": ev.get("name", ""),
|
||||||
|
"description": ev.get("description", ""),
|
||||||
|
"start_at": ev.get("start_at", ""),
|
||||||
|
"end_at": ev.get("end_at", ""),
|
||||||
|
"timezone": ev.get("timezone", ""),
|
||||||
|
"location_name": geo.get("name", ""),
|
||||||
|
"address": geo.get("address", ""),
|
||||||
|
"url": ev.get("url", ""),
|
||||||
|
"meeting_url": ev.get("meeting_url", ""),
|
||||||
|
"visibility": ev.get("visibility", ""),
|
||||||
|
"cover_url": ev.get("cover_url", ""),
|
||||||
|
}
|
||||||
|
|
||||||
|
return {"status": "success", "event": event_detail}
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
from langgraph.errors import GraphInterrupt
|
||||||
|
|
||||||
|
if isinstance(e, GraphInterrupt):
|
||||||
|
raise
|
||||||
|
logger.error("Error reading Luma event: %s", e, exc_info=True)
|
||||||
|
return {"status": "error", "message": "Failed to read Luma event."}
|
||||||
|
|
||||||
|
return read_luma_event
|
||||||
Loading…
Add table
Add a link
Reference in a new issue