fuli-context 0.7.4-beta.3 → 0.7.5

package/graph-provider/fuli_graph/store_project_agent_task_staffing.py

@@ -0,0 +1,409 @@
+"""Agent assignment selection, history continuity, and parallel staffing."""
+
+from fastapi import HTTPException
+
+from .project_agent_access import authorize_project_agent
+from .project_agent_models import ProjectAgentProfile
+from .project_agent_task_models import ProjectAgentParallelPlan
+
+
+class StoreProjectAgentTaskStaffing:
+    """Keep task routing's staffing mechanics separate from task persistence."""
+
+    async def _select_agents(self, actor, space, request):
+        if request.staffing_intent == 'unassigned':
+            return [], [], 'explicit_unassigned', ['user explicitly left task unassigned']
+        assignment_rows = await self._assignment_candidates(
+            request.personal_space_id,
+            request.personal_project_id,
+        )
+        if request.lead_agent_id:
+            await authorize_project_agent(
+                self,
+                actor,
+                space,
+                request.personal_project_id,
+                request.lead_agent_id,
+                require_active=True,
+            )
+            matched = [
+                row for row in assignment_rows
+                if row['agent_id'] == request.lead_agent_id
+            ]
+            if not matched:
+                raise HTTPException(status_code=409, detail='lead Agent assignment is inactive')
+            self._require_agent_client_allowed(
+                matched[0],
+                request.source_application,
+            )
+            return matched[:1], matched, 'explicit_agent', ['explicit Agent selected']
+        rows = [
+            row for row in assignment_rows
+            if self._agent_client_allowed(row, request.source_application)
+        ]
+        if request.staffing_intent in {'new_durable', 'temporary'}:
+            return (
+                [],
+                rows,
+                'explicit_temporary_agent'
+                if request.staffing_intent == 'temporary'
+                else 'explicit_new_agent',
+                ['user explicitly requested recruitment'],
+            )
+        normalized_work_kind = request.work_kind.casefold()
+        required = {item.casefold() for item in request.required_capabilities}
+        candidates, history = await self._rank_agent_candidates(rows, request)
+        if not candidates:
+            unavailable_match = any(
+                not self._agent_client_allowed(row, request.source_application)
+                and (
+                    normalized_work_kind in {
+                        item.casefold() for item in row.get('work_kinds', [])
+                    }
+                    or (
+                        required
+                        and required.issubset({
+                            item.casefold()
+                            for item in row.get('capabilities', [])
+                        })
+                    )
+                )
+                for row in assignment_rows
+            )
+            if unavailable_match:
+                return [], rows, 'agent_unavailable', [
+                    'matching Agent is unavailable to source client: '
+                    f'{request.source_application or "other"}',
+                ]
+            return [], rows, 'no_match', ['no active assignment matched exactly']
+        selected = candidates[0]
+        if normalized_work_kind in {
+            item.casefold() for item in selected.get('work_kinds', [])
+        }:
+            match_basis = [
+                f'exact work kind: {request.work_kind}',
+            ]
+            reason = 'exact_work_kind'
+        else:
+            match_basis = ['all required capabilities matched exactly']
+            reason = 'exact_capability'
+        history_basis = self._history_match_basis(
+            history.get(selected['agent_id']),
+            request.work_kind,
+        )
+        if history_basis:
+            match_basis.append(history_basis)
+        return [selected], candidates, reason, match_basis
+
+    async def _rank_agent_candidates(self, rows, request):
+        """Rank already source-eligible assignments with task-history continuity."""
+
+        normalized_work_kind = request.work_kind.casefold()
+        required = {item.casefold() for item in request.required_capabilities}
+        scored = []
+        for row in rows:
+            work_kinds = {item.casefold() for item in row.get('work_kinds', [])}
+            capabilities = {item.casefold() for item in row.get('capabilities', [])}
+            work_match = normalized_work_kind in work_kinds
+            capability_match = required.issubset(capabilities)
+            if not work_match and not (required and capability_match):
+                continue
+            score = (2 if work_match else 0) + (1 if capability_match else 0)
+            scored.append((score, row))
+        if not scored:
+            return [], {}
+        history = await self._historical_agent_outcomes(
+            request.personal_space_id,
+            request.personal_project_id,
+            request.work_kind,
+            [row['agent_id'] for _, row in scored],
+        )
+        scored.sort(
+            key=lambda item: (
+                -item[0],
+                *self._history_sort_key(history.get(item[1]['agent_id'])),
+                item[1]['assigned_at'],
+                item[1]['agent_id'],
+            )
+        )
+        return [item[1] for item in scored], history
+
+    async def _historical_agent_outcomes(
+        self,
+        personal_space_id,
+        personal_project_id,
+        work_kind,
+        agent_ids,
+    ):
+        """Read project/work-kind participation without making history mandatory."""
+
+        if not agent_ids:
+            return {}
+        runtime = getattr(self, 'runtime', None)
+        driver = getattr(runtime, 'driver', None)
+        if driver is None:
+            return {}
+        records, _, _ = await driver.execute_query(
+            '''
+            MATCH (space:FuliSpace {id: $personal_space_id, kind: 'personal'})-
+                  [:HAS_PROJECT_AGENT_TASK]->
+                  (task:FuliProjectAgentTask {
+                    personal_space_id: $personal_space_id,
+                    personal_project_id: $personal_project_id
+                  })-[participant:HAS_PARTICIPANT]->
+                  (agent:FuliProjectAgent)
+            WHERE agent.agent_id IN $agent_ids
+              AND toLower(task.work_kind) = toLower($work_kind)
+            OPTIONAL MATCH (task)-[:HAS_TASK_EVENT]->
+                  (event:FuliProjectAgentTaskEvent)-[:EVENT_AGENT]->(agent)
+            WHERE event.status IN ['completed', 'failed', 'cancelled']
+            RETURN agent.agent_id AS agent_id,
+                   count(DISTINCT task.task_id) AS participation_count,
+                   count(DISTINCT CASE
+                     WHEN participant.status = 'completed'
+                       OR event.status = 'completed'
+                     THEN task.task_id END
+                   ) AS completed_count,
+                   count(DISTINCT CASE
+                     WHEN participant.status = 'failed'
+                       OR event.status = 'failed'
+                     THEN task.task_id END
+                   ) AS failed_count,
+                   count(DISTINCT CASE
+                     WHEN participant.status = 'cancelled'
+                       OR event.status = 'cancelled'
+                     THEN task.task_id END
+                   ) AS cancelled_count,
+                   max(event.created_at) AS last_outcome_at
+            ''',
+            personal_space_id=personal_space_id,
+            personal_project_id=personal_project_id,
+            work_kind=work_kind,
+            agent_ids=list(agent_ids),
+            routing_='r',
+        )
+        result = {}
+        for row in records or []:
+            raw = dict(row)
+            agent_id = raw.get('agent_id')
+            if not agent_id:
+                continue
+            result[agent_id] = {
+                'participation_count': self._metric_int(
+                    raw.get('participation_count')
+                ),
+                'completed_count': self._metric_int(raw.get('completed_count')),
+                'failed_count': self._metric_int(raw.get('failed_count')),
+                'cancelled_count': self._metric_int(
+                    raw.get('cancelled_count')
+                ),
+                'last_outcome_at': raw.get('last_outcome_at'),
+            }
+        return result
+
+    @staticmethod
+    def _metric_int(value):
+        try:
+            return max(0, int(value or 0))
+        except (TypeError, ValueError):
+            return 0
+
+    @classmethod
+    def _history_sort_key(cls, history):
+        history = history or {}
+        completed = cls._metric_int(history.get('completed_count'))
+        failed = cls._metric_int(history.get('failed_count'))
+        cancelled = cls._metric_int(history.get('cancelled_count'))
+        participation = cls._metric_int(history.get('participation_count'))
+        # A successful prior task is the strongest continuity signal. Failed
+        # and cancelled outcomes lower the signal, while participation remains
+        # a deterministic tie-breaker for otherwise equivalent outcomes.
+        outcome_score = completed * 3 - failed * 2 - cancelled
+        return (
+            -outcome_score,
+            -completed,
+            failed,
+            cancelled,
+            -participation,
+        )
+
+    @classmethod
+    def _history_match_basis(cls, history, work_kind):
+        if not history:
+            return None
+        participation = cls._metric_int(history.get('participation_count'))
+        if participation <= 0:
+            return None
+        completed = cls._metric_int(history.get('completed_count'))
+        failed = cls._metric_int(history.get('failed_count'))
+        cancelled = cls._metric_int(history.get('cancelled_count'))
+        return (
+            f'historical continuity for {work_kind}: '
+            f'{participation} prior task(s), '
+            f'{completed} completed, {failed} failed, {cancelled} cancelled'
+        )
+
+    async def _parallel_staffing_candidates(
+        self,
+        request,
+        candidates,
+        participants,
+    ):
+        capacity = len(request.parallel_plan.workstream_boundaries) - len(participants)
+        if capacity <= 0:
+            return []
+        pool = list(candidates)
+        known_ids = {item['agent_id'] for item in pool}
+        for row in await self._assignment_candidates(
+            request.personal_space_id,
+            request.personal_project_id,
+        ):
+            if row['agent_id'] not in known_ids:
+                pool.append(row)
+                known_ids.add(row['agent_id'])
+        eligible = [
+            row
+            for row in pool
+            if self._agent_client_allowed(row, request.source_application)
+        ]
+        ranked, _ = await self._rank_agent_candidates(eligible, request)
+        participant_ids = {item['agent_id'] for item in participants}
+        return [
+            row
+            for row in ranked
+            if row['agent_id'] not in participant_ids
+        ][:capacity]
+
+    @staticmethod
+    def _add_parallel_collaborators(plan, participants, candidates):
+        if not plan.enabled:
+            return []
+        limit = len(plan.workstream_boundaries)
+        participant_ids = {item['agent_id'] for item in participants}
+        added = []
+        for candidate in candidates:
+            if len(participants) >= limit:
+                break
+            agent_id = candidate['agent_id']
+            if agent_id in participant_ids:
+                continue
+            participant = {
+                'agent_id': agent_id,
+                'role': 'collaborator',
+                'assignment_summary': candidate.get('responsibility'),
+            }
+            participants.append(participant)
+            participant_ids.add(agent_id)
+            added.append(candidate)
+        return added
+
+    async def _assignment_candidates(
+        self,
+        personal_space_id,
+        personal_project_id,
+        *,
+        include_temporary=False,
+    ):
+        records, _, _ = await self.runtime.driver.execute_query(
+            '''
+            MATCH (:FuliSpace {id: $personal_space_id, kind: 'personal'})-
+                  [:CONTAINS_PROJECT]->
+                  (:FuliPersonalProject {project_id: $personal_project_id})-
+                  [:HAS_PROJECT_AGENT_ASSIGNMENT]->
+                  (assignment:FuliProjectAgentAssignment {status: 'active'})-
+                  [:ASSIGNED_AGENT]->(agent:FuliProjectAgent {status: 'active'})
+            WHERE agent.agent_type = 'durable'
+               OR ($include_temporary = true AND agent.agent_type = 'temporary')
+            RETURN assignment, agent
+            ORDER BY assignment.assigned_at, agent.agent_id
+            ''',
+            personal_space_id=personal_space_id,
+            personal_project_id=personal_project_id,
+            include_temporary=include_temporary,
+            routing_='r',
+        )
+        result = []
+        for row in records:
+            assignment = dict(row['assignment'])
+            agent = dict(row['agent'])
+            profile = ProjectAgentProfile.model_validate_json(agent['profile_json'])
+            result.append({
+                'agent_id': agent['agent_id'],
+                'assignment_id': assignment.get('assignment_id') or assignment['id'],
+                'responsibility': assignment.get('responsibility'),
+                'work_kinds': list(
+                    assignment.get('work_kinds') or profile.work_kinds
+                ),
+                'capabilities': list(
+                    assignment.get('capabilities') or profile.capabilities
+                ),
+                'model_strategy_override': assignment.get('model_strategy_json'),
+                'executor_policy_override': assignment.get(
+                    'executor_policy_json'
+                ),
+                'profile': profile,
+                'assigned_at': str(assignment.get('assigned_at') or ''),
+            })
+        return result
+
+    async def _explicit_collaborators(self, actor, space, request):
+        if not request.collaborator_agent_ids:
+            return []
+        rows = await self._assignment_candidates(
+            request.personal_space_id,
+            request.personal_project_id,
+        )
+        result = []
+        for agent_id in request.collaborator_agent_ids:
+            await authorize_project_agent(
+                self,
+                actor,
+                space,
+                request.personal_project_id,
+                agent_id,
+                require_active=True,
+            )
+            row = next((item for item in rows if item['agent_id'] == agent_id), None)
+            if not row:
+                raise HTTPException(
+                    status_code=409,
+                    detail=f'collaborator Agent assignment is inactive: {agent_id}',
+                )
+            self._require_agent_client_allowed(
+                row,
+                request.source_application,
+            )
+            result.append(row)
+        return result
+
+    @staticmethod
+    def _agent_client_allowed(candidate, source_application):
+        profile = candidate.get('profile') if isinstance(candidate, dict) else candidate
+        allowed_clients = set(profile.allowed_clients)
+        return (source_application or 'other') in allowed_clients
+
+    @classmethod
+    def _require_agent_client_allowed(cls, candidate, source_application):
+        if cls._agent_client_allowed(candidate, source_application):
+            return
+        agent_id = candidate.get('agent_id') if isinstance(candidate, dict) else None
+        raise HTTPException(
+            status_code=403,
+            detail=(
+                f'project Agent is not available to this source client: '
+                f'{agent_id or "selected Agent"}'
+            ),
+        )
+
+    @staticmethod
+    def _verify_parallel_plan(plan: ProjectAgentParallelPlan, participants: list[dict]):
+        if plan.enabled and len(participants) > len(plan.workstream_boundaries):
+            raise HTTPException(
+                status_code=422,
+                detail='parallel work exceeds declared workstream boundaries',
+            )
+        if plan.enabled and len(participants) < 2:
+            raise HTTPException(
+                status_code=422,
+                detail='parallel work requires at least two active Agents',
+            )