Spaces:
Running
Running
Jeremiah Lowin commited on
Commit ·
7fb0010
1
Parent(s): fa6e614
Improve type inference
Browse files
src/fastmcp/client/client.py
CHANGED
|
@@ -125,8 +125,6 @@ class Client(Generic[ClientTransportT]):
|
|
| 125 |
instance = super().__new__(cls)
|
| 126 |
return instance
|
| 127 |
|
| 128 |
-
transport: ClientTransportT
|
| 129 |
-
|
| 130 |
def __init__(
|
| 131 |
self,
|
| 132 |
transport: ClientTransportT
|
|
@@ -145,7 +143,7 @@ class Client(Generic[ClientTransportT]):
|
|
| 145 |
timeout: datetime.timedelta | float | int | None = None,
|
| 146 |
init_timeout: datetime.timedelta | float | int | None = None,
|
| 147 |
):
|
| 148 |
-
self.transport = infer_transport(transport)
|
| 149 |
self._session: ClientSession | None = None
|
| 150 |
self._exit_stack: AsyncExitStack | None = None
|
| 151 |
self._nesting_counter: int = 0
|
|
|
|
| 125 |
instance = super().__new__(cls)
|
| 126 |
return instance
|
| 127 |
|
|
|
|
|
|
|
| 128 |
def __init__(
|
| 129 |
self,
|
| 130 |
transport: ClientTransportT
|
|
|
|
| 143 |
timeout: datetime.timedelta | float | int | None = None,
|
| 144 |
init_timeout: datetime.timedelta | float | int | None = None,
|
| 145 |
):
|
| 146 |
+
self.transport = cast(ClientTransportT, infer_transport(transport))
|
| 147 |
self._session: ClientSession | None = None
|
| 148 |
self._exit_stack: AsyncExitStack | None = None
|
| 149 |
self._nesting_counter: int = 0
|