sonzai 1.2.0__tar.gz → 1.2.2__tar.gz

{sonzai-1.2.0 → sonzai-1.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sonzai
-Version: 1.2.0
+Version: 1.2.2
 Summary: Python SDK for the Sonzai Mind Layer API
 Project-URL: Homepage, https://github.com/sonz-ai/sonzai-python
 Project-URL: Documentation, https://github.com/sonz-ai/sonzai-python#readme

{sonzai-1.2.0 → sonzai-1.2.2}/justfile

@@ -85,6 +85,7 @@ _bump VERSION:
     perl -pi -e 's/^version = "[^"]+"/version = "{{VERSION}}"/' pyproject.toml
     perl -pi -e 's/^__version__ = "[^"]+"/__version__ = "{{VERSION}}"/' src/sonzai/__init__.py
     perl -pi -e 's{"User-Agent": "sonzai-python/[0-9]+\.[0-9]+\.[0-9]+"}{"User-Agent": "sonzai-python/{{VERSION}}"}g' src/sonzai/_http.py
+    uv lock --quiet
     echo "bumped to {{VERSION}}"
 
 _build:
@@ -92,7 +93,7 @@ _build:
     uv build
 
 _commit VERSION:
-    git add pyproject.toml src/sonzai/__init__.py src/sonzai/_http.py
+    git add pyproject.toml uv.lock src/sonzai/__init__.py src/sonzai/_http.py
     git commit -m "release: v{{VERSION}}"
 
 _publish VERSION:

{sonzai-1.2.0 → sonzai-1.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "sonzai"
-version = "1.2.0"
+version = "1.2.2"
 description = "Python SDK for the Sonzai Mind Layer API"
 readme = "README.md"
 license = "MIT"

{sonzai-1.2.0 → sonzai-1.2.2}/src/sonzai/__init__.py

@@ -160,7 +160,7 @@ from .types import (
     WebhookRegisterResponse,
 )
 
-__version__ = "1.2.0"
+__version__ = "1.2.1"
 
 __all__ = [
     # Clients

{sonzai-1.2.0 → sonzai-1.2.2}/src/sonzai/_http.py

@@ -82,7 +82,7 @@ class HTTPClient:
                 headers={
                     "Authorization": f"Bearer {api_key}",
                     "Content-Type": "application/json",
-                    "User-Agent": "sonzai-python/1.2.0",
+                    "User-Agent": "sonzai-python/1.2.1",
                 },
                 timeout=httpx.Timeout(timeout, connect=10.0),
                 follow_redirects=True,
@@ -246,7 +246,7 @@ class AsyncHTTPClient:
                 headers={
                     "Authorization": f"Bearer {api_key}",
                     "Content-Type": "application/json",
-                    "User-Agent": "sonzai-python/1.2.0",
+                    "User-Agent": "sonzai-python/1.2.1",
                 },
                 timeout=httpx.Timeout(timeout, connect=10.0),
                 follow_redirects=True,

{sonzai-1.2.0 → sonzai-1.2.2}/src/sonzai/resources/agents.py

@@ -111,6 +111,8 @@ class Agents:
         generate_origin_story: bool | None = None,
         generate_personalized_memories: bool | None = None,
         initial_goals: list[dict[str, Any]] | None = None,
+        generate_avatar: bool | None = None,
+        capabilities: dict[str, Any] | None = None,
     ) -> Agent:
         """Create a new agent."""
         body: dict[str, Any] = {"name": name}
@@ -160,6 +162,10 @@ class Agents:
             body["generate_personalized_memories"] = generate_personalized_memories
         if initial_goals is not None:
             body["initial_goals"] = initial_goals
+        if generate_avatar is not None:
+            body["generate_avatar"] = generate_avatar
+        if capabilities is not None:
+            body["capabilities"] = capabilities
 
         data = self._http.post("/api/v1/agents", json_data=body)
         return Agent.model_validate(data)
@@ -660,9 +666,10 @@ class Agents:
         data = self._http.get(f"/api/v1/agents/{agent_id}/relationships", params=params)
         return RelationshipResponse.model_validate(data)
 
-    def get_habits(
+    def list_habits(
         self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
     ) -> HabitsResponse:
+        """List habits for an agent."""
         params: dict[str, Any] = {}
         if user_id:
             params["user_id"] = user_id
@@ -671,6 +678,12 @@ class Agents:
         data = self._http.get(f"/api/v1/agents/{agent_id}/habits", params=params)
         return HabitsResponse.model_validate(data)
 
+    def get_habits(
+        self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
+    ) -> HabitsResponse:
+        """.. deprecated:: Use :meth:`list_habits` instead."""
+        return self.list_habits(agent_id, user_id=user_id, instance_id=instance_id)
+
     def create_habit(
         self,
         agent_id: str,
@@ -740,9 +753,10 @@ class Agents:
             f"/api/v1/agents/{agent_id}/habits/{quote(habit_name, safe='')}", params=params
         )
 
-    def get_goals(
+    def list_goals(
         self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
     ) -> GoalsResponse:
+        """List goals for an agent. Pass user_id to get combined agent-global + per-user goals."""
         params: dict[str, Any] = {}
         if user_id:
             params["user_id"] = user_id
@@ -751,6 +765,12 @@ class Agents:
         data = self._http.get(f"/api/v1/agents/{agent_id}/goals", params=params)
         return GoalsResponse.model_validate(data)
 
+    def get_goals(
+        self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
+    ) -> GoalsResponse:
+        """.. deprecated:: Use :meth:`list_goals` instead."""
+        return self.list_goals(agent_id, user_id=user_id, instance_id=instance_id)
+
     def create_goal(
         self,
         agent_id: str,
@@ -936,10 +956,10 @@ class Agents:
         """Delete a constellation node."""
         self._http.delete(f"/api/v1/agents/{agent_id}/constellation/nodes/{node_id}")
 
-    def get_breakthroughs(
+    def list_breakthroughs(
         self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
     ) -> BreakthroughsResponse:
-        """Get breakthroughs for an agent."""
+        """List breakthroughs for an agent."""
         params: dict[str, Any] = {}
         if user_id:
             params["user_id"] = user_id
@@ -948,6 +968,12 @@ class Agents:
         data = self._http.get(f"/api/v1/agents/{agent_id}/breakthroughs", params=params)
         return BreakthroughsResponse.model_validate(data)
 
+    def get_breakthroughs(
+        self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
+    ) -> BreakthroughsResponse:
+        """.. deprecated:: Use :meth:`list_breakthroughs` instead."""
+        return self.list_breakthroughs(agent_id, user_id=user_id, instance_id=instance_id)
+
     def get_wakeups(
         self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
     ) -> WakeupsResponse:
@@ -1400,6 +1426,8 @@ class AsyncAgents:
         generate_origin_story: bool | None = None,
         generate_personalized_memories: bool | None = None,
         initial_goals: list[dict[str, Any]] | None = None,
+        generate_avatar: bool | None = None,
+        capabilities: dict[str, Any] | None = None,
     ) -> Agent:
         """Create a new agent."""
         body: dict[str, Any] = {"name": name}
@@ -1449,6 +1477,10 @@ class AsyncAgents:
             body["generate_personalized_memories"] = generate_personalized_memories
         if initial_goals is not None:
             body["initial_goals"] = initial_goals
+        if generate_avatar is not None:
+            body["generate_avatar"] = generate_avatar
+        if capabilities is not None:
+            body["capabilities"] = capabilities
 
         data = await self._http.post("/api/v1/agents", json_data=body)
         return Agent.model_validate(data)
@@ -1932,9 +1964,10 @@ class AsyncAgents:
         data = await self._http.get(f"/api/v1/agents/{agent_id}/relationships", params=params)
         return RelationshipResponse.model_validate(data)
 
-    async def get_habits(
+    async def list_habits(
         self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
     ) -> HabitsResponse:
+        """List habits for an agent."""
         params: dict[str, Any] = {}
         if user_id:
             params["user_id"] = user_id
@@ -1943,6 +1976,12 @@ class AsyncAgents:
         data = await self._http.get(f"/api/v1/agents/{agent_id}/habits", params=params)
         return HabitsResponse.model_validate(data)
 
+    async def get_habits(
+        self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
+    ) -> HabitsResponse:
+        """.. deprecated:: Use :meth:`list_habits` instead."""
+        return await self.list_habits(agent_id, user_id=user_id, instance_id=instance_id)
+
     async def create_habit(
         self,
         agent_id: str,
@@ -2012,9 +2051,10 @@ class AsyncAgents:
             f"/api/v1/agents/{agent_id}/habits/{quote(habit_name, safe='')}", params=params
         )
 
-    async def get_goals(
+    async def list_goals(
         self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
     ) -> GoalsResponse:
+        """List goals for an agent. Pass user_id to get combined agent-global + per-user goals."""
         params: dict[str, Any] = {}
         if user_id:
             params["user_id"] = user_id
@@ -2023,6 +2063,12 @@ class AsyncAgents:
         data = await self._http.get(f"/api/v1/agents/{agent_id}/goals", params=params)
         return GoalsResponse.model_validate(data)
 
+    async def get_goals(
+        self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
+    ) -> GoalsResponse:
+        """.. deprecated:: Use :meth:`list_goals` instead."""
+        return await self.list_goals(agent_id, user_id=user_id, instance_id=instance_id)
+
     async def create_goal(
         self,
         agent_id: str,
@@ -2210,10 +2256,10 @@ class AsyncAgents:
         """Delete a constellation node."""
         await self._http.delete(f"/api/v1/agents/{agent_id}/constellation/nodes/{node_id}")
 
-    async def get_breakthroughs(
+    async def list_breakthroughs(
         self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
     ) -> BreakthroughsResponse:
-        """Get breakthroughs for an agent."""
+        """List breakthroughs for an agent."""
         params: dict[str, Any] = {}
         if user_id:
             params["user_id"] = user_id
@@ -2222,6 +2268,12 @@ class AsyncAgents:
         data = await self._http.get(f"/api/v1/agents/{agent_id}/breakthroughs", params=params)
         return BreakthroughsResponse.model_validate(data)
 
+    async def get_breakthroughs(
+        self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
+    ) -> BreakthroughsResponse:
+        """.. deprecated:: Use :meth:`list_breakthroughs` instead."""
+        return await self.list_breakthroughs(agent_id, user_id=user_id, instance_id=instance_id)
+
     async def get_wakeups(
         self, agent_id: str, *, user_id: str | None = None, instance_id: str | None = None
     ) -> WakeupsResponse:

{sonzai-1.2.0 → sonzai-1.2.2}/src/sonzai/resources/memory.py

@@ -38,6 +38,7 @@ class Memory:
         parent_id: str | None = None,
         include_contents: bool = False,
         limit: int = 50,
+        scope: str | None = None,
     ) -> MemoryResponse:
         """Get the memory tree for an agent."""
         params: dict[str, Any] = {"limit": limit}
@@ -49,6 +50,8 @@ class Memory:
             params["parent_id"] = parent_id
         if include_contents:
             params["include_contents"] = "true"
+        if scope is not None:
+            params["scope"] = scope
 
         data = self._http.get(f"/api/v1/agents/{agent_id}/memory", params=params)
         return MemoryResponse.model_validate(data)
@@ -122,16 +125,16 @@ class Memory:
         agent_id: str,
         *,
         user_id: str | None = None,
-        category: str | None = None,
+        fact_type: str | None = None,
         limit: int | None = None,
         offset: int | None = None,
     ) -> FactListResponse:
-        """List atomic facts for an agent, optionally filtered by category."""
+        """List atomic facts for an agent, optionally filtered by fact type."""
         params: dict[str, Any] = {}
         if user_id:
             params["user_id"] = user_id
-        if category:
-            params["category"] = category
+        if fact_type:
+            params["fact_type"] = fact_type
         if limit is not None:
             params["limit"] = limit
         if offset is not None:
@@ -272,6 +275,7 @@ class AsyncMemory:
         parent_id: str | None = None,
         include_contents: bool = False,
         limit: int = 50,
+        scope: str | None = None,
     ) -> MemoryResponse:
         params: dict[str, Any] = {"limit": limit}
         if user_id:
@@ -282,6 +286,8 @@ class AsyncMemory:
             params["parent_id"] = parent_id
         if include_contents:
             params["include_contents"] = "true"
+        if scope is not None:
+            params["scope"] = scope
 
         data = await self._http.get(
             f"/api/v1/agents/{agent_id}/memory", params=params
@@ -355,16 +361,16 @@ class AsyncMemory:
         agent_id: str,
         *,
         user_id: str | None = None,
-        category: str | None = None,
+        fact_type: str | None = None,
         limit: int | None = None,
         offset: int | None = None,
     ) -> FactListResponse:
-        """List atomic facts for an agent, optionally filtered by category."""
+        """List atomic facts for an agent, optionally filtered by fact type."""
         params: dict[str, Any] = {}
         if user_id:
             params["user_id"] = user_id
-        if category:
-            params["category"] = category
+        if fact_type:
+            params["fact_type"] = fact_type
         if limit is not None:
             params["limit"] = limit
         if offset is not None:

{sonzai-1.2.0 → sonzai-1.2.2}/src/sonzai/types.py

@@ -46,9 +46,13 @@ class ChatStreamEvent(BaseModel):
     full_content: str = ""
     finish_reason: str = ""
     continuation_token: str = ""
+    response_cookie: str = ""
     message_count: int = 0
     side_effects: dict[str, Any] | None = None
     external_tool_calls: list[ExternalToolCall] = Field(default_factory=list)
+    enriched_context: dict[str, Any] | None = None
+    build_duration_ms: int = 0
+    used_fast_path: bool = False
     error_message: str = ""
     error_code: str = ""
     is_token_error: bool = False
@@ -112,10 +116,34 @@ class AtomicFact(BaseModel):
     atomic_text: str = ""
     fact_type: str = ""
     importance: float = 0.0
+    confidence: float = 0.0
     supersedes_id: str = ""
     session_id: str = ""
+    source_id: str = ""
+    source_type: str = ""
+    sentiment: str = ""
+    entities: list[str] = Field(default_factory=list)
+    inferred_entities: list[str] = Field(default_factory=list)
+    topic_tags: list[str] = Field(default_factory=list)
+    agent_framing: str = ""
+    character_salience: float = 0.0
+    emotional_intensity: float = 0.0
+    relationship_relevance: float = 0.0
+    retention_strength: float = 0.0
+    temporal_relevance: str = ""
+    time_sensitive_at: str = ""
+    episode_id: str = ""
+    event_time: str = ""
+    evidence_message_ids: list[str] = Field(default_factory=list)
+    polarity_group_id: str = ""
+    hit_count: int = 0
+    miss_count: int = 0
+    mention_count: int = 0
+    last_confirmed: str = ""
+    last_retrieved_at: str = ""
     metadata: dict[str, Any] | None = None
     created_at: str | None = None
+    updated_at: str | None = None
 
 
 class MemoryResponse(BaseModel):
@@ -193,6 +221,12 @@ class PersonalityBehaviors(BaseModel):
     conflict_approach: str = ""
 
 
+class TraitPrecision(BaseModel):
+    precision: float = 0.0
+    observation_count: int = 0
+    last_updated_at: str | None = None
+
+
 class PersonalityProfile(BaseModel):
     agent_id: str = ""
     name: str = ""
@@ -210,6 +244,7 @@ class PersonalityProfile(BaseModel):
     preferences: PersonalityPreferences = Field(default_factory=PersonalityPreferences)
     behaviors: PersonalityBehaviors = Field(default_factory=PersonalityBehaviors)
     emotional_tendencies: dict[str, float] = Field(default_factory=dict)
+    trait_precisions: dict[str, TraitPrecision] = Field(default_factory=dict)
     created_at: str | None = None
 
 
@@ -286,23 +321,31 @@ class MoodResponse(BaseModel):
 
     # Forward reference — MoodState is defined later in this module.
     mood: "MoodState" = Field(default_factory=lambda: MoodState())
-    updated_at: str = ""
 
     model_config = {"extra": "allow"}
 
 
 class MoodHistoryEntry(BaseModel):
-    """A single mood snapshot at a point in time."""
+    """A single mood history data point with flat dimensions and deltas."""
 
-    # Forward reference — MoodState is defined later in this module.
-    mood: "MoodState" = Field(default_factory=lambda: MoodState())
+    valence: float = 0.0
+    arousal: float = 0.0
+    tension: float = 0.0
+    affiliation: float = 0.0
+    label: str = ""
+    trigger_type: str = ""
+    trigger_reason: str = ""
+    delta_valence: float = 0.0
+    delta_arousal: float = 0.0
+    delta_tension: float = 0.0
+    delta_affiliation: float = 0.0
     timestamp: str = ""
 
     model_config = {"extra": "allow"}
 
 
 class MoodHistoryResponse(BaseModel):
-    history: list[MoodHistoryEntry] = Field(default_factory=list)
+    entries: list[MoodHistoryEntry] = Field(default_factory=list)
 
     model_config = {"extra": "allow"}
 
@@ -312,8 +355,10 @@ class RelationshipData(BaseModel):
 
     user_id: str = ""
     love_score: float = 0.0
+    chemistry_score: float = 0.0
     narrative: str = ""
     last_update: str = ""
+    updated_at: str = ""
 
     model_config = {"extra": "allow"}
 
@@ -433,8 +478,18 @@ class InterestData(BaseModel):
     """A single agent interest with score."""
 
     topic: str = ""
-    score: float = 0.0
+    score: float = 0.0  # Deprecated: use confidence instead.
     category: str = ""
+    agent_id: str = ""
+    user_id: str = ""
+    confidence: float = 0.0
+    engagement_level: float = 0.0
+    mention_count: int = 0
+    research_status: str = ""
+    research_findings: str = ""
+    last_mentioned_at: str = ""
+    created_at: str = ""
+    updated_at: str = ""
 
     model_config = {"extra": "allow"}
 
@@ -449,9 +504,17 @@ class DiaryEntry(BaseModel):
     """A single diary entry."""
 
     entry_id: str = ""
+    agent_id: str = ""
+    user_id: str = ""
+    date: str = ""
+    content: str = ""
     title: str = ""
-    body: str = ""
+    body_lines: list[str] = Field(default_factory=list)
+    body: str = ""  # Deprecated: use content instead.
+    mood: str = ""
+    topics: list[str] = Field(default_factory=list)
     tags: list[str] = Field(default_factory=list)
+    trigger_type: str = ""
     created_at: str = ""
 
     model_config = {"extra": "allow"}
@@ -515,7 +578,10 @@ class EvalTemplateListResponse(BaseModel):
 
 class EvalRun(BaseModel):
     id: str = ""
+    """SDK alias. The spec wire name is run_id."""
+    run_id: str = ""
     tenant_id: str = ""
+    project_id: str = ""
     agent_id: str = ""
     agent_name: str = ""
     status: str = ""
@@ -538,6 +604,7 @@ class EvalRun(BaseModel):
     evaluation_cost_usd: float = 0.0
     adaptation_template_id: str = ""
     adaptation_template_snapshot: Any = None
+    started_at: str | None = None
     created_at: str | None = None
     completed_at: str | None = None
 
@@ -718,21 +785,25 @@ class WakeupsResponse(BaseModel):
 
 
 class MoodState(BaseModel):
-    """Snapshot of an agent's four-dimensional mood vector."""
+    """Snapshot of an agent's four-dimensional mood vector (valence-arousal-tension-affiliation)."""
 
-    happiness: float = 0.0
-    energy: float = 0.0
-    calmness: float = 0.0
-    affection: float = 0.0
+    valence: float = 0.0
+    arousal: float = 0.0
+    tension: float = 0.0
+    affiliation: float = 0.0
+    label: str = ""
 
     model_config = {"extra": "allow"}
 
 
 class MoodAggregateResponse(BaseModel):
-    average: MoodState = Field(default_factory=MoodState)
-    min: MoodState = Field(default_factory=MoodState)
-    max: MoodState = Field(default_factory=MoodState)
-    data_count: int = 0
+    valence: float = 0.0
+    arousal: float = 0.0
+    tension: float = 0.0
+    affiliation: float = 0.0
+    label: str = ""
+    user_count: int = 0
+    days_window: int = 0
 
     model_config = {"extra": "allow"}
 
@@ -1094,6 +1165,7 @@ class ScheduledWakeup(BaseModel):
     event_description: str = ""
     occasion: str = ""
     interest_topic: str = ""
+    research_summary: str = ""
     executed_at: str | None = None
     created_at: str | None = None
 
@@ -1108,13 +1180,20 @@ class ScheduledWakeup(BaseModel):
 class AgentIndex(BaseModel):
     model_config = {"extra": "allow"}
     id: str = ""
+    agent_id: str = ""
     tenant_id: str = ""
     name: str = ""
     bio: str = ""
     gender: str = ""
     avatar_url: str = ""
     status: str = ""
+    is_active: bool = False
     project_id: str = ""
+    instance_count: int = 0
+    last_seen_at: str = ""
+    owner_user_id: str = ""
+    owner_display_name: str = ""
+    owner_email: str = ""
     created_at: str = ""
 
 
@@ -1122,6 +1201,7 @@ class AgentListResponse(BaseModel):
     items: list[AgentIndex] = Field(default_factory=list)
     next_cursor: str | None = None
     has_more: bool = False
+    total_count: int = 0
 
 
 # ---------------------------------------------------------------------------
@@ -1446,12 +1526,30 @@ class CustomToolDefinition(BaseModel):
     parameters: dict[str, Any] | None = None
 
 
+class PendingCapability(BaseModel):
+    capability: str = ""
+    context: str = ""
+    model_config = {"extra": "allow"}
+
+
 class AgentCapabilities(BaseModel):
     model_config = {"extra": "allow"}
     webSearch: bool = False
     rememberName: bool = False
     imageGeneration: bool = False
     inventory: bool = False
+    knowledgeBase: bool = False
+    knowledgeBaseProjectId: str = ""
+    voiceGeneration: bool = False
+    voiceId: str = ""
+    voiceTier: int = 0
+    voiceUnlockedAt: str | None = None
+    imageUnlockedAt: str | None = None
+    musicGeneration: bool = False
+    musicUnlockedAt: str | None = None
+    videoGeneration: bool = False
+    videoUnlockedAt: str | None = None
+    pendingCapabilities: list[PendingCapability] = Field(default_factory=list)
     customTools: list[CustomToolDefinition] = Field(default_factory=list)
 
 
@@ -1649,6 +1747,9 @@ class KBAnalyticsRule(BaseModel):
     config: Any = None
     enabled: bool = False
     schedule: str = ""
+    last_run_at: str = ""
+    last_run_status: str = ""
+    last_run_duration_ms: int = 0
     created_at: str = ""
     updated_at: str = ""
 
@@ -2991,6 +3092,7 @@ class UpdateCapabilitiesOptions(BaseModel):
     remember_name: bool | None = None
     image_generation: bool | None = None
     inventory: bool | None = None
+    knowledge_base: bool | None = None
 
     model_config = {"extra": "allow"}
 

{sonzai-1.2.0 → sonzai-1.2.2}/uv.lock

@@ -450,7 +450,7 @@ wheels = [
 
 [[package]]
 name = "sonzai"
-version = "1.1.4"
+version = "1.2.1"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },