Skip to content

Commit 17a2723

Browse files
committed
fix: PyCharm should correctly recognize the protocols
1 parent 34e3da6 commit 17a2723

3 files changed

Lines changed: 4 additions & 7 deletions

File tree

src/deepset_mcp/api/client.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,11 @@
33
from typing import Any, Self
44

55
from deepset_mcp.api.pipeline.resource import PipelineResource
6+
from deepset_mcp.api.protocols import AsyncClientProtocol
67
from deepset_mcp.api.transport import AsyncTransport, TransportProtocol, TransportResponse
78

89

9-
class AsyncDeepsetClient:
10+
class AsyncDeepsetClient(AsyncClientProtocol):
1011
"""Async Client for interacting with the deepset API."""
1112

1213
def __init__(

src/deepset_mcp/api/pipeline/resource.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,11 @@
22

33
from deepset_mcp.api.exceptions import UnexpectedAPIError
44
from deepset_mcp.api.pipeline.models import DeepsetPipeline, PipelineValidationResult, ValidationError
5-
from deepset_mcp.api.protocols import AsyncClientProtocol
5+
from deepset_mcp.api.protocols import AsyncClientProtocol, PipelineResourceProtocol
66
from deepset_mcp.api.transport import TransportResponse, raise_for_status
77

88

9-
class PipelineResource:
9+
class PipelineResource(PipelineResourceProtocol):
1010
"""Manages interactions with the deepset pipeline API."""
1111

1212
def __init__(

src/deepset_mcp/api/protocols.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,10 +40,6 @@ def pipelines(self, workspace: str) -> "PipelineResourceProtocol":
4040
class PipelineResourceProtocol(Protocol):
4141
"""Protocol defining the implementation for PipelineResource."""
4242

43-
def __init__(self, client: AsyncClientProtocol, workspace: str) -> None:
44-
"""Initialize a PipelineResource."""
45-
...
46-
4743
async def validate(self, yaml_config: str) -> PipelineValidationResult:
4844
"""Validate a pipeline's YAML configuration against the API."""
4945
...

0 commit comments

Comments
 (0)