ceos/fastmcp
1
0
Fork 0
mirror of https://github.com/PrefectHQ/fastmcp.git synced 2026-08-22 21:44:18 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix: Client OAuth async_auth_flow() method causing MCP-SDK self.context.lock error. (#2644)

Browse source 
Co-authored-by: Jeremiah Lowin <153965+jlowin@users.noreply.github.com>
This commit is contained in:
Luke 2025-12-26 21:41:41 +01:00 • committed by GitHub
commit 650b103dbf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 19 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

32
src/fastmcp/client/auth/oauth.py
View file

@ -298,13 +298,16 @@ class OAuth(OAuthClientProvider):
# First attempt with potentially cached credentials
gen = super().async_auth_flow(request)
response = None
while True:
try:
# First iteration sends None, subsequent iterations send response
yielded_request = await gen.asend(response) # ty: ignore[invalid-argument-type]
response = yield yielded_request
except StopAsyncIteration:
break
try:
while True:
try:
# First iteration sends None, subsequent iterations send response
yielded_request = await gen.asend(response) # ty: ignore[invalid-argument-type]
response = yield yielded_request
except StopAsyncIteration:
break
finally:
await gen.aclose()
except ClientNotFoundError:
logger.debug(
@ -317,9 +320,12 @@ class OAuth(OAuthClientProvider):
# Retry with fresh registration
gen = super().async_auth_flow(request)
response = None
while True:
try:
yielded_request = await gen.asend(response) # ty: ignore[invalid-argument-type]
response = yield yielded_request
except StopAsyncIteration:
break
try:
while True:
try:
yielded_request = await gen.asend(response) # ty: ignore[invalid-argument-type]
response = yield yielded_request
except StopAsyncIteration:
break
finally:
await gen.aclose()