Skip to content

Commit 1647415

Browse files
committed
fix: ignore missing imports for all optional packages in pyright
Signed-off-by: Luca Muscariello <muscariello@ieee.org>
1 parent d1a4852 commit 1647415

5 files changed

Lines changed: 29 additions & 15 deletions

File tree

src/a2a/client/transports/grpc.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55

66
try:
7-
import grpc
7+
import grpc # type: ignore[reportMissingModuleSource, reportMissingImports]
88
except ImportError as e:
99
raise ImportError(
1010
'A2AGrpcClient requires grpcio and grpcio-tools to be installed. '

src/a2a/server/request_handlers/grpc_handler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,10 @@
77

88

99
try:
10-
import grpc
11-
import grpc.aio
10+
import grpc # type: ignore[reportMissingModuleSource]
11+
import grpc.aio # type: ignore[reportMissingModuleSource]
1212

13-
from grpc.aio import Metadata
13+
from grpc.aio import Metadata # type: ignore[reportMissingModuleSource]
1414
except ImportError as e:
1515
raise ImportError(
1616
'GrpcHandler requires grpcio and grpcio-tools to be installed. '

src/a2a/server/tasks/database_push_notification_config_store.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,17 +8,19 @@
88

99

1010
try:
11-
from sqlalchemy import (
11+
from sqlalchemy import ( # type: ignore[reportMissingImports]
1212
Table,
1313
delete,
1414
select,
1515
)
16-
from sqlalchemy.ext.asyncio import (
16+
from sqlalchemy.ext.asyncio import ( # type: ignore[reportMissingImports]
1717
AsyncEngine,
1818
AsyncSession,
1919
async_sessionmaker,
2020
)
21-
from sqlalchemy.orm import class_mapper
21+
from sqlalchemy.orm import (
22+
class_mapper, # type: ignore[reportMissingImports]
23+
)
2224
except ImportError as e:
2325
raise ImportError(
2426
'DatabasePushNotificationConfigStore requires SQLAlchemy and a database driver. '
@@ -41,7 +43,7 @@
4143

4244

4345
if TYPE_CHECKING:
44-
from cryptography.fernet import Fernet
46+
from cryptography.fernet import Fernet # type: ignore[reportMissingImports]
4547

4648

4749
logger = logging.getLogger(__name__)

src/a2a/server/tasks/database_task_store.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,19 @@
44

55

66
try:
7-
from sqlalchemy import Table, delete, select
8-
from sqlalchemy.ext.asyncio import (
7+
from sqlalchemy import ( # type: ignore[reportMissingImports]
8+
Table,
9+
delete,
10+
select,
11+
)
12+
from sqlalchemy.ext.asyncio import ( # type: ignore[reportMissingImports]
913
AsyncEngine,
1014
AsyncSession,
1115
async_sessionmaker,
1216
)
13-
from sqlalchemy.orm import class_mapper
17+
from sqlalchemy.orm import (
18+
class_mapper, # type: ignore[reportMissingImports]
19+
)
1420
except ImportError as e:
1521
raise ImportError(
1622
'DatabaseTaskStore requires SQLAlchemy and a database driver. '

src/a2a/utils/telemetry.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -65,16 +65,22 @@ def internal_method(self):
6565

6666

6767
if TYPE_CHECKING:
68-
from opentelemetry.trace import SpanKind as SpanKindType
68+
from opentelemetry.trace import (
69+
SpanKind as SpanKindType, # type: ignore[reportMissingImports]
70+
)
6971
else:
7072
SpanKindType = object
7173

7274
logger = logging.getLogger(__name__)
7375

7476
try:
75-
from opentelemetry import trace
76-
from opentelemetry.trace import SpanKind as _SpanKind
77-
from opentelemetry.trace import StatusCode
77+
from opentelemetry import trace # type: ignore[reportMissingImports]
78+
from opentelemetry.trace import ( # type: ignore[reportMissingImports]
79+
SpanKind as _SpanKind,
80+
)
81+
from opentelemetry.trace import ( # type: ignore[reportMissingImports]
82+
StatusCode,
83+
)
7884

7985
except ImportError:
8086
logger.debug(

0 commit comments

Comments
 (0)