hindsight-client 0.2.1__py3-none-any.whl → 0.4.0__py3-none-any.whl

hindsight_client/hindsight_client.py

@@ -6,23 +6,23 @@ easy-to-use interface on top of the auto-generated OpenAPI client.
 """
 
 import asyncio
-from typing import Optional, List, Dict, Any
 from datetime import datetime
+from typing import Any, Literal
 
 import hindsight_client_api
-from hindsight_client_api.api import memory_api, banks_api
+from hindsight_client_api.api import banks_api, directives_api, memory_api, mental_models_api
 from hindsight_client_api.models import (
-    recall_request,
-    retain_request,
     memory_item,
+    recall_request,
     reflect_request,
+    retain_request,
 )
-from hindsight_client_api.models.retain_response import RetainResponse
+from hindsight_client_api.models.bank_profile_response import BankProfileResponse
+from hindsight_client_api.models.list_memory_units_response import ListMemoryUnitsResponse
 from hindsight_client_api.models.recall_response import RecallResponse
 from hindsight_client_api.models.recall_result import RecallResult
 from hindsight_client_api.models.reflect_response import ReflectResponse
-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.retain_response import RetainResponse
 
 
 def _run_async(coro):
@@ -63,7 +63,7 @@ class Hindsight:
         ```
     """
 
-    def __init__(self, base_url: str, api_key: Optional[str] = None, timeout: float = 30.0):
+    def __init__(self, base_url: str, api_key: str | None = None, timeout: float = 30.0):
         """
         Initialize the Hindsight client.
 
@@ -74,8 +74,12 @@ class Hindsight:
         """
         config = hindsight_client_api.Configuration(host=base_url, access_token=api_key)
         self._api_client = hindsight_client_api.ApiClient(config)
+        if api_key:
+            self._api_client.set_default_header("Authorization", f"Bearer {api_key}")
         self._memory_api = memory_api.MemoryApi(self._api_client)
         self._banks_api = banks_api.BanksApi(self._api_client)
+        self._mental_models_api = mental_models_api.MentalModelsApi(self._api_client)
+        self._directives_api = directives_api.DirectivesApi(self._api_client)
 
     def __enter__(self):
         """Context manager entry."""
@@ -108,11 +112,12 @@ class Hindsight:
         self,
         bank_id: str,
         content: str,
