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

chore: ran linting

Browse source
This commit is contained in:
Anish Sarkar 2026-03-22 00:43:53 +05:30
parent 44f59bfb61
commit 2bc6a0c3bc
24 changed files with 166 additions and 160 deletions
Download patch file Download diff file Expand all files Collapse all files

6
surfsense_backend/app/agents/new_chat/tools/gmail/create_draft.py
View file

@ -144,7 +144,7 @@ def create_create_gmail_draft_tool(
from app.db import SearchSourceConnector, SearchSourceConnectorType
_GMAIL_TYPES = [
_gmail_types = [
SearchSourceConnectorType.GOOGLE_GMAIL_CONNECTOR,
SearchSourceConnectorType.COMPOSIO_GMAIL_CONNECTOR,
]
@ -155,7 +155,7 @@ def create_create_gmail_draft_tool(
SearchSourceConnector.id == final_connector_id,
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_GMAIL_TYPES),
SearchSourceConnector.connector_type.in_(_gmail_types),
)
)
connector = result.scalars().first()
@ -170,7 +170,7 @@ def create_create_gmail_draft_tool(
select(SearchSourceConnector).filter(
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_GMAIL_TYPES),
SearchSourceConnector.connector_type.in_(_gmail_types),
)
)
connector = result.scalars().first()

6
surfsense_backend/app/agents/new_chat/tools/gmail/send_email.py
View file

@ -145,7 +145,7 @@ def create_send_gmail_email_tool(
from app.db import SearchSourceConnector, SearchSourceConnectorType
_GMAIL_TYPES = [
_gmail_types = [
SearchSourceConnectorType.GOOGLE_GMAIL_CONNECTOR,
SearchSourceConnectorType.COMPOSIO_GMAIL_CONNECTOR,
]
@ -156,7 +156,7 @@ def create_send_gmail_email_tool(
SearchSourceConnector.id == final_connector_id,
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_GMAIL_TYPES),
SearchSourceConnector.connector_type.in_(_gmail_types),
)
)
connector = result.scalars().first()
@ -171,7 +171,7 @@ def create_send_gmail_email_tool(
select(SearchSourceConnector).filter(
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_GMAIL_TYPES),
SearchSourceConnector.connector_type.in_(_gmail_types),
)
)
connector = result.scalars().first()

4
surfsense_backend/app/agents/new_chat/tools/gmail/trash_email.py
View file

@ -162,7 +162,7 @@ def create_trash_gmail_email_tool(
from app.db import SearchSourceConnector, SearchSourceConnectorType
_GMAIL_TYPES = [
_gmail_types = [
SearchSourceConnectorType.GOOGLE_GMAIL_CONNECTOR,
SearchSourceConnectorType.COMPOSIO_GMAIL_CONNECTOR,
]
@ -172,7 +172,7 @@ def create_trash_gmail_email_tool(
SearchSourceConnector.id == final_connector_id,
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_GMAIL_TYPES),
SearchSourceConnector.connector_type.in_(_gmail_types),
)
)
connector = result.scalars().first()

8
surfsense_backend/app/agents/new_chat/tools/gmail/update_draft.py
View file

@ -192,7 +192,7 @@ def create_update_gmail_draft_tool(
from app.db import SearchSourceConnector, SearchSourceConnectorType
_GMAIL_TYPES = [
_gmail_types = [
SearchSourceConnectorType.GOOGLE_GMAIL_CONNECTOR,
SearchSourceConnectorType.COMPOSIO_GMAIL_CONNECTOR,
]
@ -202,7 +202,7 @@ def create_update_gmail_draft_tool(
SearchSourceConnector.id == final_connector_id,
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_GMAIL_TYPES),
SearchSourceConnector.connector_type.in_(_gmail_types),
)
)
connector = result.scalars().first()
@ -419,7 +419,9 @@ async def _find_draft_id_by_message(gmail_service: Any, message_id: str) -> str
response = await asyncio.get_event_loop().run_in_executor(
None,
lambda: gmail_service.users().drafts().list(**kwargs).execute(),
lambda kwargs=kwargs: (
gmail_service.users().drafts().list(**kwargs).execute()
),
)
for draft in response.get("drafts", []):

6
surfsense_backend/app/agents/new_chat/tools/google_calendar/create_event.py
View file

