Skip to content

Commit 7f3c636

Browse files
⚡ fix: use async httpx client in MCP handler
1 parent 1a5ab27 commit 7f3c636

1 file changed

Lines changed: 20 additions & 17 deletions

File tree

main.py

Lines changed: 20 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -183,11 +183,12 @@ async def mcp_post_handler(fastapi_request: Request):
183183
f"{fastapi_request.url.scheme}://{fastapi_request.url.netloc}"
184184
)
185185

186-
response = httpx.get(
187-
f"{base_url}/v1/check-email/{email}",
188-
follow_redirects=True,
189-
timeout=30.0,
190-
)
186+
async with httpx.AsyncClient() as client:
187+
response = await client.get(
188+
f"{base_url}/v1/check-email/{email}",
189+
follow_redirects=True,
190+
timeout=30.0,
191+
)
191192
response.raise_for_status()
192193
result = response.json()
193194

@@ -233,12 +234,13 @@ async def mcp_post_handler(fastapi_request: Request):
233234
if token:
234235
params["token"] = token
235236

236-
response = httpx.get(
237-
f"{base_url}/v1/breach-analytics",
238-
params=params,
239-
follow_redirects=True,
240-
timeout=30.0,
241-
)
237+
async with httpx.AsyncClient() as client:
238+
response = await client.get(
239+
f"{base_url}/v1/breach-analytics",
240+
params=params,
241+
follow_redirects=True,
242+
timeout=30.0,
243+
)
242244
response.raise_for_status()
243245
result = response.json()
244246

@@ -283,12 +285,13 @@ async def mcp_post_handler(fastapi_request: Request):
283285
if tool_args.get("breach_id"):
284286
params["breach_id"] = tool_args.get("breach_id")
285287

286-
response = httpx.get(
287-
f"{base_url}/v1/breaches",
288-
params=params if params else None,
289-
follow_redirects=True,
290-
timeout=30.0,
291-
)
288+
async with httpx.AsyncClient() as client:
289+
response = await client.get(
290+
f"{base_url}/v1/breaches",
291+
params=params if params else None,
292+
follow_redirects=True,
293+
timeout=30.0,
294+
)
292295
response.raise_for_status()
293296
result = response.json()
294297

0 commit comments

Comments
 (0)