hindsight-client 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

hindsight_client/__init__.py

@@ -15,8 +15,8 @@ Example:
     print(result.success)
 
     # Search memories
-    results = client.recall(bank_id="alice", query="What does Alice like?")
-    for r in results:
+    response = client.recall(bank_id="alice", query="What does Alice like?")
+    for r in response.results:
         print(r.text)
 
     # Generate contextual answer
@@ -29,14 +29,59 @@ from .hindsight_client import Hindsight
 
 # Re-export response types for convenient access
 from hindsight_client_api.models.retain_response import RetainResponse
-from hindsight_client_api.models.recall_response import RecallResponse
-from hindsight_client_api.models.recall_result import RecallResult
+from hindsight_client_api.models.recall_response import RecallResponse as _RecallResponse
+from hindsight_client_api.models.recall_result import RecallResult as _RecallResult
 from hindsight_client_api.models.reflect_response import ReflectResponse
 from hindsight_client_api.models.reflect_fact import ReflectFact
 from hindsight_client_api.models.list_memory_units_response import ListMemoryUnitsResponse
 from hindsight_client_api.models.bank_profile_response import BankProfileResponse
 from hindsight_client_api.models.disposition_traits import DispositionTraits
 
+
+# Add cleaner __repr__ and __iter__ for REPL usability
+def _recall_result_repr(self):
+    text_preview = self.text[:80] + "..." if len(self.text) > 80 else self.text
+    return f"RecallResult(id='{self.id[:8]}...', type='{self.type}', text='{text_preview}')"
+
+
+def _recall_response_repr(self):
+    count = len(self.results) if self.results else 0
+    extras = []
+    if self.trace:
+        extras.append("trace=True")
+    if self.entities:
+        extras.append(f"entities={len(self.entities)}")
+    if self.chunks:
+        extras.append(f"chunks={len(self.chunks)}")
+    extras_str = ", " + ", ".join(extras) if extras else ""
+    return f"RecallResponse({count} results{extras_str})"
+
+
+def _recall_response_iter(self):
+    """Iterate directly over results for convenience."""
+    return iter(self.results or [])
+
+
+def _recall_response_len(self):
+    """Return number of results."""
+    return len(self.results) if self.results else 0
+
+
+def _recall_response_getitem(self, index):
+    """Access results by index."""
+    return self.results[index]
+
+
+_RecallResult.__repr__ = _recall_result_repr
+_RecallResponse.__repr__ = _recall_response_repr
+_RecallResponse.__iter__ = _recall_response_iter
+_RecallResponse.__len__ = _recall_response_len
+_RecallResponse.__getitem__ = _recall_response_getitem
+
+# Re-export with patched repr
+RecallResult = _RecallResult
+RecallResponse = _RecallResponse
+
 __all__ = [
     "Hindsight",
     # Response types

hindsight_client/hindsight_client.py

@@ -50,7 +50,9 @@ class Hindsight:
         client.retain(bank_id="alice", content="Alice loves AI")
 
         # Recall memories
-        results = client.recall(bank_id="alice", query="What does Alice like?")
+        response = client.recall(bank_id="alice", query="What does Alice like?")
+        for r in response.results:
+            print(r.text)
 
         # Generate contextual answer
         answer = client.reflect(bank_id="alice", query="What are my interests?")
@@ -125,8 +127,8 @@ class Hindsight:
 
         Args:
             bank_id: The memory bank ID
-            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata'
-            document_id: Optional document ID for grouping memories
+            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata', 'document_id'
+            document_id: Optional document ID for grouping memories (applied to items that don't have their own)
             retain_async: If True, process asynchronously in background (default: False)
 
         Returns:
@@ -138,13 +140,14 @@ class Hindsight:
                 timestamp=item.get("timestamp"),
                 context=item.get("context"),
                 metadata=item.get("metadata"),
+                # Use item's document_id if provided, otherwise fall back to batch-level document_id
+                document_id=item.get("document_id") or document_id,
             )
             for item in items
         ]
 
         request_obj = retain_request.RetainRequest(
             items=memory_items,
-            document_id=document_id,
             async_=retain_async,
         )
 
@@ -161,6 +164,8 @@ class Hindsight:
         query_timestamp: Optional[str] = None,
         include_entities: bool = False,
         max_entity_tokens: int = 500,
