From 2053ee3ffd4c6359530dd0cc963d683738204c16 Mon Sep 17 00:00:00 2001 From: jackwotherspoon Date: Wed, 26 Jun 2024 20:09:13 +0000 Subject: [PATCH] fix: use lazy refresh for Cloud SQL Connector --- pyproject.toml | 2 +- requirements.txt | 2 +- src/langchain_google_cloud_sql_mysql/engine.py | 6 ++++-- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 3e427bc..21356a3 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,7 +13,7 @@ dependencies = [ "langchain-community>=0.0.18, <1.0.0", "numpy>=1.24.4, <2.0.0", "SQLAlchemy>=2.0.7, <3.0.0", - "cloud-sql-python-connector[pymysql]>=1.7.0, <2.0.0" + "cloud-sql-python-connector[pymysql]>=1.10.0, <2.0.0" ] classifiers = [ diff --git a/requirements.txt b/requirements.txt index 108f72f..703b0b7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,5 +3,5 @@ langchain-community==0.2.0 numpy===1.24.4; python_version <= "3.8" numpy==1.26.4; python_version > "3.8" SQLAlchemy==2.0.30 -cloud-sql-python-connector[pymysql]==1.9.2 +cloud-sql-python-connector[pymysql]==1.10.0 diff --git a/src/langchain_google_cloud_sql_mysql/engine.py b/src/langchain_google_cloud_sql_mysql/engine.py index 3581dd6..7cdc8d6 100644 --- a/src/langchain_google_cloud_sql_mysql/engine.py +++ b/src/langchain_google_cloud_sql_mysql/engine.py @@ -21,7 +21,7 @@ import google.auth.transport.requests import requests import sqlalchemy -from google.cloud.sql.connector import Connector +from google.cloud.sql.connector import Connector, RefreshStrategy from .version import __version__ @@ -193,7 +193,9 @@ def _create_connector_engine( enable_iam_auth = True if cls._connector is None: - cls._connector = Connector(user_agent=USER_AGENT) + cls._connector = Connector( + user_agent=USER_AGENT, refresh_strategy=RefreshStrategy.LAZY + ) # anonymous function to be used for SQLAlchemy 'creator' argument def getconn() -> pymysql.Connection: