feat: add user_agent to client (#35)

* feat: add user_agent to client * Update engine.py

Author: kurtisvg

pyproject.toml

@@ -6,7 +6,7 @@ readme = "README.md"
 license = {file = "LICENSE"}
 requires-python = ">=3.8"
 dependencies = [
- "cloud-sql-python-connector[asyncpg] >= 1.6.0, <2.0.0",
+ "cloud-sql-python-connector[asyncpg] >= 1.7.0, <2.0.0",
  "langchain-core>=0.1.1, <2.0.0 ",
  "langchain-community>=0.0.18, <0.1.0",
  "pgvector>=0.2.5, <1.0.0",

src/langchain_google_cloud_sql_pg/engine.py

@@ -26,12 +26,16 @@
 from sqlalchemy import text # Column,
 from sqlalchemy.ext.asyncio import AsyncEngine, create_async_engine
 
+from .version import __version__
+
 if TYPE_CHECKING:
  import asyncpg # type: ignore
  import google.auth.credentials # type: ignore
 
 T = TypeVar("T")
 
+USER_AGENT = "langchain-google-cloud-sql-pg-python/" + __version__
+
 
 async def _get_iam_principal_email(
  credentials: google.auth.credentials.Credentials,
@@ -143,7 +147,9 @@ async def _create(
  "authentication or neither for IAM DB authentication."
  )
  if cls._connector is None:
- cls._connector = await create_async_connector()
+ cls._connector = Connector(
+ loop=asyncio.get_event_loop(), user_agent=USER_AGENT
+ )
  # if user and password are given, use basic auth
  if user and password:
  enable_iam_auth = False

src/langchain_google_cloud_sql_pg/version.py

@@ -13,4 +13,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__version__ = "0.1.0"
+__version__ = "0.0.1"