fuli-context 0.7.6 → 0.7.8

package/graph-provider/fuli_graph/project_agent_memory_routes.py

@@ -0,0 +1,46 @@
+"""Transport boundary for host-independent Agent working memory."""
+
+from typing import Annotated
+
+from fastapi import HTTPException, Query
+
+from .project_agent_memory_models import (
+    ProjectAgentMemoryRecord,
+    ProjectAgentMemoryView,
+    ProjectAgentMemoryWrite,
+)
+from .project_agent_context_models import (
+    ProjectAgentContextRequest, ProjectAgentContextResolution,
+)
+from .task_context_routes import register_task_context_routes
+
+
+def register_project_agent_memory_routes(application, store, Actor):
+    register_task_context_routes(application, store, Actor)
+    @application.post(
+        '/v1/project-agent-context/resolve', response_model=ProjectAgentContextResolution,
+    )
+    async def resolve_context(request: ProjectAgentContextRequest, actor: Actor):
+        return await store.resolve_project_agent_context(actor, request)
+
+    @application.get(
+        '/v1/project-agents/{agent_id}/memory', response_model=ProjectAgentMemoryView,
+    )
+    async def get_memory(
+        agent_id: str,
+        actor: Actor,
+        personal_space_id: Annotated[str, Query(min_length=1, max_length=128)],
+        personal_project_id: Annotated[str, Query(min_length=1, max_length=128)],
+        limit: Annotated[int, Query(ge=1, le=10)] = 1,
+    ):
+        return await store.get_project_agent_memory(
+            actor, personal_space_id, personal_project_id, agent_id, limit=limit,
+        )
+
+    @application.put(
+        '/v1/project-agents/{agent_id}/memory', response_model=ProjectAgentMemoryRecord,
+    )
+    async def write_memory(agent_id: str, request: ProjectAgentMemoryWrite, actor: Actor):
+        if request.agent_id != agent_id:
+            raise HTTPException(status_code=422, detail='Agent memory target does not match path')
+        return await store.write_project_agent_memory(actor, request)

package/graph-provider/fuli_graph/project_agent_models.py

