Set charset on connection from AIO pool.

tjni · Dec 8, 2024 · 2792a53 · 2792a53
1 parent d6089b1
commit 2792a53
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/langgraph/checkpoint/mysql/aio.py b/langgraph/checkpoint/mysql/aio.py
@@ -37,6 +37,9 @@ async def _get_connection(
         yield conn
     elif isinstance(conn, aiomysql.Pool):
         async with conn.acquire() as _conn:
+            # This seems necessary until https://github.com/PyMySQL/PyMySQL/pull/1119
+            # is merged into aiomysql.
+            await _conn.set_charset(pymysql.connections.DEFAULT_CHARSET)
             yield _conn
     else:
         raise TypeError(f"Invalid connection type: {type(conn)}")