nvidia-nat-redis 1.3.0a20250910__py3-none-any.whl → 1.3.0a20250922__py3-none-any.whl

nat/plugins/redis/redis_editor.py

@@ -171,17 +171,12 @@ class RedisEditor(MemoryEditor):
             for i, doc in enumerate(results.docs):
                 try:
                     logger.debug("Processing result %d/%d", i + 1, len(results.docs))
-                    # Get the document data from the correct attribute
-                    memory_data = {
-                        "conversation": getattr(doc, 'conversation', []),
-                        "user_id": getattr(doc, 'user_id', user_id),
-                        "tags": getattr(doc, 'tags', []),
-                        "metadata": getattr(doc, 'metadata', {}),
-                        "memory": getattr(doc, 'memory', "")
-                    }
                     logger.debug("Similarity score: %d", getattr(doc, 'score', 0))
-                    logger.debug("Extracted data for result %d: %s", i + 1, memory_data)
-                    memory_item = self._create_memory_item(memory_data, user_id)
+
+                    # Get the full document data
+                    full_doc = await self._client.json().get(doc.id)
+                    logger.debug("Extracted data for result %d: %s", i + 1, full_doc)
+                    memory_item = self._create_memory_item(dict(full_doc), user_id)
                     memories.append(memory_item)
                     logger.debug("Successfully created MemoryItem for result %d", i + 1)
                 except Exception as e:

nat/plugins/redis/schema.py

@@ -56,10 +56,11 @@ def create_schema(embedding_dim: int = DEFAULT_DIM):
     logger.info("Created embedding field with dimension %d", embedding_dim)
 
     schema = (
+        # Redis search can't directly index complex objects (e.g. conversation and metadata) in return_fields
+        # They need to be retrieved via json().get() for full object access
         TextField("$.user_id", as_name="user_id"),
         TagField("$.tags[*]", as_name="tags"),
         TextField("$.memory", as_name="memory"),
-        # TextField("$.conversations[*]", as_name="conversations"), # TODO: figure out if/how this should be done
         embedding_field)
 
     # Log the schema details

{nvidia_nat_redis-1.3.0a20250910.dist-info → nvidia_nat_redis-1.3.0a20250922.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-redis
-Version: 1.3.0a20250910
+Version: 1.3.0a20250922
 Summary: Subpackage for Redis integration in NeMo Agent toolkit
 Keywords: ai,agents,memory
 Classifier: Programming Language :: Python
@@ -9,7 +9,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: nvidia-nat==v1.3.0a20250910
+Requires-Dist: nvidia-nat==v1.3.0a20250922
 Requires-Dist: redis~=4.3.4
 
 <!--

{nvidia_nat_redis-1.3.0a20250910.dist-info → nvidia_nat_redis-1.3.0a20250922.dist-info}/RECORD

@@ -2,12 +2,12 @@ nat/meta/pypi.md,sha256=TpeNbVZJxzvEf0Gh3BGvLHPYsKnXjgM_KQVCayBPXso,1090
 nat/plugins/redis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nat/plugins/redis/memory.py,sha256=wqj_UYqnllc4kTCtFDz3eu-OZnyPWXxNZ3e6G992OGQ,2546
 nat/plugins/redis/object_store.py,sha256=f_GtCTZ3KHfxE4f0lAzKaoAoInAZZki4Dfo7hrKCAHA,1681
-nat/plugins/redis/redis_editor.py,sha256=OjdVaP1OBBR6CoIZ2PIqCrRfsD7l8U6Pn59oq_paGw4,10084
+nat/plugins/redis/redis_editor.py,sha256=nkSIWi1HUPald088fXTuF0rmZ0uS_3V65Vxy20vLSgk,9746
 nat/plugins/redis/redis_object_store.py,sha256=DX46GEQl4H1Ivf2wrRaSimNcq6EfvRsm-xhyiECudoQ,4302
 nat/plugins/redis/register.py,sha256=dJBKi-7W72ipkmZTOIo1E3ETffmJIlYhQTOlrkiFH3A,834
-nat/plugins/redis/schema.py,sha256=4Or-rrkdj6h4okGsnEm6cKMNGLCeSSFwqecQNmu6T9g,5517
-nvidia_nat_redis-1.3.0a20250910.dist-info/METADATA,sha256=w4VQ8wVrYHX0DrCv4tj766ekh6-bnsiWsJ-oTB3lROk,1588
-nvidia_nat_redis-1.3.0a20250910.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_redis-1.3.0a20250910.dist-info/entry_points.txt,sha256=nyS8t8L9CbRFIMlE70RQBtJXrflBP4Ltl5zAkIl44So,56
-nvidia_nat_redis-1.3.0a20250910.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_redis-1.3.0a20250910.dist-info/RECORD,,
+nat/plugins/redis/schema.py,sha256=Zcas3hIIqG7wuR94baYRFycmnccB3CuGmTs4p8Vv4mA,5589
+nvidia_nat_redis-1.3.0a20250922.dist-info/METADATA,sha256=T930mwYtYNRuY4sS9cNx-mHoOUHHtXCEZSHizY0OXSo,1588
+nvidia_nat_redis-1.3.0a20250922.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_redis-1.3.0a20250922.dist-info/entry_points.txt,sha256=nyS8t8L9CbRFIMlE70RQBtJXrflBP4Ltl5zAkIl44So,56
+nvidia_nat_redis-1.3.0a20250922.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_redis-1.3.0a20250922.dist-info/RECORD,,