@@ -27,6 +27,7 @@ ProjectAgentReasoningEffort = Literal['default', 'low', 'medium', 'high']
 ProjectAgentExecutorPolicyMode = Literal['flexible', 'locked']
 PROJECT_AGENT_CLIENTS: tuple[SourceApplication, ...] = (
     'codex',
+    'claude',
     'claude_code',
     'cursor',
     'kiro',
@@ -162,6 +163,14 @@ class ProjectAgentUpsert(StrictModel):
     agent_id: str = Field(min_length=1, max_length=128)
     profile: ProjectAgentProfile
 
+    @model_validator(mode='after')
+    def require_dedicated_archive_operation(self):
+        if self.profile.status == 'archived':
+            raise ValueError(
+                'archive Project Agents through the dedicated archive operation'
+            )
+        return self
+
 
 class ProjectAgentAssignmentCreate(StrictModel):
     personal_space_id: str = Field(min_length=1, max_length=128)

package/graph-provider/fuli_graph/project_agent_recruitment_values.py

@@ -0,0 +1,48 @@
+"""Value conversion helpers for durable project Agent recruitment records."""
+
+from .project_agent_models import ProjectAgentProfile
+from .project_agent_task_models import ProjectAgentRecruitmentRecord
+from .provider_values import native_datetime
+
+
+def project_agent_recruitment_record(raw):
+    proposed_profile = ProjectAgentProfile.model_validate_json(
+        raw['proposed_profile_json']
+    )
+    occupation_emoji = raw.get('occupation_emoji') or raw.get(
+        'occupationEmoji'
+    )
+    if not proposed_profile.occupation_emoji and occupation_emoji:
+        proposed_profile = proposed_profile.model_copy(
+            update={'occupation_emoji': occupation_emoji}
+        )
+    fields = {
+        'recruitment_id': raw.get('recruitment_id') or raw['id'],
+        'personal_space_id': raw['personal_space_id'],
+        'personal_project_id': raw['personal_project_id'],
+        'task_id': raw['task_id'],
+        'coordinator_agent_id': raw['coordinator_agent_id'],
+        'hr_agent_id': raw.get('hr_agent_id'),
+        'position_kind': raw['position_kind'],
+        'work_kind': raw['work_kind'],
+        'required_capabilities': list(raw.get('required_capabilities') or []),
+        'reason_code': raw['reason_code'],
+        'reason': raw['reason'],
+        'status': raw['status'],
+        'confirmation_mode': raw['confirmation_mode'],
+        'proposed_agent_id': raw['proposed_agent_id'],
+        'proposed_profile': proposed_profile,
+        'participant_role': raw.get('participant_role') or 'lead',
+        'recruitment_slot': raw.get('recruitment_slot') or 'lead',
+        'trigger_source_application': raw.get('trigger_source_application'),
+        'trigger_source_session_id': raw.get('trigger_source_session_id'),
+        'revision': int(raw.get('revision') or 0),
+        'recruited_agent_id': raw.get('recruited_agent_id'),
+        'created_at': native_datetime(raw['created_at']),
+        'updated_at': native_datetime(raw['updated_at']),
+        'fulfilled_at': native_datetime(raw.get('fulfilled_at')),
+    }
+    for name in ('test_source', 'cleanup_eligible'):
+        if name in ProjectAgentRecruitmentRecord.model_fields:
+            fields[name] = raw.get(name)
+    return ProjectAgentRecruitmentRecord(**fields)

package/graph-provider/fuli_graph/project_agent_routes.py

@@ -5,6 +5,8 @@ from typing import Annotated, Any
 
 from fastapi import FastAPI, HTTPException, Query
 
+from .project_agent_memory_routes import register_project_agent_memory_routes
+
 from .project_agent_models import (
     ProjectAgentAssignmentCreate,
     ProjectAgentAssignmentEnd,
@@ -57,6 +59,8 @@ def register_project_agent_routes(
 ) -> None:
     """Attach Project Agent routes while keeping ``create_app`` orchestration small."""
 
+    register_project_agent_memory_routes(application, store, Actor)
+
     @application.put('/v1/project-agents', response_model=ProjectAgentRecord)
     async def upsert_project_agent(
         request: ProjectAgentUpsert,
@@ -233,8 +237,17 @@ def register_project_agent_routes(
         task_id: str,
         actor: Actor,
         personal_space_id: Annotated[str, Query(min_length=1, max_length=128)],
+        personal_project_id: Annotated[
+            str | None,
+            Query(min_length=1, max_length=128),
+        ] = None,
     ) -> ProjectAgentTaskRecord:
-        return await store.get_project_agent_task(actor, personal_space_id, task_id)
+        return await store.get_project_agent_task(
+            actor,
+            personal_space_id,
+            task_id,
+            personal_project_id=personal_project_id,
+        )
 
     @application.post(
         '/v1/project-agent-tasks/{task_id}/events',
@@ -259,6 +272,10 @@ def register_project_agent_routes(
         personal_space_id: Annotated[str, Query(min_length=1, max_length=128)],
         from_date: Annotated[date, Query(alias='from')],
         to_date: Annotated[date, Query(alias='to')],
+        personal_project_id: Annotated[
+            str | None,
+            Query(min_length=1, max_length=128),
+        ] = None,
     ) -> ProjectAgentActivityResult:
         return await store.get_project_agent_activity(
             actor,
@@ -266,6 +283,7 @@ def register_project_agent_routes(
             agent_id,
             from_date,
             to_date,
+            personal_project_id=personal_project_id,
         )
 
     @application.get(
@@ -296,6 +314,7 @@ def register_project_agent_routes(
     @application.get(
         '/v1/project-agent-recruitment-policy',
         response_model=ProjectAgentRecruitmentPolicyRecord,
+        deprecated=True,
     )
     async def get_project_agent_recruitment_policy(
         actor: Actor,
@@ -309,6 +328,7 @@ def register_project_agent_routes(
     @application.put(
         '/v1/project-agent-recruitment-policy',
         response_model=ProjectAgentRecruitmentPolicyRecord,
+        deprecated=True,
     )
     async def update_project_agent_recruitment_policy(
         request: ProjectAgentRecruitmentPolicyUpdate,

package/graph-provider/fuli_graph/project_agent_task_models.py

@@ -3,7 +3,11 @@ from typing import Literal
 
 from pydantic import AliasChoices, Field, field_validator, model_validator
 
-from .model_validation import reject_credentials, validate_emoji_sequence
+from .model_validation import (
+    normalize_source_session_url,
+    reject_credentials,
+    validate_emoji_sequence,
+)
 from .models import SourceApplication, StrictModel
 from .project_agent_models import (
     ProjectAgentExecutorPolicy,
@@ -42,6 +46,7 @@ ProjectAgentRoutingOutcome = Literal[
 ProjectAgentRoutingReason = Literal[
     'explicit_agent',
     'project_continuity',
+    'project_default',
     'sole_active_assignment',
     'manual_agent_selection',
     'exact_work_kind',
@@ -72,6 +77,16 @@ ProjectAgentOptimizationPriority = Literal[
     'token_and_cost',
     'time',
 ]
+ProjectAgentTokenUsageSource = Literal['executor', 'host', 'dingdong']
+
+_WORKER_EVIDENCE_ALIASES = {
+    'worker_id': 'workerId',
+    'worker_label': 'workerLabel',
+    'worker_occupation_emoji': 'workerOccupationEmoji',
+    'worker_status': 'workerStatus',
+    'token_usage': 'tokenUsage',
+    'worker_runtime': 'workerRuntime',
+}
 
 
 def _unique_text(value: list[str], label: str) -> list[str]:
@@ -83,6 +98,19 @@ def _unique_text(value: list[str], label: str) -> list[str]:
     return normalized
 
 
+def _normalize_worker_evidence_aliases(value):
+    if not isinstance(value, dict):
+        return value
+    normalized = dict(value)
+    for field_name, alias in _WORKER_EVIDENCE_ALIASES.items():
+        if alias not in normalized:
+            continue
+        if field_name in normalized and normalized[field_name] != normalized[alias]:
+            raise ValueError(f'conflicting {field_name} and {alias} values')
+        normalized[field_name] = normalized.pop(alias)
+    return normalized
+
+
 class ProjectAgentParallelPlan(StrictModel):
     enabled: bool = False
     independent_verification: bool = False
@@ -120,6 +148,10 @@ class ProjectAgentTaskSubmit(StrictModel):
     objective: str = Field(min_length=1, max_length=4096)
     work_kind: str = Field(min_length=1, max_length=128)
     required_capabilities: list[str] = Field(default_factory=list, max_length=16)
+    executor_capability_hints: list[str] = Field(
+        default_factory=list,
+        max_length=16,
+    )
     duration: ProjectAgentTaskDuration = 'ongoing'
     staffing_intent: ProjectAgentStaffingIntent = 'reuse_preferred'
     lead_agent_id: str | None = Field(default=None, min_length=1, max_length=128)
@@ -143,6 +175,14 @@ class ProjectAgentTaskSubmit(StrictModel):
     def validate_capabilities(cls, value: list[str]) -> list[str]:
         return _unique_text(value, 'required capabilities')
 
+    @field_validator('executor_capability_hints')
+    @classmethod
+    def validate_executor_capability_hints(
+        cls,
+        value: list[str],
+    ) -> list[str]:
+        return _unique_text(value, 'executor capability hints')
+
     @field_validator('collaborator_agent_ids')
     @classmethod
     def validate_collaborators(cls, value: list[str]) -> list[str]:
@@ -185,6 +225,80 @@ class ProjectAgentTaskParticipantRecord(StrictModel):
     ended_at: datetime | None = None
 
 
+class ProjectAgentTokenUsage(StrictModel):
+    """Cumulative token snapshot reported for one concrete worker run."""
+
+    source: ProjectAgentTokenUsageSource
+    total_tokens: int = Field(
+        ge=0,
+        validation_alias=AliasChoices('total_tokens', 'totalTokens'),
+    )
+    input_tokens: int | None = Field(
+        default=None,
+        ge=0,
+        validation_alias=AliasChoices('input_tokens', 'inputTokens'),
+    )
+    output_tokens: int | None = Field(
+        default=None,
+        ge=0,
+        validation_alias=AliasChoices('output_tokens', 'outputTokens'),
+    )
+    cached_input_tokens: int | None = Field(
+        default=None,
+        ge=0,
+        validation_alias=AliasChoices(
+            'cached_input_tokens',
+            'cachedInputTokens',
+        ),
+    )
+    cache_write_input_tokens: int | None = Field(
+        default=None,
+        ge=0,
+        validation_alias=AliasChoices(
+            'cache_write_input_tokens',
+            'cacheWriteInputTokens',
+        ),
+    )
+    reasoning_output_tokens: int | None = Field(
+        default=None,
+        ge=0,
+        validation_alias=AliasChoices(
+            'reasoning_output_tokens',
+            'reasoningOutputTokens',
+        ),
+    )
+
+
+class ProjectAgentWorkerRuntime(StrictModel):
+    """Host-reported worker location, never the reporter's trusted identity."""
+
+    application: SourceApplication
+    session_id: str | None = Field(
+        default=None, min_length=1, max_length=256,
+        validation_alias=AliasChoices('session_id', 'sessionId'),
+    )
+    session_url: str | None = Field(
+        default=None, max_length=2048,
+        validation_alias=AliasChoices('session_url', 'sessionUrl'),
+    )
+
+    _validate_session_url = field_validator('session_url')(
+        normalize_source_session_url
+    )
+
+    @field_validator('session_id')
+    @classmethod
+    def validate_session_id(cls, value):
+        if value is not None and not value.strip():
+            raise ValueError('worker session ID must not be blank')
+        return value.strip() if value is not None else None
+
+    @model_validator(mode='after')
+    def validate_evidence(self):
+        reject_credentials(self, 'worker runtime')
+        return self
+
+
 class ProjectAgentTaskExecutionSummary(StrictModel):
     """Auditable execution evidence projected onto a real task participant.
 
@@ -203,17 +317,31 @@ class ProjectAgentTaskExecutionSummary(StrictModel):
     source_application: SourceApplication | None = None
     actual_model_provider: str | None = None
     actual_model: str | None = None
+    source_session_id: str | None = None
+    source_session_url: str | None = None
+    tools_used: list[str] | None = Field(default=None, max_length=32)
+    token_usage: ProjectAgentTokenUsage | None = None
     work_summary: str | None = None
     status: ProjectAgentTaskStatus
     worker_id: str | None = None
     worker_label: str | None = None
     worker_occupation_emoji: str | None = None
+    worker_runtime: ProjectAgentWorkerRuntime | None = None
 
     @field_validator('occupation_emoji', 'worker_occupation_emoji', mode='before')
     @classmethod
     def validate_emojis(cls, value):
         return validate_emoji_sequence(value, 'occupation emoji')
 
+    _validate_session_url = field_validator('source_session_url')(
+        normalize_source_session_url
+    )
+
+    @field_validator('tools_used')
+    @classmethod
+    def validate_tools_used(cls, value):
+        return _unique_text(value, 'tools used') if value is not None else None
+
 
 class ProjectAgentTaskEventRecord(StrictModel):
     event_id: str
@@ -224,6 +352,8 @@ class ProjectAgentTaskEventRecord(StrictModel):
     summary: str
     source_application: SourceApplication | None = None
     source_session_id: str | None = None
+    source_session_url: str | None = None
+    tools_used: list[str] | None = Field(default=None, max_length=32)
     actual_model_provider: str | None = None
     actual_model: str | None = None
     actual_executor_id: str | None = None
@@ -232,32 +362,32 @@ class ProjectAgentTaskEventRecord(StrictModel):
     executor_selection_reason: str | None = None
     executor_fallback_reason: str | None = None
     executor_blocked_reason: str | None = None
-    worker_id: str | None = Field(
-        default=None,
-        validation_alias=AliasChoices('worker_id', 'workerId'),
-    )
-    worker_label: str | None = Field(
-        default=None,
-        validation_alias=AliasChoices('worker_label', 'workerLabel'),
-    )
-    worker_occupation_emoji: str | None = Field(
-        default=None,
-        validation_alias=AliasChoices(
-            'worker_occupation_emoji',
-            'workerOccupationEmoji',
-        ),
-    )
-    worker_status: ProjectAgentTaskStatus | None = Field(
-        default=None,
-        validation_alias=AliasChoices('worker_status', 'workerStatus'),
-    )
+    worker_id: str | None = None
+    worker_label: str | None = None
+    worker_occupation_emoji: str | None = None
+    worker_status: ProjectAgentTaskStatus | None = None
+    token_usage: ProjectAgentTokenUsage | None = None
+    worker_runtime: ProjectAgentWorkerRuntime | None = None
     created_at: datetime
 
+    _normalize_worker_aliases = model_validator(mode='before')(
+        _normalize_worker_evidence_aliases
+    )
+
     @field_validator('worker_occupation_emoji', mode='before')
     @classmethod
     def validate_worker_occupation_emoji(cls, value):
         return validate_emoji_sequence(value, 'worker occupation emoji')
 
+    _validate_session_url = field_validator('source_session_url')(
+        normalize_source_session_url
+    )
+
+    @field_validator('tools_used')
+    @classmethod
+    def validate_tools_used(cls, value):
+        return _unique_text(value, 'tools used') if value is not None else None
+
 
 class ProjectAgentRoutingDecisionRecord(StrictModel):
     decision_id: str
@@ -302,6 +432,7 @@ class ProjectAgentTaskRecord(StrictModel):
     objective: str
     work_kind: str
     required_capabilities: list[str]
+    executor_capability_hints: list[str] = Field(default_factory=list)
     duration: ProjectAgentTaskDuration
     staffing_intent: ProjectAgentStaffingIntent
     status: ProjectAgentTaskStatus
@@ -366,6 +497,8 @@ class ProjectAgentTaskActivityCreate(StrictModel):
     actor_kind: Literal['agent', 'human'] = 'agent'
     source_application: SourceApplication | None = None
     source_session_id: str | None = Field(default=None, min_length=1, max_length=256)
+    source_session_url: str | None = Field(default=None, max_length=2048)
+    tools_used: list[str] | None = Field(default=None, max_length=32)
     actual_model_provider: str | None = Field(default=None, min_length=1, max_length=128)
     actual_model: str | None = Field(default=None, min_length=1, max_length=256)
     actual_executor_id: str | None = Field(default=None, min_length=1, max_length=128)
@@ -393,33 +526,46 @@ class ProjectAgentTaskActivityCreate(StrictModel):
         default=None,
         min_length=1,
         max_length=128,
-        validation_alias=AliasChoices('worker_id', 'workerId'),
     )
     worker_label: str | None = Field(
         default=None,
         min_length=1,
         max_length=160,
-        validation_alias=AliasChoices('worker_label', 'workerLabel'),
     )
     worker_occupation_emoji: str | None = Field(
         default=None,
         min_length=1,
         max_length=32,
-        validation_alias=AliasChoices(
-            'worker_occupation_emoji',
-            'workerOccupationEmoji',
-        ),
     )
-    worker_status: ProjectAgentTaskStatus | None = Field(
-        default=None,
-        validation_alias=AliasChoices('worker_status', 'workerStatus'),
+    worker_status: ProjectAgentTaskStatus | None = None
+    token_usage: ProjectAgentTokenUsage | None = None
+    worker_runtime: ProjectAgentWorkerRuntime | None = None
+
+    _normalize_worker_aliases = model_validator(mode='before')(
+        _normalize_worker_evidence_aliases
     )
 
+    @property
+    def reported_client_applications(self):
+        # Worker claims can impose another restriction, never replace the
+        # host identity used for authorization and audit attribution.
+        return (self.source_application, self.worker_runtime.application) \
+            if self.worker_runtime else (self.source_application,)
+
     @field_validator('worker_occupation_emoji', mode='before')
     @classmethod
     def validate_worker_occupation_emoji(cls, value):
         return validate_emoji_sequence(value, 'worker occupation emoji')
 
+    _validate_session_url = field_validator('source_session_url')(
+        normalize_source_session_url
+    )
+
+    @field_validator('tools_used')
+    @classmethod
+    def validate_tools_used(cls, value):
+        return _unique_text(value, 'tools used') if value is not None else None
+
     @model_validator(mode='after')
     def validate_actual_model(self):
         if self.status in {'completed', 'failed'} and not self.agent_id:
@@ -436,6 +582,24 @@ class ProjectAgentTaskActivityCreate(StrictModel):
             raise ValueError('actual execution report requires the participating Agent ID')
         if self.worker_id and not self.agent_id:
             raise ValueError('worker execution report requires the participating Agent ID')
+        if self.worker_status and not self.worker_id:
+            raise ValueError('worker status requires a worker ID')
+        if self.worker_runtime and not self.worker_id:
+            raise ValueError('worker runtime requires a worker ID')
+        if self.token_usage and not self.agent_id:
+            raise ValueError('token usage report requires the participating Agent ID')
+        if (self.source_session_url or self.tools_used) and not self.agent_id:
+            raise ValueError(
+                'execution evidence requires the participating Agent ID'
+            )
+        if (
+            self.token_usage
+            and self.token_usage.source == 'executor'
+            and not self.actual_executor_id
+        ):
+            raise ValueError(
+                'executor token usage requires the actual executor report'
+            )
         if (self.worker_label or self.worker_occupation_emoji) and not self.worker_id:
             raise ValueError(
                 'worker label or occupation emoji requires a worker ID'
@@ -460,6 +624,8 @@ class ProjectAgentRecruitmentRecord(StrictModel):
     confirmation_mode: ProjectAgentRecruitmentConfirmationMode
     proposed_agent_id: str
     proposed_profile: ProjectAgentProfile
+    participant_role: ProjectAgentParticipantRole = 'lead'
+    recruitment_slot: str = Field(default='lead', min_length=1, max_length=128)
     trigger_source_application: SourceApplication | None = None
     trigger_source_session_id: str | None = None
     revision: int = Field(ge=0)
@@ -475,6 +641,7 @@ class ProjectAgentTaskRouteResult(StrictModel):
     task: ProjectAgentTaskRecord
     assigned_agent: ProjectAgentRecord | None = None
     recruitment: ProjectAgentRecruitmentRecord | None = None
+    recruitments: list[ProjectAgentRecruitmentRecord] = Field(default_factory=list)
     decision: Literal[
         'reused',
         'recruited',
@@ -490,6 +657,13 @@ class ProjectAgentRecruitmentPolicyRecord(StrictModel):
     personal_space_id: str
     confirmation_mode: ProjectAgentRecruitmentConfirmationMode = 'automatic'
     updated_at: datetime | None = None
+    policy_status: Literal['superseded'] = 'superseded'
+    applies_to_recruitment: Literal[False] = False
+    warning: str = (
+        'Legacy storage only; this setting does not control recruitment. '
+        'Use update_project_agent_coordination_policy with the exact project '
+        'and askBeforeRecruitment to change effective authorization.'
+    )
 
 
 class ProjectAgentRecruitmentPolicyUpdate(StrictModel):
@@ -521,12 +695,26 @@ class ProjectAgentActivityTask(StrictModel):
     actual_model_provider: str | None = None
     actual_model: str | None = None
     actual_executor_id: str | None = None
+    source_session_id: str | None = None
+    source_session_url: str | None = Field(default=None, max_length=2048)
+    tools_used: list[str] | None = Field(default=None, max_length=32)
+    token_usage: ProjectAgentTokenUsage | None = None
+    worker_runtime: ProjectAgentWorkerRuntime | None = None
     matched_executor_rule_id: str | None = None
     worker_id: str | None = None
     worker_label: str | None = None
     worker_occupation_emoji: str | None = None
     worker_status: ProjectAgentTaskStatus | None = None
 
+    _validate_session_url = field_validator('source_session_url')(
+        normalize_source_session_url
+    )
+
+    @field_validator('tools_used')
+    @classmethod
+    def validate_tools_used(cls, value):
+        return _unique_text(value, 'tools used') if value is not None else None
+
 
 class ProjectAgentActivityDay(StrictModel):
     date: date

package/graph-provider/fuli_graph/runtime.py

@@ -108,11 +108,37 @@ class LocalLexicalReranker(CrossEncoderClient):
         )
 
 
+class ManagedNeo4jDriver(Neo4jDriver):
+    """Let GraphitiRuntime own index initialization exactly once.
+
+    graphiti-core schedules an eager background index build whenever its driver
+    is constructed inside a running event loop. Fuli has an explicit awaited
+    initialization sequence, so the dependency's untracked task would race it
+    and can leak unawaited query coroutines when one index request fails.
+    """
+
+    def __init__(self, *args, **kwargs):
+        self._fuli_constructing = True
+        try:
+            super().__init__(*args, **kwargs)
+        finally:
+            self._fuli_constructing = False
+
+    def build_indices_and_constraints(self, delete_existing: bool = False):
+        if self._fuli_constructing:
+            return _completed_index_build()
+        return super().build_indices_and_constraints(delete_existing)
+
+
+async def _completed_index_build() -> None:
+    return None
+
+
 class GraphitiRuntime:
     def __init__(self, settings: Settings):
         self.settings = settings
         self.embedder = LocalHashEmbedder(settings.embedding_dim)
-        driver = Neo4jDriver(
+        driver = ManagedNeo4jDriver(
             uri=settings.neo4j_uri,
             user=settings.neo4j_user,
             password=settings.neo4j_password,
@@ -247,6 +273,16 @@ class GraphitiRuntime:
             'FOR (n:FuliProjectAgent) REQUIRE n.id IS UNIQUE',
             'CREATE CONSTRAINT fuli_project_agent_assignment_id IF NOT EXISTS '
             'FOR (n:FuliProjectAgentAssignment) REQUIRE n.id IS UNIQUE',
+            'CREATE CONSTRAINT fuli_project_agent_memory_id IF NOT EXISTS '
+            'FOR (n:FuliProjectAgentMemory) REQUIRE n.id IS UNIQUE',
+            'CREATE CONSTRAINT fuli_project_agent_memory_checkpoint_id IF NOT EXISTS '
+            'FOR (n:FuliProjectAgentMemoryCheckpoint) REQUIRE n.id IS UNIQUE',
+            'CREATE CONSTRAINT fuli_task_context_session_id IF NOT EXISTS '
+            'FOR (n:FuliTaskContextSession) REQUIRE n.id IS UNIQUE',
+            'CREATE CONSTRAINT fuli_task_context_id IF NOT EXISTS '
+            'FOR (n:FuliTaskContext) REQUIRE n.id IS UNIQUE',
+            'CREATE INDEX fuli_task_context_agent_scope IF NOT EXISTS '
+            'FOR (n:FuliTaskContext) ON (n.personal_space_id, n.project_agent_id)',
             'CREATE CONSTRAINT fuli_project_agent_task_id IF NOT EXISTS '
             'FOR (n:FuliProjectAgentTask) REQUIRE n.id IS UNIQUE',
             'CREATE CONSTRAINT fuli_project_agent_task_event_id IF NOT EXISTS '

package/graph-provider/fuli_graph/store.py

@@ -4,17 +4,23 @@ from .runtime import GraphitiRuntime
 from .store_identity import StoreIdentity
 from .store_knowledge import StoreKnowledge
 from .store_project_agent_executors import StoreProjectAgentExecutors
+from .store_project_agent_memory import StoreProjectAgentMemory
+from .store_project_agent_context import StoreProjectAgentContext
 from .store_project_agents import StoreProjectAgents
 from .store_project_agent_tasks import StoreProjectAgentTasks
 from .store_projects import StoreProjects
 from .store_publication import StorePublication
 from .store_records import StoreRecords
+from .store_task_contexts import StoreTaskContexts
 
 
 class GraphStore(
     StoreIdentity,
     StoreProjects,
     StoreProjectAgents,
+    StoreProjectAgentMemory,
+    StoreProjectAgentContext,
+    StoreTaskContexts,
     StoreProjectAgentExecutors,
     StoreProjectAgentTasks,
     StorePublication,