-        timestamp: Optional[datetime] = None,
-        context: Optional[str] = None,
-        document_id: Optional[str] = None,
-        metadata: Optional[Dict[str, str]] = None,
-        entities: Optional[List[Dict[str, str]]] = None,
+        timestamp: datetime | None = None,
+        context: str | None = None,
+        document_id: str | None = None,
+        metadata: dict[str, str] | None = None,
+        entities: list[dict[str, str]] | None = None,
+        tags: list[str] | None = None,
     ) -> RetainResponse:
         """
         Store a single memory (simplified interface).
@@ -125,21 +130,32 @@ class Hindsight:
             document_id: Optional document ID for grouping
             metadata: Optional user-defined metadata
             entities: Optional list of entities [{"text": "...", "type": "..."}]
+            tags: Optional list of tags for filtering memories during recall/reflect
 
         Returns:
             RetainResponse with success status
         """
         return self.retain_batch(
             bank_id=bank_id,
-            items=[{"content": content, "timestamp": timestamp, "context": context, "metadata": metadata, "entities": entities}],
+            items=[
+                {
+                    "content": content,
+                    "timestamp": timestamp,
+                    "context": context,
+                    "metadata": metadata,
+                    "entities": entities,
+                    "tags": tags,
+                }
+            ],
             document_id=document_id,
         )
 
     def retain_batch(
         self,
         bank_id: str,
-        items: List[Dict[str, Any]],
-        document_id: Optional[str] = None,
+        items: list[dict[str, Any]],
+        document_id: str | None = None,
+        document_tags: list[str] | None = None,
         retain_async: bool = False,
     ) -> RetainResponse:
         """
@@ -147,8 +163,9 @@ class Hindsight:
 
         Args:
             bank_id: The memory bank ID
-            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata', 'document_id', 'entities'
+            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata', 'document_id', 'entities', 'tags'
             document_id: Optional document ID for grouping memories (applied to items that don't have their own)
+            document_tags: Optional list of tags applied to all items in this batch (merged with per-item tags)
             retain_async: If True, process asynchronously in background (default: False)
 
         Returns:
@@ -160,10 +177,7 @@ class Hindsight:
         for item in items:
             entities = None
             if item.get("entities"):
-                entities = [
-                    EntityInput(text=e["text"], type=e.get("type"))
-                    for e in item["entities"]
-                ]
+                entities = [EntityInput(text=e["text"], type=e.get("type")) for e in item["entities"]]
             memory_items.append(
                 memory_item.MemoryItem(
                     content=item["content"],
@@ -173,12 +187,14 @@ class Hindsight:
                     # Use item's document_id if provided, otherwise fall back to batch-level document_id
                     document_id=item.get("document_id") or document_id,
                     entities=entities,
+                    tags=item.get("tags"),
                 )
             )
 
         request_obj = retain_request.RetainRequest(
             items=memory_items,
             async_=retain_async,
+            document_tags=document_tags,
         )
 
         return _run_async(self._memory_api.retain_memories(bank_id, request_obj))
@@ -187,15 +203,17 @@ class Hindsight:
         self,
         bank_id: str,
         query: str,
-        types: Optional[List[str]] = None,
+        types: list[str] | None = None,
         max_tokens: int = 4096,
         budget: str = "mid",
         trace: bool = False,
-        query_timestamp: Optional[str] = None,
+        query_timestamp: str | None = None,
         include_entities: bool = False,
         max_entity_tokens: int = 500,
         include_chunks: bool = False,
         max_chunk_tokens: int = 8192,
+        tags: list[str] | None = None,
+        tags_match: Literal["any", "all", "any_strict", "all_strict"] = "any",
     ) -> RecallResponse:
         """
         Recall memories using semantic similarity.
@@ -212,14 +230,19 @@ class Hindsight:
             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)
+            tags: Optional list of tags to filter memories by
+            tags_match: How to match tags - "any" (OR, includes untagged), "all" (AND, includes untagged),
+                "any_strict" (OR, excludes untagged), "all_strict" (AND, excludes untagged). Default: "any"
 
         Returns:
             RecallResponse with results, optional entities, optional chunks, and optional trace
         """
-        from hindsight_client_api.models import include_options, entity_include_options, chunk_include_options
+        from hindsight_client_api.models import chunk_include_options, entity_include_options, 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,
         )
 
@@ -231,6 +254,8 @@ class Hindsight:
             trace=trace,
             query_timestamp=query_timestamp,
             include=include_opts,
+            tags=tags,
+            tags_match=tags_match,
         )
 
         return _run_async(self._memory_api.recall_memories(bank_id, request_obj))
@@ -240,9 +265,11 @@ class Hindsight:
         bank_id: str,
         query: str,
         budget: str = "low",
-        context: Optional[str] = None,
-        max_tokens: Optional[int] = None,
-        response_schema: Optional[Dict[str, Any]] = None,
+        context: str | None = None,
+        max_tokens: int | None = None,
+        response_schema: dict[str, Any] | None = None,
+        tags: list[str] | None = None,
+        tags_match: Literal["any", "all", "any_strict", "all_strict"] = "any",
     ) -> ReflectResponse:
         """
         Generate a contextual answer based on bank identity and memories.
@@ -256,6 +283,9 @@ class Hindsight:
             response_schema: Optional JSON Schema for structured output. When provided,
                 the response will include a 'structured_output' field with the LLM
                 response parsed according to this schema.
+            tags: Optional list of tags to filter memories by
+            tags_match: How to match tags - "any" (OR, includes untagged), "all" (AND, includes untagged),
+                "any_strict" (OR, excludes untagged), "all_strict" (AND, excludes untagged). Default: "any"
 
         Returns:
             ReflectResponse with answer text, optionally facts used, and optionally
