chore: ran linting

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

View file

@ -23,7 +23,7 @@ async def test_search_google_drive_includes_legacy_composio_docs(
async with patched_session_factory() as session:
service = ConnectorService(session, search_space_id=space_id)
result_object, raw_docs = await service.search_google_drive(
_, raw_docs = await service.search_google_drive(
user_query="quarterly budget",
search_space_id=space_id,
top_k=10,
@ -55,7 +55,7 @@ async def test_search_files_does_not_include_google_types(
async with patched_session_factory() as session:
service = ConnectorService(session, search_space_id=space_id)
result_object, raw_docs = await service.search_files(
_, raw_docs = await service.search_files(
user_query="quarterly budget",
search_space_id=space_id,
top_k=10,

View file

@ -15,11 +15,11 @@ pytestmark = pytest.mark.unit
@patch("app.services.composio_service.ComposioService")
def test_returns_credentials_with_token_and_expiry(MockComposioService):
def test_returns_credentials_with_token_and_expiry(mock_composio_service):
"""build_composio_credentials returns a Credentials object with the Composio access token."""
mock_service = MagicMock()
mock_service.get_access_token.return_value = "fake-access-token"
MockComposioService.return_value = mock_service
mock_composio_service.return_value = mock_service
from app.utils.google_credentials import build_composio_credentials
@ -32,14 +32,14 @@ def test_returns_credentials_with_token_and_expiry(MockComposioService):
@patch("app.services.composio_service.ComposioService")
def test_refresh_handler_fetches_fresh_token(MockComposioService):
def test_refresh_handler_fetches_fresh_token(mock_composio_service):
"""The refresh_handler on the returned Credentials fetches a new token from Composio."""
mock_service = MagicMock()
mock_service.get_access_token.side_effect = [
"initial-token",
"refreshed-token",
]
MockComposioService.return_value = mock_service
mock_composio_service.return_value = mock_service
from app.utils.google_credentials import build_composio_credentials

View file

@ -208,7 +208,7 @@ async def test_drive_client_uses_prebuilt_composio_credentials(mock_build):
credentials=creds,
)
files, next_token, error = await client.list_files()
files, _next_token, error = await client.list_files()
assert error is None
assert files == []