chore: ran linting

This commit is contained in:
Anish Sarkar 2026-03-21 13:21:19 +05:30
parent de8841fb86
commit e37e6d2d18
19 changed files with 36 additions and 35 deletions

View file

@ -20,10 +20,10 @@ from langgraph.types import Checkpointer
from sqlalchemy.ext.asyncio import AsyncSession
from app.agents.new_chat.context import SurfSenseContextSchema
from app.agents.new_chat.llm_config import AgentConfig
from app.agents.new_chat.middleware.dedup_tool_calls import (
DedupHITLToolCallsMiddleware,
)
from app.agents.new_chat.llm_config import AgentConfig
from app.agents.new_chat.system_prompt import (
build_configurable_system_prompt,
build_surfsense_system_prompt,

View file

@ -120,6 +120,7 @@ def create_create_confluence_page_tool(
return {"status": "error", "message": "A space must be selected."}
from sqlalchemy.future import select
from app.db import SearchSourceConnector, SearchSourceConnectorType
actual_connector_id = final_connector_id

View file

@ -124,6 +124,7 @@ def create_delete_confluence_page_tool(
final_delete_from_kb = final_params.get("delete_from_kb", delete_from_kb)
from sqlalchemy.future import select
from app.db import SearchSourceConnector, SearchSourceConnectorType
if not final_connector_id:

View file

@ -136,6 +136,7 @@ def create_update_confluence_page_tool(
final_document_id = final_params.get("document_id", document_id)
from sqlalchemy.future import select
from app.db import SearchSourceConnector, SearchSourceConnectorType
if not final_connector_id:

View file

@ -1,15 +1,15 @@
from app.agents.new_chat.tools.google_calendar.create_event import (
create_create_calendar_event_tool,
)
from app.agents.new_chat.tools.google_calendar.update_event import (
create_update_calendar_event_tool,
)
from app.agents.new_chat.tools.google_calendar.delete_event import (
create_delete_calendar_event_tool,
)
from app.agents.new_chat.tools.google_calendar.update_event import (
create_update_calendar_event_tool,
)
__all__ = [
"create_create_calendar_event_tool",
"create_update_calendar_event_tool",
"create_delete_calendar_event_tool",
"create_update_calendar_event_tool",
]

View file

@ -128,6 +128,7 @@ def create_create_jira_issue_tool(
return {"status": "error", "message": "A project must be selected."}
from sqlalchemy.future import select
from app.db import SearchSourceConnector, SearchSourceConnectorType
actual_connector_id = final_connector_id

View file

@ -121,6 +121,7 @@ def create_delete_jira_issue_tool(
final_delete_from_kb = final_params.get("delete_from_kb", delete_from_kb)
from sqlalchemy.future import select
from app.db import SearchSourceConnector, SearchSourceConnectorType
if not final_connector_id:

View file

@ -131,6 +131,7 @@ def create_update_jira_issue_tool(
final_document_id = final_params.get("document_id", document_id)
from sqlalchemy.future import select
from app.db import SearchSourceConnector, SearchSourceConnectorType
if not final_connector_id:

View file

@ -45,6 +45,11 @@ from langchain_core.tools import BaseTool
from app.db import ChatVisibility
from .confluence import (
create_create_confluence_page_tool,
create_delete_confluence_page_tool,
create_update_confluence_page_tool,
)
from .display_image import create_display_image_tool
from .generate_image import create_generate_image_tool
from .gmail import (
@ -62,6 +67,11 @@ from .google_drive import (
create_create_google_drive_file_tool,
create_delete_google_drive_file_tool,
)
from .jira import (
create_create_jira_issue_tool,
create_delete_jira_issue_tool,
create_update_jira_issue_tool,
)
from .knowledge_base import create_search_knowledge_base_tool
from .linear import (
create_create_linear_issue_tool,
@ -70,16 +80,6 @@ from .linear import (
)
from .link_preview import create_link_preview_tool
from .mcp_tool import load_mcp_tools
from .jira import (
create_create_jira_issue_tool,
create_delete_jira_issue_tool,
create_update_jira_issue_tool,
)
from .confluence import (
create_create_confluence_page_tool,
create_delete_confluence_page_tool,
create_update_confluence_page_tool,
)
from .notion import (
create_create_notion_page_tool,
create_delete_notion_page_tool,

View file

@ -1,13 +1,12 @@
import asyncio
import contextlib
import logging
import re
from collections.abc import Awaitable, Callable
from typing import Any, TypeVar
from notion_client import AsyncClient
from notion_markdown import to_notion
from notion_client.errors import APIResponseError
from notion_markdown import to_notion
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.future import select

View file

@ -12,10 +12,9 @@ import httpx
from fastapi import APIRouter, Depends, HTTPException, Request
from fastapi.responses import RedirectResponse
from pydantic import ValidationError
from sqlalchemy import select
from sqlalchemy.exc import IntegrityError
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select
from sqlalchemy.orm.attributes import flag_modified
from app.config import config

View file

@ -12,10 +12,9 @@ import httpx
from fastapi import APIRouter, Depends, HTTPException, Request
from fastapi.responses import RedirectResponse
from pydantic import ValidationError
from sqlalchemy import select
from sqlalchemy.exc import IntegrityError
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select
from sqlalchemy.orm.attributes import flag_modified
from app.config import config

View file

@ -72,6 +72,7 @@ from app.tasks.connector_indexers import (
index_slack_messages,
)
from app.users import current_active_user
from app.utils.connector_naming import ensure_unique_connector_name
from app.utils.indexing_locks import (
acquire_connector_indexing_lock,
release_connector_indexing_lock,
@ -81,7 +82,6 @@ from app.utils.periodic_scheduler import (
delete_periodic_schedule,
update_periodic_schedule,
)
from app.utils.connector_naming import ensure_unique_connector_name
from app.utils.rbac import check_permission
# Set up logging

View file

@ -1,7 +1,7 @@
"""Shared Google OAuth credential utilities for native and Composio connectors."""
import logging
from datetime import datetime, timedelta, timezone
from datetime import UTC, datetime, timedelta
from google.oauth2.credentials import Credentials
@ -31,11 +31,11 @@ def build_composio_credentials(connected_account_id: str) -> Credentials:
def composio_refresh_handler(request, scopes):
fresh_token = service.get_access_token(connected_account_id)
expiry = datetime.now(timezone.utc).replace(tzinfo=None) + timedelta(minutes=55)
expiry = datetime.now(UTC).replace(tzinfo=None) + timedelta(minutes=55)
return fresh_token, expiry
return Credentials(
token=access_token,
expiry=datetime.now(timezone.utc).replace(tzinfo=None) + timedelta(minutes=55),
expiry=datetime.now(UTC).replace(tzinfo=None) + timedelta(minutes=55),
refresh_handler=composio_refresh_handler,
)