@@ -267,6 +297,8 @@ class Hindsight:
             context=context,
             max_tokens=max_tokens,
             response_schema=response_schema,
+            tags=tags,
+            tags_match=tags_match,
         )
 
         return _run_async(self._memory_api.reflect(bank_id, request_obj))
@@ -274,28 +306,37 @@ class Hindsight:
     def list_memories(
         self,
         bank_id: str,
-        type: Optional[str] = None,
-        search_query: Optional[str] = None,
+        type: str | None = None,
+        search_query: str | None = None,
         limit: int = 100,
         offset: int = 0,
     ) -> ListMemoryUnitsResponse:
         """List memory units with pagination."""
-        return _run_async(self._memory_api.list_memories(
-            bank_id=bank_id,
-            type=type,
-            q=search_query,
-            limit=limit,
-            offset=offset,
-        ))
+        return _run_async(
+            self._memory_api.list_memories(
+                bank_id=bank_id,
+                type=type,
+                q=search_query,
+                limit=limit,
+                offset=offset,
+            )
+        )
 
     def create_bank(
         self,
         bank_id: str,
-        name: Optional[str] = None,
-        background: Optional[str] = None,
-        disposition: Optional[Dict[str, float]] = None,
+        name: str | None = None,
+        mission: str | None = None,
+        disposition: dict[str, float] | None = None,
     ) -> BankProfileResponse:
-        """Create or update a memory bank."""
+        """Create or update a memory bank.
+
+        Args:
+            bank_id: Unique identifier for the bank
+            name: Human-readable display name
+            mission: Instructions guiding what Hindsight should learn and remember (for mental models)
+            disposition: Optional disposition traits (skepticism, literalism, empathy)
+        """
         from hindsight_client_api.models import create_bank_request, disposition_traits
 
         disposition_obj = None
@@ -304,19 +345,40 @@ class Hindsight:
 
         request_obj = create_bank_request.CreateBankRequest(
             name=name,
-            background=background,
+            mission=mission,
             disposition=disposition_obj,
         )
 
         return _run_async(self._banks_api.create_or_update_bank(bank_id, request_obj))
 
+    def set_mission(
+        self,
+        bank_id: str,
+        mission: str,
+    ) -> BankProfileResponse:
+        """
+        Set or update the mission for a memory bank.
+
+        Args:
+            bank_id: The memory bank ID
+            mission: The mission text describing the agent's purpose
+
+        Returns:
+            BankProfileResponse with updated bank profile
+        """
+        from hindsight_client_api.models import create_bank_request
+
+        request_obj = create_bank_request.CreateBankRequest(mission=mission)
+        return _run_async(self._banks_api.create_or_update_bank(bank_id, request_obj))
+
     # Async methods (native async, no _run_async wrapper)
 
     async def aretain_batch(
         self,
         bank_id: str,
-        items: List[Dict[str, Any]],
-        document_id: Optional[str] = None,
+        items: list[dict[str, Any]],
+        document_id: str | None = None,
+        document_tags: list[str] | None = None,
         retain_async: bool = False,
     ) -> RetainResponse:
         """
@@ -324,8 +386,9 @@ class Hindsight:
 
         Args:
             bank_id: The memory bank ID
-            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata', 'document_id', 'entities'
+            items: List of memory items with 'content' and optional 'timestamp', 'context', 'metadata', 'document_id', 'entities', 'tags'
             document_id: Optional document ID for grouping memories (applied to items that don't have their own)
+            document_tags: Optional list of tags applied to all items in this batch (merged with per-item tags)
             retain_async: If True, process asynchronously in background (default: False)
 
         Returns:
@@ -337,10 +400,7 @@ class Hindsight:
         for item in items:
             entities = None
             if item.get("entities"):
-                entities = [
-                    EntityInput(text=e["text"], type=e.get("type"))
-                    for e in item["entities"]
-                ]
+                entities = [EntityInput(text=e["text"], type=e.get("type")) for e in item["entities"]]
             memory_items.append(
                 memory_item.MemoryItem(
                     content=item["content"],
@@ -350,12 +410,14 @@ class Hindsight:
                     # Use item's document_id if provided, otherwise fall back to batch-level document_id
                     document_id=item.get("document_id") or document_id,
                     entities=entities,
+                    tags=item.get("tags"),
                 )
             )
 
         request_obj = retain_request.RetainRequest(
             items=memory_items,
             async_=retain_async,
+            document_tags=document_tags,
         )
 
         return await self._memory_api.retain_memories(bank_id, request_obj)
@@ -364,11 +426,12 @@ class Hindsight:
         self,
         bank_id: str,
         content: str,
-        timestamp: Optional[datetime] = None,
-        context: Optional[str] = None,
-        document_id: Optional[str] = None,
-        metadata: Optional[Dict[str, str]] = None,
-        entities: Optional[List[Dict[str, str]]] = None,
+        timestamp: datetime | None = None,
+        context: str | None = None,
+        document_id: str | None = None,
+        metadata: dict[str, str] | None = None,
+        entities: list[dict[str, str]] | None = None,
+        tags: list[str] | None = None,
     ) -> RetainResponse:
         """
         Store a single memory (async).
