hindsight-client 0.1.3__tar.gz → 0.1.4__tar.gz

{hindsight_client-0.1.3 → hindsight_client-0.1.4}/PKG-INFO

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

{hindsight_client-0.1.3 → hindsight_client-0.1.4}/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-0.1.3 → hindsight_client-0.1.4}/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 → hindsight_client-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hindsight-client"
-version = "0.1.3"
+version = "0.1.4"
 description = "Python client for Hindsight - Semantic memory system with personality-driven thinking"
 authors = [
     {name = "Hindsight Team"}