Skip to content

Commit d3743fd

Browse files
committed
fix: ruff
1 parent 0fc8a3c commit d3743fd

File tree

2 files changed

+6
-6
lines changed

2 files changed

+6
-6
lines changed

src/mcp_scan/models.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
11
from datetime import datetime
22
from hashlib import md5
3+
from itertools import chain
34
from typing import Any, Literal, TypeAlias
45

56
from mcp.types import Prompt, Resource, Tool
6-
from pydantic import BaseModel, ConfigDict, RootModel, field_serializer, field_validator, model_serializer
7+
from pydantic import BaseModel, ConfigDict, RootModel, field_serializer, field_validator
78

89
Entity: TypeAlias = Prompt | Resource | Tool
910

@@ -178,4 +179,4 @@ class ScanPathResult(BaseModel):
178179

179180
@property
180181
def entities(self) -> list[Entity]:
181-
return sum([server.entities for server in self.servers], start=[],)
182+
return list(chain.from_iterable(server.entities for server in self.servers))

src/mcp_scan/verify_api.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,3 @@
1-
import ast
2-
import json
3-
import rich
41
import aiohttp
52

63
from .models import EntityScanResult, ScanPathResult
@@ -27,5 +24,7 @@ async def verify_server(scan_path: ScanPathResult, base_url: str) -> ScanPathRes
2724
errstr = ""
2825
result = scan_path.model_copy(deep=True)
2926
for server in result.servers:
30-
server.result = [EntityScanResult(status="could not reach verification server " + errstr) for _ in server.entities]
27+
server.result = [
28+
EntityScanResult(status="could not reach verification server " + errstr) for _ in server.entities
29+
]
3130
return result

0 commit comments

Comments
 (0)