hindsight-api 0.0.18__tar.gz → 0.0.21__tar.gz

{hindsight_api-0.0.18 → hindsight_api-0.0.21}/.gitignore

@@ -31,4 +31,6 @@ logs/
 
 
 hindsight-dev/benchmarks/locomo/results/
-hindsight-dev/benchmarks/longmemeval/results/
+hindsight-dev/benchmarks/longmemeval/results/
+hindsight-cli/target
+hindsight-clients/rust/target

{hindsight_api-0.0.18 → hindsight_api-0.0.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hindsight-api
-Version: 0.0.18
+Version: 0.0.21
 Summary: Temporal + Semantic + Entity Memory System for AI agents using PostgreSQL
 Requires-Python: >=3.11
 Requires-Dist: alembic>=1.17.1
@@ -8,6 +8,7 @@ Requires-Dist: asyncpg>=0.29.0
 Requires-Dist: dateparser>=1.2.2
 Requires-Dist: fastapi[standard]>=0.120.3
 Requires-Dist: fastmcp>=2.0.0
+Requires-Dist: google-genai>=1.0.0
 Requires-Dist: greenlet>=3.2.4
 Requires-Dist: httpx>=0.27.0
 Requires-Dist: langchain-text-splitters>=0.3.0

{hindsight_api-0.0.18 → hindsight_api-0.0.21}/alembic/env.py

@@ -105,6 +105,8 @@ def run_migrations_offline() -> None:
 
 def run_migrations_online() -> None:
     """Run migrations in 'online' mode with synchronous engine."""
+    from sqlalchemy import event, text
+
     get_database_url()  # Process and set the database URL in config
 
     connectable = engine_from_config(
@@ -113,7 +115,19 @@ def run_migrations_online() -> None:
         poolclass=pool.NullPool,
     )
 
+    # Add event listener to ensure connection is in read-write mode
+    # This is needed for Supabase which may start connections in read-only mode
+    @event.listens_for(connectable, "connect")
+    def set_read_write_mode(dbapi_connection, connection_record):
+        cursor = dbapi_connection.cursor()
+        cursor.execute("SET SESSION CHARACTERISTICS AS TRANSACTION READ WRITE")
+        cursor.close()
+
     with connectable.connect() as connection:
+        # Also explicitly set read-write mode on this connection
+        connection.execute(text("SET SESSION CHARACTERISTICS AS TRANSACTION READ WRITE"))
+        connection.commit()  # Commit the SET command
+
         context.configure(
             connection=connection,
             target_metadata=target_metadata
@@ -122,6 +136,9 @@ def run_migrations_online() -> None:
         with context.begin_transaction():
             context.run_migrations()
 
+        # Explicit commit to ensure changes are persisted (especially for Supabase)
+        connection.commit()
+
 
 if context.is_offline_mode():
     run_migrations_offline()

hindsight_api-0.0.21/alembic/versions/d9f6a3b4c5e2_rename_bank_to_interactions.py

@@ -0,0 +1,48 @@
+"""Rename fact_type 'bank' to 'experience'
+
+Revision ID: d9f6a3b4c5e2
+Revises: c8e5f2a3b4d1
+Create Date: 2024-12-04 15:00:00.000000
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'd9f6a3b4c5e2'
+down_revision = 'c8e5f2a3b4d1'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # Drop old check constraint FIRST (before updating data)
+    op.drop_constraint('memory_units_fact_type_check', 'memory_units', type_='check')
+
+    # Update existing 'bank' values to 'experience'
+    op.execute("UPDATE memory_units SET fact_type = 'experience' WHERE fact_type = 'bank'")
+    # Also update any 'interactions' values (in case of partial migration)
+    op.execute("UPDATE memory_units SET fact_type = 'experience' WHERE fact_type = 'interactions'")
+
+    # Create new check constraint with 'experience' instead of 'bank'
+    op.create_check_constraint(
+        'memory_units_fact_type_check',
+        'memory_units',
+        "fact_type IN ('world', 'experience', 'opinion', 'observation')"
+    )
+
+
+def downgrade():
+    # Drop new check constraint FIRST
+    op.drop_constraint('memory_units_fact_type_check', 'memory_units', type_='check')
+
+    # Update 'experience' back to 'bank'
+    op.execute("UPDATE memory_units SET fact_type = 'bank' WHERE fact_type = 'experience'")
+
+    # Recreate old check constraint
+    op.create_check_constraint(
+        'memory_units_fact_type_check',
+        'memory_units',
+        "fact_type IN ('world', 'bank', 'opinion', 'observation')"
+    )

hindsight_api-0.0.21/alembic/versions/rename_personality_to_disposition.py

@@ -0,0 +1,65 @@
+"""rename_personality_to_disposition
+
+Revision ID: rename_personality
+Revises: d9f6a3b4c5e2
+Create Date: 2024-12-04
+
+"""
+from typing import Sequence, Union
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import postgresql
+
+
+# revision identifiers, used by Alembic.
+revision: str = 'rename_personality'
+down_revision: Union[str, Sequence[str], None] = 'd9f6a3b4c5e2'
+branch_labels: Union[str, Sequence[str], None] = None
+depends_on: Union[str, Sequence[str], None] = None
+
+
+def upgrade() -> None:
+    """Rename personality column to disposition in banks table (if it exists)."""
+    conn = op.get_bind()
+
+    # Check if 'personality' column exists (old database)
+    result = conn.execute(sa.text("""
+        SELECT column_name
+        FROM information_schema.columns
+        WHERE table_name = 'banks' AND column_name = 'personality'
+    """))
+    has_personality = result.fetchone() is not None
+
+    # Check if 'disposition' column exists (new database)
+    result = conn.execute(sa.text("""
+        SELECT column_name
+        FROM information_schema.columns
+        WHERE table_name = 'banks' AND column_name = 'disposition'
+    """))
+    has_disposition = result.fetchone() is not None
+
+    if has_personality and not has_disposition:
+        # Old database: rename personality -> disposition
+        op.alter_column('banks', 'personality', new_column_name='disposition')
+    elif not has_personality and not has_disposition:
+        # Neither exists (shouldn't happen, but be safe): add disposition column
+        op.add_column('banks', sa.Column(
+            'disposition',
+            postgresql.JSONB(astext_type=sa.Text()),
+            server_default=sa.text("'{}'::jsonb"),
+            nullable=False
+        ))
+    # else: disposition already exists, nothing to do
+
+
+def downgrade() -> None:
+    """Revert disposition column back to personality."""
+    conn = op.get_bind()
+    result = conn.execute(sa.text("""
+        SELECT column_name
+        FROM information_schema.columns
+        WHERE table_name = 'banks' AND column_name = 'disposition'
+    """))
+    if result.fetchone():
+        op.alter_column('banks', 'disposition', new_column_name='personality')

{hindsight_api-0.0.18 → hindsight_api-0.0.21}/hindsight_api/api/__init__.py

@@ -87,7 +87,7 @@ from .http import (
     ReflectRequest,
     ReflectResponse,
     CreateBankRequest,
-    PersonalityTraits,
+    DispositionTraits,
 )
 
 __all__ = [
@@ -100,5 +100,5 @@ __all__ = [
     "ReflectRequest",
     "ReflectResponse",
     "CreateBankRequest",
-    "PersonalityTraits",
+    "DispositionTraits",
 ]

{hindsight_api-0.0.18 → hindsight_api-0.0.21}/hindsight_api/api/http.py

@@ -84,7 +84,7 @@ class RecallRequest(BaseModel):
     model_config = ConfigDict(json_schema_extra={
         "example": {
             "query": "What did Alice say about machine learning?",
-            "types": ["world", "bank"],
+            "types": ["world", "experience"],
             "budget": "mid",
             "max_tokens": 4096,
             "trace": True,
@@ -131,7 +131,7 @@ class RecallResult(BaseModel):
 
     id: str
     text: str
-    type: Optional[str] = None  # fact type: world, agent, opinion, observation
+    type: Optional[str] = None  # fact type: world, experience, opinion, observation
     entities: Optional[List[str]] = None  # Entity names mentioned in this fact
     context: Optional[str] = None
     occurred_start: Optional[str] = None  # ISO format date when the event started
@@ -397,7 +397,7 @@ class ReflectFact(BaseModel):
 
     id: Optional[str] = None
     text: str
-    type: Optional[str] = None  # fact type: world, agent, opinion
+    type: Optional[str] = None  # fact type: world, experience, opinion
     context: Optional[str] = None
     occurred_start: Optional[str] = None
     occurred_end: Optional[str] = None
@@ -417,7 +417,7 @@ class ReflectResponse(BaseModel):
                 {
                     "id": "456",
                     "text": "I discussed AI applications last week",
-                    "type": "bank"
+                    "type": "experience"
                 }
             ]
         }
@@ -438,8 +438,8 @@ class BanksResponse(BaseModel):
     banks: List[str]
 
 
-class PersonalityTraits(BaseModel):
-    """Personality traits based on Big Five model."""
+class DispositionTraits(BaseModel):
+    """Disposition traits based on Big Five model."""
     model_config = ConfigDict(json_schema_extra={
         "example": {
             "openness": 0.8,
@@ -456,7 +456,7 @@ class PersonalityTraits(BaseModel):
     extraversion: float = Field(ge=0.0, le=1.0, description="Extraversion (0-1)")
     agreeableness: float = Field(ge=0.0, le=1.0, description="Agreeableness (0-1)")
     neuroticism: float = Field(ge=0.0, le=1.0, description="Neuroticism (0-1)")
-    bias_strength: float = Field(ge=0.0, le=1.0, description="How strongly personality influences opinions (0-1)")
+    bias_strength: float = Field(ge=0.0, le=1.0, description="How strongly disposition influences opinions (0-1)")
 
 
 class BankProfileResponse(BaseModel):
@@ -465,7 +465,7 @@ class BankProfileResponse(BaseModel):
         "example": {
             "bank_id": "user123",
             "name": "Alice",
-            "personality": {
+            "disposition": {
                 "openness": 0.8,
                 "conscientiousness": 0.6,
                 "extraversion": 0.5,
@@ -479,13 +479,13 @@ class BankProfileResponse(BaseModel):
 
     bank_id: str
     name: str
-    personality: PersonalityTraits
+    disposition: DispositionTraits
     background: str
 
 
-class UpdatePersonalityRequest(BaseModel):
-    """Request model for updating personality traits."""
-    personality: PersonalityTraits
+class UpdateDispositionRequest(BaseModel):
+    """Request model for updating disposition traits."""
+    disposition: DispositionTraits
 
 
 class AddBackgroundRequest(BaseModel):
@@ -493,14 +493,14 @@ class AddBackgroundRequest(BaseModel):
     model_config = ConfigDict(json_schema_extra={
         "example": {
             "content": "I was born in Texas",
-            "update_personality": True
+            "update_disposition": True
         }
     })
 
     content: str = Field(description="New background information to add or merge")
-    update_personality: bool = Field(
+    update_disposition: bool = Field(
         default=True,
-        description="If true, infer Big Five personality traits from the merged background (default: true)"
+        description="If true, infer Big Five disposition traits from the merged background (default: true)"
     )
 
 
@@ -509,7 +509,7 @@ class BackgroundResponse(BaseModel):
     model_config = ConfigDict(json_schema_extra={
         "example": {
             "background": "I was born in Texas. I am a software engineer with 10 years of experience.",
-            "personality": {
+            "disposition": {
                 "openness": 0.7,
                 "conscientiousness": 0.6,
                 "extraversion": 0.5,
@@ -521,14 +521,14 @@ class BackgroundResponse(BaseModel):
     })
 
     background: str
-    personality: Optional[PersonalityTraits] = None
+    disposition: Optional[DispositionTraits] = None
 
 
 class BankListItem(BaseModel):
     """Bank list item with profile summary."""
     bank_id: str
     name: str
-    personality: PersonalityTraits
+    disposition: DispositionTraits
     background: str
     created_at: Optional[str] = None
     updated_at: Optional[str] = None
@@ -542,7 +542,7 @@ class BankListResponse(BaseModel):
                 {
                     "bank_id": "user123",
                     "name": "Alice",
-                    "personality": {
+                    "disposition": {
                         "openness": 0.5,
                         "conscientiousness": 0.5,
                         "extraversion": 0.5,
@@ -566,7 +566,7 @@ class CreateBankRequest(BaseModel):
     model_config = ConfigDict(json_schema_extra={
         "example": {
             "name": "Alice",
-            "personality": {
+            "disposition": {
                 "openness": 0.8,
                 "conscientiousness": 0.6,
                 "extraversion": 0.5,
@@ -579,7 +579,7 @@ class CreateBankRequest(BaseModel):
     })
 
     name: Optional[str] = None
-    personality: Optional[PersonalityTraits] = None
+    disposition: Optional[DispositionTraits] = None
     background: Optional[str] = None
 
 
@@ -833,7 +833,7 @@ def _register_routes(app: FastAPI):
         "/v1/default/banks/{bank_id}/graph",
         response_model=GraphDataResponse,
         summary="Get memory graph data",
-        description="Retrieve graph data for visualization, optionally filtered by type (world/agent/opinion). Limited to 1000 most recent items.",
+        description="Retrieve graph data for visualization, optionally filtered by type (world/experience/opinion). Limited to 1000 most recent items.",
         operation_id="get_graph"
     )
     async def api_graph(bank_id: str,
@@ -871,7 +871,7 @@ def _register_routes(app: FastAPI):
 
         Args:
             bank_id: Memory Bank ID (from path)
-            type: Filter by fact type (world, agent, opinion)
+            type: Filter by fact type (world, experience, opinion)
             q: Search query for full-text search (searches text and context)
             limit: Maximum number of results (default: 100)
             offset: Offset for pagination (default: 0)
@@ -901,7 +901,7 @@ def _register_routes(app: FastAPI):
 
     The type parameter is optional and must be one of:
     - 'world': General knowledge about people, places, events, and things that happen
-    - 'bank': Memories about what the AI agent did, actions taken, and tasks performed
+    - 'experience': Memories about experience, conversations, actions taken, and tasks performed
     - 'opinion': The bank's formed beliefs, perspectives, and viewpoints
 
     Set include_entities=true to get entity observations alongside recall results.
@@ -914,10 +914,10 @@ def _register_routes(app: FastAPI):
 
         try:
             # Validate types
-            valid_fact_types = ["world", "bank", "opinion"]
+            valid_fact_types = ["world", "experience", "opinion"]
 
-            # Default to world, agent, opinion if not specified (exclude observation by default)
-            fact_types = request.types if request.types else ["world", "bank", "opinion"]
+            # Default to world, experience, opinion if not specified (exclude observation by default)
+            fact_types = request.types if request.types else ["world", "experience", "opinion"]
             for ft in fact_types:
                 if ft not in valid_fact_types:
                     raise HTTPException(
@@ -1026,7 +1026,7 @@ def _register_routes(app: FastAPI):
     Reflect and formulate an answer using bank identity, world facts, and opinions.
 
     This endpoint:
-    1. Retrieves agent facts (bank's identity)
+    1. Retrieves experience (conversations and events)
     2. Retrieves world facts relevant to the query
     3. Retrieves existing opinions (bank's perspectives)
     4. Uses LLM to formulate a contextual answer
@@ -1579,19 +1579,19 @@ This operation cannot be undone.
         "/v1/default/banks/{bank_id}/profile",
         response_model=BankProfileResponse,
         summary="Get memory bank profile",
-        description="Get personality traits and background for a memory bank. Auto-creates agent with defaults if not exists.",
+        description="Get disposition traits and background for a memory bank. Auto-creates agent with defaults if not exists.",
         operation_id="get_bank_profile"
     )
     async def api_get_bank_profile(bank_id: str):
-        """Get memory bank profile (personality + background)."""
+        """Get memory bank profile (disposition + background)."""
         try:
             profile = await app.state.memory.get_bank_profile(bank_id)
-            # Convert PersonalityTraits object to dict for Pydantic
-            personality_dict = profile["personality"].model_dump() if hasattr(profile["personality"], 'model_dump') else dict(profile["personality"])
+            # Convert DispositionTraits object to dict for Pydantic
+            disposition_dict = profile["disposition"].model_dump() if hasattr(profile["disposition"], 'model_dump') else dict(profile["disposition"])
             return BankProfileResponse(
                 bank_id=bank_id,
                 name=profile["name"],
-                personality=PersonalityTraits(**personality_dict),
+                disposition=DispositionTraits(**disposition_dict),
                 background=profile["background"]
             )
         except Exception as e:
@@ -1604,28 +1604,28 @@ This operation cannot be undone.
     @app.put(
         "/v1/default/banks/{bank_id}/profile",
         response_model=BankProfileResponse,
-        summary="Update memory bank personality",
-        description="Update bank's Big Five personality traits and bias strength",
-        operation_id="update_bank_personality"
+        summary="Update memory bank disposition",
+        description="Update bank's Big Five disposition traits and bias strength",
+        operation_id="update_bank_disposition"
     )
-    async def api_update_bank_personality(bank_id: str,
-        request: UpdatePersonalityRequest
+    async def api_update_bank_disposition(bank_id: str,
+        request: UpdateDispositionRequest
     ):
-        """Update bank personality traits."""
+        """Update bank disposition traits."""
         try:
-            # Update personality
-            await app.state.memory.update_bank_personality(
+            # Update disposition
+            await app.state.memory.update_bank_disposition(
                 bank_id,
-                request.personality.model_dump()
+                request.disposition.model_dump()
             )
 
             # Get updated profile
             profile = await app.state.memory.get_bank_profile(bank_id)
-            personality_dict = profile["personality"].model_dump() if hasattr(profile["personality"], 'model_dump') else dict(profile["personality"])
+            disposition_dict = profile["disposition"].model_dump() if hasattr(profile["disposition"], 'model_dump') else dict(profile["disposition"])
             return BankProfileResponse(
                 bank_id=bank_id,
                 name=profile["name"],
-                personality=PersonalityTraits(**personality_dict),
+                disposition=DispositionTraits(**disposition_dict),
                 background=profile["background"]
             )
         except Exception as e:
@@ -1639,23 +1639,23 @@ This operation cannot be undone.
         "/v1/default/banks/{bank_id}/background",
         response_model=BackgroundResponse,
         summary="Add/merge memory bank background",
-        description="Add new background information or merge with existing. LLM intelligently resolves conflicts, normalizes to first person, and optionally infers personality traits.",
+        description="Add new background information or merge with existing. LLM intelligently resolves conflicts, normalizes to first person, and optionally infers disposition traits.",
         operation_id="add_bank_background"
     )
     async def api_add_bank_background(bank_id: str,
         request: AddBackgroundRequest
     ):
-        """Add or merge bank background information. Optionally infer personality traits."""
+        """Add or merge bank background information. Optionally infer disposition traits."""
         try:
             result = await app.state.memory.merge_bank_background(
                 bank_id,
                 request.content,
-                update_personality=request.update_personality
+                update_disposition=request.update_disposition
             )
 
             response = BackgroundResponse(background=result["background"])
-            if "personality" in result:
-                response.personality = PersonalityTraits(**result["personality"])
+            if "disposition" in result:
+                response.disposition = DispositionTraits(**result["disposition"])
 
             return response
         except Exception as e:
@@ -1669,13 +1669,13 @@ This operation cannot be undone.
         "/v1/default/banks/{bank_id}",
         response_model=BankProfileResponse,
         summary="Create or update memory bank",
-        description="Create a new agent or update existing agent with personality and background. Auto-fills missing fields with defaults.",
+        description="Create a new agent or update existing agent with disposition and background. Auto-fills missing fields with defaults.",
         operation_id="create_or_update_bank"
     )
     async def api_create_or_update_bank(bank_id: str,
         request: CreateBankRequest
     ):
-        """Create or update an agent with personality and background."""
+        """Create or update an agent with disposition and background."""
         try:
             # Get existing profile or create with defaults
             profile = await app.state.memory.get_bank_profile(bank_id)
@@ -1696,13 +1696,13 @@ This operation cannot be undone.
                     )
                 profile["name"] = request.name
 
-            # Update personality if provided
-            if request.personality is not None:
-                await app.state.memory.update_bank_personality(
+            # Update disposition if provided
+            if request.disposition is not None:
+                await app.state.memory.update_bank_disposition(
                     bank_id,
-                    request.personality.model_dump()
+                    request.disposition.model_dump()
                 )
-                profile["personality"] = request.personality.model_dump()
+                profile["disposition"] = request.disposition.model_dump()
 
             # Update background if provided (replace, not merge)
             if request.background is not None:
@@ -1722,11 +1722,11 @@ This operation cannot be undone.
 
             # Get final profile
             final_profile = await app.state.memory.get_bank_profile(bank_id)
-            personality_dict = final_profile["personality"].model_dump() if hasattr(final_profile["personality"], 'model_dump') else dict(final_profile["personality"])
+            disposition_dict = final_profile["disposition"].model_dump() if hasattr(final_profile["disposition"], 'model_dump') else dict(final_profile["disposition"])
             return BankProfileResponse(
                 bank_id=bank_id,
                 name=final_profile["name"],
-                personality=PersonalityTraits(**personality_dict),
+                disposition=DispositionTraits(**disposition_dict),
                 background=final_profile["background"]
             )
         except Exception as e:
@@ -1852,11 +1852,11 @@ This operation cannot be undone.
         "/v1/default/banks/{bank_id}/memories",
         response_model=DeleteResponse,
         summary="Clear memory bank memories",
-        description="Delete memory units for a memory bank. Optionally filter by type (world, agent, opinion) to delete only specific types. This is a destructive operation that cannot be undone. The bank profile (personality and background) will be preserved.",
+        description="Delete memory units for a memory bank. Optionally filter by type (world, experience, opinion) to delete only specific types. This is a destructive operation that cannot be undone. The bank profile (personality and background) will be preserved.",
         operation_id="clear_bank_memories"
     )
     async def api_clear_bank_memories(bank_id: str,
-        type: Optional[str] = Query(None, description="Optional fact type filter (world, agent, opinion)")
+        type: Optional[str] = Query(None, description="Optional fact type filter (world, experience, opinion)")
     ):
         """Clear memories for a memory bank, optionally filtered by type."""
         try:

{hindsight_api-0.0.18 → hindsight_api-0.0.21}/hindsight_api/api/mcp.py

@@ -90,7 +90,7 @@ def create_mcp_server(memory: MemoryEngine) -> FastMCP:
             search_result = await memory.recall_async(
                 bank_id=bank_id,
                 query=query,
-                fact_type=["world", "bank", "opinion"],
+                fact_type=["world", "experience", "opinion"],
                 budget=Budget.LOW
             )