We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 11fff7f commit 8b858c1Copy full SHA for 8b858c1
1 file changed
pipelines/google/google_gemini.py
@@ -1621,7 +1621,7 @@ async def _handle_streaming_response_with_cleanup(
1621
finally:
1622
# Ensure client is properly closed after streaming completes
1623
try:
1624
- await client.aclose()
+ await client.aio.aclose()
1625
self.log.debug("Streaming client closed successfully")
1626
except Exception as close_error:
1627
self.log.warning(f"Error closing streaming client: {close_error}")
@@ -2210,7 +2210,7 @@ async def get_response():
2210
# Ensure client is properly closed for non-streaming responses
2211
if not stream or supports_image_generation:
2212
2213
2214
self.log.debug(
2215
f"Request {request_id}: Client closed successfully"
2216
)
0 commit comments