nvidia-nat-mysql 1.3.dev0__py3-none-any.whl → 1.3.0.dev2__py3-none-any.whl

nat/plugins/mysql/mysql_object_store.py

@@ -14,7 +14,6 @@
 # limitations under the License.
 
 import logging
-import pickle
 
 import aiomysql
 from aiomysql.pool import Pool
@@ -57,8 +56,10 @@ class MySQLObjectStore(ObjectStore):
         )
         assert self._conn_pool is not None
 
-        logger.info(
-            f"Created connection pool for {self._config.bucket_name} at {self._config.host}:{self._config.port}")
+        logger.info("Created connection pool for %s at %s:%s",
+                    self._config.bucket_name,
+                    self._config.host,
+                    self._config.port)
 
         async with self._conn_pool.acquire() as conn:
             async with conn.cursor() as cur:
@@ -88,8 +89,10 @@ class MySQLObjectStore(ObjectStore):
 
             await conn.commit()
 
-        logger.info(
-            f"Created schema and tables for {self._config.bucket_name} at {self._config.host}:{self._config.port}")
+        logger.info("Created schema and tables for %s at %s:%s",
+                    self._config.bucket_name,
+                    self._config.host,
+                    self._config.port)
 
         return self
 
@@ -123,8 +126,7 @@ class MySQLObjectStore(ObjectStore):
                             key=key, additional_message=f"MySQL table {self._config.bucket_name} already has key {key}")
                     await cur.execute("SELECT id FROM object_meta WHERE path=%s FOR UPDATE;", (key, ))
                     (obj_id, ) = await cur.fetchone()
-
-                    blob = pickle.dumps(item)
+                    blob = item.model_dump_json()
                     await cur.execute("INSERT INTO object_data (id, data) VALUES (%s, %s)", (obj_id, blob))
                     await conn.commit()
                 except Exception:
@@ -151,7 +153,7 @@ class MySQLObjectStore(ObjectStore):
                     await cur.execute("SELECT id FROM object_meta WHERE path=%s FOR UPDATE;", (key, ))
                     (obj_id, ) = await cur.fetchone()
 
-                    blob = pickle.dumps(item)
+                    blob = item.model_dump_json()
                     await cur.execute("REPLACE INTO object_data (id, data) VALUES (%s, %s)", (obj_id, blob))
                     await conn.commit()
                 except Exception:
@@ -178,7 +180,7 @@ class MySQLObjectStore(ObjectStore):
                 if not row:
                     raise NoSuchKeyError(
                         key=key, additional_message=f"MySQL table {self._config.bucket_name} does not have key {key}")
-                return pickle.loads(row[0])
+                return ObjectStoreItem.model_validate_json(row[0].decode("utf-8"))
 
     @override
     async def delete_object(self, key: str):

{nvidia_nat_mysql-1.3.dev0.dist-info → nvidia_nat_mysql-1.3.0.dev2.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-mysql
-Version: 1.3.dev0
+Version: 1.3.0.dev2
 Summary: Subpackage for MySQL integration in NeMo Agent toolkit
 Keywords: ai,agents,memory,data store
 Classifier: Programming Language :: Python
-Requires-Python: >=3.12
+Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: nvidia-nat==v1.3-dev
+Requires-Dist: nvidia-nat==v1.3.0-dev2
 Requires-Dist: aiomysql>=0.2.0

nvidia_nat_mysql-1.3.0.dev2.dist-info/RECORD

@@ -0,0 +1,9 @@
+nat/plugins/mysql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nat/plugins/mysql/mysql_object_store.py,sha256=wgsX797j21gw9ZFt9ZCrRHRT5GDrqPHywwFmc1HxdD4,8178
+nat/plugins/mysql/object_store.py,sha256=IWMJFKBhn0lH7UGHuYkYht82TwIe5KXS7VUhQrlNRyI,2600
+nat/plugins/mysql/register.py,sha256=7gqnwyDrYttIlEaa7lo9AASYt-2GrZJE0YT2jpKjepo,845
+nvidia_nat_mysql-1.3.0.dev2.dist-info/METADATA,sha256=0nDmPlYVNyo6YJtSCIC8ADmgiENEe8k7uXdLH1-ZuMM,350
+nvidia_nat_mysql-1.3.0.dev2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_mysql-1.3.0.dev2.dist-info/entry_points.txt,sha256=ZI7ielmDX-k_OPXGvpRB5tKFVR5kMCHmHGwLWqxRKh0,56
+nvidia_nat_mysql-1.3.0.dev2.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_mysql-1.3.0.dev2.dist-info/RECORD,,

nvidia_nat_mysql-1.3.dev0.dist-info/RECORD

@@ -1,9 +0,0 @@
-nat/plugins/mysql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nat/plugins/mysql/mysql_object_store.py,sha256=EXHuA6z28__lH6qhSyGuIPWLPZ-5BzH9EHVCkpDYyvE,8042
-nat/plugins/mysql/object_store.py,sha256=IWMJFKBhn0lH7UGHuYkYht82TwIe5KXS7VUhQrlNRyI,2600
-nat/plugins/mysql/register.py,sha256=7gqnwyDrYttIlEaa7lo9AASYt-2GrZJE0YT2jpKjepo,845
-nvidia_nat_mysql-1.3.dev0.dist-info/METADATA,sha256=g11nXToNcwBEvbspx29N-29LrI_sGnp3Z0h2z2Wh7XI,339
-nvidia_nat_mysql-1.3.dev0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_mysql-1.3.dev0.dist-info/entry_points.txt,sha256=ZI7ielmDX-k_OPXGvpRB5tKFVR5kMCHmHGwLWqxRKh0,56
-nvidia_nat_mysql-1.3.dev0.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_mysql-1.3.dev0.dist-info/RECORD,,