@@ -381,13 +444,23 @@ class Hindsight:
             document_id: Optional document ID for grouping
             metadata: Optional user-defined metadata
             entities: Optional list of entities [{"text": "...", "type": "..."}]
+            tags: Optional list of tags for filtering memories during recall/reflect
 
         Returns:
             RetainResponse with success status
         """
         return await self.aretain_batch(
             bank_id=bank_id,
-            items=[{"content": content, "timestamp": timestamp, "context": context, "metadata": metadata, "entities": entities}],
+            items=[
+                {
+                    "content": content,
+                    "timestamp": timestamp,
+                    "context": context,
+                    "metadata": metadata,
+                    "entities": entities,
+                    "tags": tags,
+                }
+            ],
             document_id=document_id,
         )
 
@@ -395,10 +468,12 @@ class Hindsight:
         self,
         bank_id: str,
         query: str,
-        types: Optional[List[str]] = None,
+        types: list[str] | None = None,
         max_tokens: int = 4096,
         budget: str = "mid",
-    ) -> List[RecallResult]:
+        tags: list[str] | None = None,
+        tags_match: Literal["any", "all", "any_strict", "all_strict"] = "any",
+    ) -> list[RecallResult]:
         """
         Recall memories using semantic similarity (async).
 
@@ -408,6 +483,9 @@ class Hindsight:
             types: Optional list of fact types to filter (world, experience, opinion, observation)
             max_tokens: Maximum tokens in results (default: 4096)
             budget: Budget level for recall - "low", "mid", or "high" (default: "mid")
+            tags: Optional list of tags to filter memories by
+            tags_match: How to match tags - "any" (OR, includes untagged), "all" (AND, includes untagged),
+                "any_strict" (OR, excludes untagged), "all_strict" (AND, excludes untagged). Default: "any"
 
         Returns:
             List of RecallResult objects
@@ -418,17 +496,21 @@ class Hindsight:
             budget=budget,
             max_tokens=max_tokens,
             trace=False,
+            tags=tags,
+            tags_match=tags_match,
         )
 
         response = await self._memory_api.recall_memories(bank_id, request_obj)