@ -155,7 +155,7 @@ def create_create_calendar_event_tool(
from app.db import SearchSourceConnector, SearchSourceConnectorType
_CALENDAR_TYPES = [
_calendar_types = [
SearchSourceConnectorType.GOOGLE_CALENDAR_CONNECTOR,
SearchSourceConnectorType.COMPOSIO_GOOGLE_CALENDAR_CONNECTOR,
]
@ -166,7 +166,7 @@ def create_create_calendar_event_tool(
SearchSourceConnector.id == final_connector_id,
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_CALENDAR_TYPES),
SearchSourceConnector.connector_type.in_(_calendar_types),
)
)
connector = result.scalars().first()
@ -181,7 +181,7 @@ def create_create_calendar_event_tool(
select(SearchSourceConnector).filter(
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_CALENDAR_TYPES),
SearchSourceConnector.connector_type.in_(_calendar_types),
)
)
connector = result.scalars().first()

4
surfsense_backend/app/agents/new_chat/tools/google_calendar/delete_event.py
View file

@ -161,7 +161,7 @@ def create_delete_calendar_event_tool(
from app.db import SearchSourceConnector, SearchSourceConnectorType
_CALENDAR_TYPES = [
_calendar_types = [
SearchSourceConnectorType.GOOGLE_CALENDAR_CONNECTOR,
SearchSourceConnectorType.COMPOSIO_GOOGLE_CALENDAR_CONNECTOR,
]
@ -171,7 +171,7 @@ def create_delete_calendar_event_tool(
SearchSourceConnector.id == final_connector_id,
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_CALENDAR_TYPES),
SearchSourceConnector.connector_type.in_(_calendar_types),
)
)
connector = result.scalars().first()

4
surfsense_backend/app/agents/new_chat/tools/google_calendar/update_event.py
View file

@ -178,7 +178,7 @@ def create_update_calendar_event_tool(
from app.db import SearchSourceConnector, SearchSourceConnectorType
_CALENDAR_TYPES = [
_calendar_types = [
SearchSourceConnectorType.GOOGLE_CALENDAR_CONNECTOR,
SearchSourceConnectorType.COMPOSIO_GOOGLE_CALENDAR_CONNECTOR,
]
@ -188,7 +188,7 @@ def create_update_calendar_event_tool(
SearchSourceConnector.id == final_connector_id,
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_CALENDAR_TYPES),
SearchSourceConnector.connector_type.in_(_calendar_types),
)
)
connector = result.scalars().first()

6
surfsense_backend/app/agents/new_chat/tools/google_drive/create_file.py
View file

@ -166,7 +166,7 @@ def create_create_google_drive_file_tool(
from app.db import SearchSourceConnector, SearchSourceConnectorType
_DRIVE_TYPES = [
_drive_types = [
SearchSourceConnectorType.GOOGLE_DRIVE_CONNECTOR,
SearchSourceConnectorType.COMPOSIO_GOOGLE_DRIVE_CONNECTOR,
]
@ -177,7 +177,7 @@ def create_create_google_drive_file_tool(
SearchSourceConnector.id == final_connector_id,
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_DRIVE_TYPES),
SearchSourceConnector.connector_type.in_(_drive_types),
)
)
connector = result.scalars().first()
@ -192,7 +192,7 @@ def create_create_google_drive_file_tool(
select(SearchSourceConnector).filter(
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_DRIVE_TYPES),
SearchSourceConnector.connector_type.in_(_drive_types),
)
)
connector = result.scalars().first()

4
surfsense_backend/app/agents/new_chat/tools/google_drive/trash_file.py
View file

@ -162,7 +162,7 @@ def create_delete_google_drive_file_tool(
from app.db import SearchSourceConnector, SearchSourceConnectorType
_DRIVE_TYPES = [
_drive_types = [
SearchSourceConnectorType.GOOGLE_DRIVE_CONNECTOR,
SearchSourceConnectorType.COMPOSIO_GOOGLE_DRIVE_CONNECTOR,
]
@ -172,7 +172,7 @@ def create_delete_google_drive_file_tool(
SearchSourceConnector.id == final_connector_id,
SearchSourceConnector.search_space_id == search_space_id,
SearchSourceConnector.user_id == user_id,
SearchSourceConnector.connector_type.in_(_DRIVE_TYPES),
SearchSourceConnector.connector_type.in_(_drive_types),
)
)
connector = result.scalars().first()

5
surfsense_backend/app/agents/new_chat/tools/knowledge_base.py
View file

@ -9,6 +9,7 @@ This module provides:
"""
import asyncio
import contextlib
import json
import re
import time
@ -89,10 +90,8 @@ async def _browse_recent_documents(
doc_type_enums = []
for dt in type_list:
if isinstance(dt, str):
try:
with contextlib.suppress(KeyError):
doc_type_enums.append(DocumentType[dt])
except KeyError:
pass
else:
doc_type_enums.append(dt)
if not doc_type_enums: