Initial commit

This commit is contained in:
LZH-YS1998
2026-07-01 17:56:31 +08:00
commit d78931979d
731 changed files with 311088 additions and 0 deletions
+1
View File
@@ -0,0 +1 @@
File diff suppressed because it is too large Load Diff
+131
View File
@@ -0,0 +1,131 @@
"""Context loader — assembles all relevant context before routing."""
from __future__ import annotations
from typing import Any
from opc.database.store import OPCStore
from opc.layer5_memory.memory_manager import MemoryManager
from opc.layer5_memory.preference import PreferenceManager
from opc.layer5_memory.secretary_policy import SecretaryPolicyManager
from opc.layer5_memory.capability_manager import CapabilityManager
from opc.layer5_memory.skill_library import SkillLibrary
from opc.layer3_agent.adapters.registry import AdapterRegistry
from opc.layer2_organization.org_engine import OrgEngine
class LoadedContext:
"""Container for all context assembled for a task."""
def __init__(self) -> None:
self.preferences: dict[str, Any] = {}
self.memory: str = ""
self.project_memory: str = ""
self.session_memory: str = ""
self.skills_context: str = ""
self.available_external_agents: list[str] = []
self.external_agent_profiles: list[dict[str, Any]] = []
self.company_profile: str = "corporate"
self.company_profiles: list[str] = ["corporate", "custom"]
self.company_profile_descriptions: dict[str, str] = {}
self.autonomy_preferences: dict[str, Any] = {}
self.autonomy_stats: dict[str, Any] = {}
self.external_sessions: list[dict[str, Any]] = []
self.session_execution_defaults: dict[str, Any] = {}
self.project_id: str | None = None
self.capability_catalog_summary: str = ""
self.secretary_context: str = ""
self.default_channel: str = "cli"
self.origin_chat_id: str = ""
self.origin_thread_id: str = ""
def has_capable_external_agent(self, domains: list[str]) -> bool:
_ = domains
return bool(self.available_external_agents)
class ContextLoader:
"""Loads all relevant context for a task before routing."""
def __init__(
self,
memory: MemoryManager,
preferences: PreferenceManager,
secretary_policies: SecretaryPolicyManager,
skills: SkillLibrary,
capability_manager: CapabilityManager,
adapter_registry: AdapterRegistry,
org_engine: OrgEngine,
store: OPCStore,
) -> None:
self.memory = memory
self.preferences = preferences
self.secretary_policies = secretary_policies
self.skills = skills
self.capability_manager = capability_manager
self.adapters = adapter_registry
self.org_engine = org_engine
self.store = store
async def load(
self,
project_id: str | None = None,
session_id: str | None = None,
domains: list[str] | None = None,
*,
include_project_knowledge: bool = False,
) -> LoadedContext:
ctx = LoadedContext()
ctx.project_id = project_id
ctx.preferences = {}
ctx.autonomy_preferences = {}
ctx.secretary_context = ""
if session_id:
session = await self.store.get_session(session_id)
if session:
defaults = session.metadata.get("execution_defaults", {})
if isinstance(defaults, dict):
ctx.session_execution_defaults = dict(defaults)
ctx.project_memory = await self.memory.build_memory_context(
project_id=project_id,
session_id=None,
include_project_knowledge=include_project_knowledge,
)
ctx.session_memory = (
await self.memory.build_session_prompt_context(
session_id,
include_latest_user_turn=False,
)
if session_id
else ""
)
ctx.memory = "\n\n".join(part for part in (ctx.project_memory, ctx.session_memory) if part)
# Pre-routing catalog: we don't yet know which execution mode
# the user will land in, so mode-restricted skills (e.g. a
# collaboration playbook scoped to company_mode) are hidden
# from this top-level catalog. They surface later via the
# per-turn prompt harness, which does know the execution mode.
ctx.skills_context = self.skills.build_skills_summary(project_id, execution_mode=None)
ctx.capability_catalog_summary = self.capability_manager.build_catalog_summary()
ctx.available_external_agents = self.adapters.list_available()
ctx.external_agent_profiles = self.adapters.describe_all()
ctx.company_profile = self.org_engine.get_company_profile()
ctx.company_profiles = list(self.org_engine.config.org.company_profiles)
ctx.company_profile_descriptions = self.org_engine.get_company_profile_descriptions()
ctx.autonomy_stats = await self.store.get_autonomy_stats(project_id=project_id)
sessions = []
for agent in ctx.available_external_agents:
session = await self.store.get_external_session(agent_type=agent, project_id=project_id or "default")
if session:
sessions.append({
"agent_type": session.agent_type,
"session_id": session.session_id,
"run_mode": session.run_mode,
"status": session.status,
"updated_at": session.updated_at.isoformat(),
"last_activity_at": session.metadata.get("last_activity_at", ""),
"activity_count": session.metadata.get("activity_count", 0),
"pid": session.metadata.get("pid"),
})
ctx.external_sessions = sessions
return ctx
+47
View File
@@ -0,0 +1,47 @@
"""Task router — DEPRECATED.
Mode selection is now determined by user metadata, not LLM-based routing.
This module is retained for backward compatibility only. Importing TaskRouter
still works, but ``route()`` returns a default ModeSelection(TASK_MODE).
"""
from __future__ import annotations
import warnings
from typing import Any
from loguru import logger
from opc.core.models import ExecutionMode, ModeSelection
from opc.layer1_perception.context_loader import LoadedContext
RouterDecision = ModeSelection
class TaskRouter:
"""Deprecated — kept for backward compatibility.
``route()`` now returns a default ``ModeSelection`` with ``TASK_MODE``
without making any LLM calls. Callers should migrate to reading mode
from user metadata directly.
"""
def __init__(self, llm: Any = None) -> None:
self.llm = llm
warnings.warn(
"TaskRouter is deprecated. Mode is now determined by user metadata.",
DeprecationWarning,
stacklevel=2,
)
async def route(
self,
user_message: str,
context: LoadedContext | None = None,
preferences: dict[str, Any] | None = None,
) -> ModeSelection:
logger.debug("TaskRouter.route() called — returning default TASK_MODE")
return ModeSelection(
mode=ExecutionMode.TASK_MODE,
domains=["general"],
)