-        return response.results if hasattr(response, 'results') else []
+        return response.results if hasattr(response, "results") else []
 
     async def areflect(
         self,
         bank_id: str,
         query: str,
         budget: str = "low",
-        context: Optional[str] = None,
+        context: str | None = None,
+        tags: list[str] | None = None,
+        tags_match: Literal["any", "all", "any_strict", "all_strict"] = "any",
     ) -> ReflectResponse:
         """
         Generate a contextual answer based on bank identity and memories (async).
@@ -438,6 +520,9 @@ class Hindsight:
             query: The question or prompt
             budget: Budget level for reflection - "low", "mid", or "high" (default: "low")
             context: Optional additional context
+            tags: Optional list of tags to filter memories by
+            tags_match: How to match tags - "any" (OR, includes untagged), "all" (AND, includes untagged),
+                "any_strict" (OR, excludes untagged), "all_strict" (AND, excludes untagged). Default: "any"
 
         Returns:
             ReflectResponse with answer text and optionally facts used
@@ -446,6 +531,258 @@ class Hindsight:
             query=query,
             budget=budget,
             context=context,
+            tags=tags,
+            tags_match=tags_match,
         )
 
         return await self._memory_api.reflect(bank_id, request_obj)
+
+    # Mental Models methods
+
+    def create_mental_model(
+        self,
+        bank_id: str,
+        name: str,
+        source_query: str,
+        tags: list[str] | None = None,
+        max_tokens: int | None = None,
+        trigger: dict[str, Any] | None = None,
+    ):
+        """
+        Create a mental model (runs reflect in background).
+
+        Args:
+            bank_id: The memory bank ID
+            name: Human-readable name for the mental model
+            source_query: The query to run to generate content
+            tags: Optional tags for filtering during retrieval
+            max_tokens: Optional maximum tokens for the mental model content
+            trigger: Optional trigger settings (e.g., {"refresh_after_consolidation": True})
+
+        Returns:
+            CreateMentalModelResponse with operation_id
+        """
+        from hindsight_client_api.models import create_mental_model_request, mental_model_trigger
+
+        trigger_obj = None
+        if trigger:
+            trigger_obj = mental_model_trigger.MentalModelTrigger(**trigger)
+
+        request_obj = create_mental_model_request.CreateMentalModelRequest(
+            name=name,
+            source_query=source_query,
+            tags=tags,
+            max_tokens=max_tokens,
+            trigger=trigger_obj,
+        )
+
+        return _run_async(self._mental_models_api.create_mental_model(bank_id, request_obj))
+
+    def list_mental_models(self, bank_id: str, tags: list[str] | None = None):
+        """
+        List all mental models in a bank.
+
+        Args:
+            bank_id: The memory bank ID
+            tags: Optional tags to filter by
+
+        Returns:
+            ListMentalModelsResponse with items
+        """
+        return _run_async(self._mental_models_api.list_mental_models(bank_id, tags=tags))
+
+    def get_mental_model(self, bank_id: str, mental_model_id: str):
+        """
+        Get a specific mental model.
+
+        Args:
+            bank_id: The memory bank ID
+            mental_model_id: The mental model ID
+
+        Returns:
+            MentalModelResponse
+        """
+        return _run_async(self._mental_models_api.get_mental_model(bank_id, mental_model_id))
+
+    def refresh_mental_model(self, bank_id: str, mental_model_id: str):
+        """
+        Refresh a mental model to update with current knowledge.
+
+        Args:
+            bank_id: The memory bank ID
+            mental_model_id: The mental model ID
+
+        Returns:
+            RefreshMentalModelResponse with operation_id
+        """
+        return _run_async(self._mental_models_api.refresh_mental_model(bank_id, mental_model_id))
+
+    def update_mental_model(
+        self,
+        bank_id: str,
+        mental_model_id: str,
+        name: str | None = None,
+        source_query: str | None = None,
+        tags: list[str] | None = None,
+        max_tokens: int | None = None,
+        trigger: dict[str, Any] | None = None,
+    ):
+        """
+        Update a mental model's metadata.
+
+        Args:
+            bank_id: The memory bank ID
+            mental_model_id: The mental model ID
+            name: Optional new name
+            source_query: Optional new source query
+            tags: Optional new tags
+            max_tokens: Optional new max tokens
+            trigger: Optional trigger settings (e.g., {"refresh_after_consolidation": True})
+
+        Returns:
+            MentalModelResponse
+        """
+        from hindsight_client_api.models import mental_model_trigger, update_mental_model_request
+
+        trigger_obj = None
+        if trigger:
+            trigger_obj = mental_model_trigger.MentalModelTrigger(**trigger)
+
+        request_obj = update_mental_model_request.UpdateMentalModelRequest(
+            name=name,
+            source_query=source_query,
+            tags=tags,
+            max_tokens=max_tokens,
+            trigger=trigger_obj,
+        )
+
+        return _run_async(self._mental_models_api.update_mental_model(bank_id, mental_model_id, request_obj))
+
+    def delete_mental_model(self, bank_id: str, mental_model_id: str):
+        """
+        Delete a mental model.
+
+        Args:
+            bank_id: The memory bank ID
+            mental_model_id: The mental model ID
+        """
+        return _run_async(self._mental_models_api.delete_mental_model(bank_id, mental_model_id))
+
+    # Directives methods
+
+    def create_directive(
+        self,
+        bank_id: str,
+        name: str,
+        content: str,
+        priority: int = 0,
+        is_active: bool = True,
+        tags: list[str] | None = None,
+    ):
+        """
+        Create a directive (hard rule for reflect).
+
+        Args:
+            bank_id: The memory bank ID
+            name: Human-readable name for the directive
+            content: The directive content/rules
+            priority: Priority level (higher = injected first)
+            is_active: Whether the directive is active
+            tags: Optional tags for filtering
+
+        Returns:
+            DirectiveResponse
+        """
+        from hindsight_client_api.models import create_directive_request
+
+        request_obj = create_directive_request.CreateDirectiveRequest(
+            name=name,
+            content=content,
+            priority=priority,
+            is_active=is_active,
+            tags=tags,
+        )
+
+        return _run_async(self._directives_api.create_directive(bank_id, request_obj))
+
+    def list_directives(self, bank_id: str, tags: list[str] | None = None):
+        """
+        List all directives in a bank.
+
+        Args:
+            bank_id: The memory bank ID
+            tags: Optional tags to filter by
+
+        Returns:
+            ListDirectivesResponse with items
+        """
+        return _run_async(self._directives_api.list_directives(bank_id, tags=tags))
+
+    def get_directive(self, bank_id: str, directive_id: str):
+        """
+        Get a specific directive.
+
+        Args:
+            bank_id: The memory bank ID
+            directive_id: The directive ID
+
+        Returns:
+            DirectiveResponse
+        """
+        return _run_async(self._directives_api.get_directive(bank_id, directive_id))
+
+    def update_directive(
+        self,
+        bank_id: str,
+        directive_id: str,
+        name: str | None = None,
+        content: str | None = None,
+        priority: int | None = None,
+        is_active: bool | None = None,
+        tags: list[str] | None = None,
+    ):
+        """
+        Update a directive.
+
+        Args:
+            bank_id: The memory bank ID
+            directive_id: The directive ID
+            name: Optional new name
+            content: Optional new content
+            priority: Optional new priority
+            is_active: Optional new active status
+            tags: Optional new tags
+
+        Returns:
+            DirectiveResponse
+        """
+        from hindsight_client_api.models import update_directive_request
+
+        request_obj = update_directive_request.UpdateDirectiveRequest(
+            name=name,
+            content=content,
+            priority=priority,
+            is_active=is_active,
+            tags=tags,
+        )
+
+        return _run_async(self._directives_api.update_directive(bank_id, directive_id, request_obj))
+
+    def delete_directive(self, bank_id: str, directive_id: str):
+        """
+        Delete a directive.
+
+        Args:
+            bank_id: The memory bank ID
+            directive_id: The directive ID
+        """
+        return _run_async(self._directives_api.delete_directive(bank_id, directive_id))
+
+    def delete_bank(self, bank_id: str):
+        """
+        Delete a memory bank.
+
+        Args:
+            bank_id: The memory bank ID
+        """
+        return _run_async(self._banks_api.delete_bank(bank_id))