From bd246732fba585b9fd24923b9a256bd8ba844d37 Mon Sep 17 00:00:00 2001
From: jackwotherspoon <jackwoth@google.com>
Date: Wed, 26 Jun 2024 20:11:54 +0000
Subject: [PATCH] fix: use lazy refresh for Cloud SQL Connector

---
 pyproject.toml                                 | 2 +-
 src/langchain_google_cloud_sql_mssql/engine.py | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 222d76d..88420b7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,7 +13,7 @@ dependencies = [
     "langchain-community>=0.0.18, <1.0.0",
     "SQLAlchemy>=2.0.7, <3.0.0",
     "sqlalchemy-pytds>=1.0.0, <2.0.0",
-    "cloud-sql-python-connector[pytds]>=1.7.0, <2.0.0"
+    "cloud-sql-python-connector[pytds]>=1.10.0, <2.0.0"
 ]
 classifiers = [
     "Intended Audience :: Developers",
diff --git a/src/langchain_google_cloud_sql_mssql/engine.py b/src/langchain_google_cloud_sql_mssql/engine.py
index 875d9eb..1dc7feb 100644
--- a/src/langchain_google_cloud_sql_mssql/engine.py
+++ b/src/langchain_google_cloud_sql_mssql/engine.py
@@ -18,7 +18,7 @@
 from typing import List, Optional
 
 import sqlalchemy
-from google.cloud.sql.connector import Connector
+from google.cloud.sql.connector import Connector, RefreshStrategy
 
 from .version import __version__
 
@@ -96,7 +96,9 @@ def _create_connector_engine(
                 Python Connector.
         """
         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():