+        include_chunks: bool = False,
+        max_chunk_tokens: int = 8192,
     ) -> RecallResponse:
         """
         Recall memories using semantic similarity.
@@ -175,14 +180,17 @@ class Hindsight:
             query_timestamp: Optional ISO format date string (e.g., '2023-05-30T23:40:00')
             include_entities: Include entity observations in results (default: False)
             max_entity_tokens: Maximum tokens for entity observations (default: 500)
+            include_chunks: Include raw text chunks in results (default: False)
+            max_chunk_tokens: Maximum tokens for chunks (default: 8192)
 
         Returns:
-            RecallResponse with results, optional entities, and optional trace
+            RecallResponse with results, optional entities, optional chunks, and optional trace
         """
-        from hindsight_client_api.models import include_options, entity_include_options
+        from hindsight_client_api.models import include_options, entity_include_options, chunk_include_options
 
         include_opts = include_options.IncludeOptions(
-            entities=entity_include_options.EntityIncludeOptions(max_tokens=max_entity_tokens) if include_entities else None
+            entities=entity_include_options.EntityIncludeOptions(max_tokens=max_entity_tokens) if include_entities else None,
+            chunks=chunk_include_options.ChunkIncludeOptions(max_tokens=max_chunk_tokens) if include_chunks else None,
         )
 
         request_obj = recall_request.RecallRequest(
@@ -277,8 +285,8 @@ class Hindsight:
 
         Args:
             bank_id: The memory bank ID
-            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata'
-            document_id: Optional document ID for grouping memories
+            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata', 'document_id'
+            document_id: Optional document ID for grouping memories (applied to items that don't have their own)
             retain_async: If True, process asynchronously in background (default: False)
 
         Returns:
@@ -290,13 +298,14 @@ class Hindsight:
                 timestamp=item.get("timestamp"),
                 context=item.get("context"),
                 metadata=item.get("metadata"),
+                # Use item's document_id if provided, otherwise fall back to batch-level document_id
+                document_id=item.get("document_id") or document_id,
             )
             for item in items
         ]
 
         request_obj = retain_request.RetainRequest(
             items=memory_items,
-            document_id=document_id,
             async_=retain_async,
         )
 

{hindsight_client-0.1.3.dist-info → hindsight_client-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hindsight-client
-Version: 0.1.3
+Version: 0.1.5
 Summary: Python client for Hindsight - Semantic memory system with personality-driven thinking
 Author: Hindsight Team
 Requires-Python: >=3.10

{hindsight_client-0.1.3.dist-info → hindsight_client-0.1.5.dist-info}/RECORD

@@ -1,5 +1,5 @@
-hindsight_client/__init__.py,sha256=Qq8DtPOglZOQfPkAyEWulmyBU4QMeNm6C5bb6TAY0jI,1679
-hindsight_client/hindsight_client.py,sha256=kTb-tpvinPgcPEasBmQgbuQIjTNs0IrL_zQxqvZp9pc,12647
+hindsight_client/__init__.py,sha256=PyDJ4UVKmtRN5OeBs0-rl-tUtqS8OoX53qvejKGC3JU,3114
+hindsight_client/hindsight_client.py,sha256=tKqeqcbYAgCFIgYBgcyKZi-DYC10FF9suA1VhooOrxU,13481
 hindsight_client_api/__init__.py,sha256=7hfGgOu50_bmmuoz4-e4JVqubo5W39g_p3HHgaWZGRs,5941
 hindsight_client_api/api_client.py,sha256=gO_s4kVFGPJMawvNcdoqbpqptHIIKELylVni48b7rqU,27860
 hindsight_client_api/api_response.py,sha256=eMxw1mpmJcoGZ3gs9z6jM4oYoZ10Gjk333s9sKxGv7s,652
@@ -126,6 +126,6 @@ hindsight_client_api/test/test_retain_response.py,sha256=L_83q5aYZAb9mZhHwF7WAe4
 hindsight_client_api/test/test_update_disposition_request.py,sha256=bHQG7SqMD_KESLs61kVOTknYH-flvX5UjW9yJ9UK3RI,1550
 hindsight_client_api/test/test_validation_error.py,sha256=FiG2lcCSznxrbqRk2rcKv1ctpin0gGM5YGReZWMRnkw,1549
 hindsight_client_api/test/test_validation_error_loc_inner.py,sha256=Cy-VdSn1aBCRzpiI-2OSCrK376C5eHDywvbF16KNhMQ,1398
-hindsight_client-0.1.3.dist-info/METADATA,sha256=sIReA4-WRcJSeF6T2P1npijz8N4iMS_rfWBpertbpBw,598
-hindsight_client-0.1.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-hindsight_client-0.1.3.dist-info/RECORD,,
+hindsight_client-0.1.5.dist-info/METADATA,sha256=Em5Vh-0LfFwgpsGd4USYCopnv0TxmqiHcqSMdqU6aN4,598
+hindsight_client-0.1.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+hindsight_client-0.1.5.dist-info/RECORD,,