Coverage for agentos/__init__.py: 100%
121 statements
« prev ^ index » next coverage.py v7.14.3, created at 2026-07-06 12:29 +0800
« prev ^ index » next coverage.py v7.14.3, created at 2026-07-06 12:29 +0800
1"""NexusAgent - Production-grade Agent Framework SDK
3v1.12.1: Async Parallel Primitives (fan-out/fan-in, parallel_gather, parallel_map, structured concurrency).
4v1.12.0: Letta-style Virtual Memory Pager (page-out/page-in + swap store + smart recall).
5v1.11.0: Background Task Manager + Agent Supervision Tree + Full Checkpoint Integration + Auto-Context Paging.
6v1.10.0: All-in — Deploy (Docker/K8s) + Eval (SWE-bench/GAIA) + Multimodal (Vision/Audio) + Prompt Hub (versioned) + Cost Tracker (pricing).
7v1.9.9: GuardPipeline (PII/Injection/Toxicity safety with strict/permissive modes).
8v1.9.5: CodeSandbox (safe code gen + test case validation) + Human-in-the-Loop breakpoints.
9v1.9.4: TaskDecomposer + ResultFusion + EvalFeedbackLoop (P0 three-bottleneck fix).
10v1.9.3: CompositeScorer V2 (BLEU smoothing + LLM-as-Judge), 50+ benchmark cases, 80% pass rate.
11v1.9.2: Swarm MESH 5x parallel acceleration, CompositeScorer (ROUGE-L+BLEU+contains+exact), 14 built-in benchmarks, AutoPilot self-healing.
12v1.7.1: System layer + Desktop client: visual approval engine (Agent applies → user clicks allow/deny), native desktop shell (pywebview wrapper), tiered file/shell/browser ops。
13v1.0.0: Production release — ToolUsingAgent CLI, Mock fallback mode,
14weather demo (`agentos demo`), unified LLM Provider abstraction,
15streaming/retry/checkpoint/resume, PyPI + TestPyPI dual publish.
17v1.3.38: +Tool-Using Agent streaming/retry/checkpoint/resume(run_stream/重试逻辑/断点恢复),
18+MockLLMProvider 集成测试支持。11+10 条 Agent 测试全过。
19v1.3.37: +Tool-Using Agent (agentos.agent) — 基于 LLM Function Calling 的自主 Agent 循环:
20ToolExecutor 工具注册/执行、多步推理闭环、同步/异步运行、成本追踪、端到端天气 Agent 示例。
21v1.3.36: +LLM Provider Module (agentos.llm) — unified abstraction with OpenAI/DeepSeek/Anthropic
22providers, Function Calling / Tool Use, streaming, cost estimation. 零 SDK 依赖 AnthropicProvider
23(pure httpx).
24v1.3.15: +SubAgent Parent-Child Communication (SharedState, ChildContext/ChildHandle, heartbeat,
25lifecycle management: pause/resume/cancel/timeout, heartbeat monitoring).
26v1.3.14: +OpenTelemetry Integration (otel_bridge: OtelConfig/OtelTracer/OtelMeter/OtelMiddleware).
27v1.3.13: +A2A Protocol v2 (Task Store with InMemory/SQLite backends, Streaming SSE task lifecycle
28notifications, enhanced A2AClient with retry/auth/connection pooling, A2AServer with FastAPI
29route builder + streaming + auth + pluggable persistence).
30v1.3.12: +Prompt Optimizer (DSPy-inspired iterative refinement, bootstrapping, multi-strategy),
31+Few-Shot Selector (similarity/diversity/label-balanced strategies),
32+SSE Streaming (ASGI SSE with heartbeats, backpressure, typed events).
33v1.3.11: +Guardrails (Input/Output safety engine with PII/Injection/Keyword/Toxicity rules, PolicyEnforcer),
34+HITL (Human-in-the-Loop approval workflows with RiskLevel auto-decision, caching, preset policies).
35v1.3.10: +Conversation Manager (multi-turn dialog, sliding window, branching, summarization).
36v1.3.9: +Schema Enforcer (Pydantic output validation/auto-repair).
37v1.3.8: +Quality (docstrings, bare-except/type-ignore fixes).
39v1.4.0: +End-to-end examples (multi_agent_research.py, file_ops_agent.py),
40+Professional README with feature comparison table,
41+CLI demo upgraded with self-check mode, +Agent marketplace listing.
42"""
44__version__ = "1.16.1"
46# v1.15.0: Tool Output Validation Layer (structured result validation + error classification + auto-repair suggestions).
47# v1.14.9: Memory Persistence Checkpoint Delivery - all 6 memory subsystems get_state()/restore_state() + ServerDaemon lifecycle integration + DaemonConfig + /api/daemon/memory endpoint.
48# v1.14.8: P0 regression fix (imports, version, dependencies).
50# Core - DI system
51# Tool-Using Agent (v1.3.38)
52from agentos.agent import (
53 AgentConfig,
54 AgentResult,
55 AgentStep,
56 MockLLMProvider,
57 ToolAgent,
58 ToolExecutor,
59)
61# Agent Marketplace (v1.3.0)
62from agentos.agents.market import (
63 AgentCategory,
64 AgentMarket,
65 AgentSkill,
66)
68# API Middleware (v1.3.0)
69from agentos.api.middleware import (
70 AuthConfig,
71 CORSConfig,
72 CORSMiddleware,
73 RequestContext,
74 RequestIDMiddleware,
75)
77# API Server (v1.3.2)
78from agentos.api.server import (
79 AgentAPI,
80 RunRequest,
81 RunResponse,
82)
84# API Streaming (v1.3.0)
85from agentos.api.streaming import (
86 StreamEvent,
87 StreamingAgent,
88 StreamSession,
89)
91# API Versioning (v1.3.0)
92from agentos.api.versioning import (
93 APIVersion,
94 VersionConfig,
95 VersionNegotiator,
96 VersionStrategy,
97)
99# Benchmark Runner (v1.3.0)
100from agentos.benchmarks.runner import (
101 BenchmarkConfig,
102 BenchmarkReport,
103 BenchmarkRunner,
104 BenchmarkScenario,
105)
107# Cache - LLM Response Cache (v1.2.7)
108from agentos.cache import (
109 BaseEmbedder,
110 CacheEntry,
111 CacheKeyStrategy,
112 CohereEmbedder,
113 LLMCache,
114 LocalEmbedder,
115 OpenAIEmbedder,
116 ResponseCache,
117)
119# CLI Init (v1.4.1)
120from agentos.cli.init import (
121 scaffold,
122)
124# CLI Main (v1.0.0)
125from agentos.cli.main import main as cli_main
127# CLI Serve (v1.3.0)
128from agentos.cli.serve import (
129 ServeConfig,
130 start_api_server,
131)
133# Communication layer
134from agentos.comm.layer import (
135 Blackboard,
136 CommunicationLayer,
137 EventBus,
138 Mailbox,
139)
141# Concurrency (v1.1.3)
142from agentos.concurrency.batch import (
143 AsyncBatchExecutor,
144 BatchConfig,
145 BatchResult,
146 BatchStrategy,
147 TaskSpec,
148)
149from agentos.concurrency.batch import (
150 TaskResult as BatchTaskResult,
151)
152from agentos.concurrency.batch import (
153 TaskStatus as BatchTaskStatus,
154)
156# Config System (v1.2.7)
157from agentos.config import (
158 AgentOSConfig,
159 AgentOSPreset,
160 ValidationResult,
161)
163# Core extensions (v1.2.9)
164from agentos.core import (
165 AgentContext,
166 AgentState,
167 AgentStateMachine,
168 AsyncAgentLoop,
169 AsyncContextManager,
170 AsyncInvocationResult,
171 AsyncLoopConfig,
172 ContextManager,
173 CoreMessage,
174 CoreToolCall,
175 CoreToolResult,
176 ResponseCollector,
177 Session,
178 SessionStore,
179 StateTimeoutError,
180 StateTransition,
181 StreamChunk,
182 StreamEmitter,
183 StreamEvent,
184 TransitionError,
185)
187# Core - CodeAgent
188from agentos.core.code_agent import (
189 CodeAgent,
190 CodeResult,
191 CodeStep,
192)
194# Cost Tracking (v1.20.0)
195from agentos.core.cost_tracker import (
196 CostTracker,
197 ModelPricing,
198 UsageRecord,
199)
200from agentos.core.di import (
201 Agent,
202 Depends,
203 RunContext,
204 inject_tool,
205 requires_context,
206)
208# Feature Flags (v1.20.0)
209# Core - Handoff protocol
210from agentos.core.handoff import (
211 Handoff,
212 HandoffAwareAgent,
213 HandoffResult,
214 can_handle,
215 execute_with_handoff,
216 transfer_to,
217)
219# Agent Loop (v1.3.2)
220from agentos.core.loop import (
221 AgentLoop,
222 AgentResult,
223 HumanInterruptNeeded,
224 LoopConfig,
225 LoopState,
226 MaxIterationsExceeded,
227)
229# Core - Middleware Pipeline (v1.2.7)
230from agentos.core.middleware import (
231 AgentMiddleware,
232 MiddlewareContext,
233 MiddlewareDecision,
234 MiddlewarePhase,
235 MiddlewarePipeline,
236)
238# Cost - Token Counter (v1.2.9)
239from agentos.cost import (
240 CostEstimate,
241 ModelFamily,
242 TokenCount,
243 TokenCounter,
244)
246# Cost tracking (v1.1.4)
247from agentos.cost.tracker import (
248 PRICING,
249 CostTracker,
250 ModelPricing,
251 RunCostSession,
252 UsageRecord,
253)
255# Deployment (v1.2.8)
256from agentos.deployment import (
257 ComposeConfig,
258 ComposeService,
259 DockerConfig,
260)
262# Docs Generator (v1.3.2)
263from agentos.docs.generator import (
264 DocConfig,
265 generate_api_docs,
266 generate_quickstart,
267)
269# Errors (v1.2.8)
270from agentos.errors import (
271 ErrorCategory,
272 ErrorContext,
273 ErrorFormatter,
274 HumanError,
275)
277# Evaluation Framework (v1.3.18)
278from agentos.evaluation import (
279 EvalReport,
280 Evaluator,
281 Scorer,
282)
283from agentos.evaluation.scorers import (
284 CompositeScorer,
285)
287# Evolution engine
288from agentos.evolution.engine import (
289 EvolutionEngine,
290 EvolutionProposal,
291 EvolutionStatus,
292)
293from agentos.experiments import (
294 Evaluator as ExperimentEvaluator,
295)
297# Experiments (v1.2.8)
298from agentos.experiments import (
299 ExperimentConfig,
300 ExperimentReport,
301 ExperimentRunner,
302 PromptVariant,
303 TrialResult,
304)
306# Feedback (v1.2.8)
307from agentos.feedback import (
308 FeedbackCollector,
309 FeedbackRecord,
310 FeedbackType,
311 PreferenceLearner,
312)
314# Health (v1.2.9)
315from agentos.health import (
316 CheckResult,
317 HealthCheck,
318 HealthChecker,
319 HealthStatus,
320)
322# Logging (v1.2.9)
323from agentos.log import (
324 JSONFormatter,
325 TraceContext,
326)
328# MCP Package (v1.3.6)
329from agentos.mcp import (
330 MCPClient as MCPFullClient,
331)
332from agentos.mcp import (
333 MCPError,
334 MCPPromptDef,
335 MCPPromptInfo,
336 MCPResource,
337 MCPResourceInfo,
338 # MCP Server (v1.5.2)
339 MCPServer,
340 MCPToolDef,
341 MCPToolInfo,
342 connect_mcp_servers,
343 create_default_server,
344 start_mcp_server,
345)
346from agentos.mcp import (
347 MCPServerConfig as MCPConfig,
348)
349from agentos.mcp.adapter import (
350 MCPToolAdapter,
351 MCPToolRegistry,
352)
354# Built-in MCP Servers (v1.7.8)
355from agentos.mcp.builtin_servers import (
356 BuiltinMCPRegistry,
357 FilesystemServer,
358 MemoryServer,
359 WebFetchServer,
360 create_default_registry,
361)
363# Memory - Retriever + Conversation (v1.2.7)
364# Memory extensions (v1.2.8)
365# Memory - Compressor (v1.2.9)
366from agentos.memory import (
367 ContextCompressor,
368 ConversationMemory,
369 ImportanceScorer,
370 LongTermMemory,
371 MemoryChunk,
372 MemoryStore,
373 MemorySummarizer,
374 SemanticMemoryRetriever,
375 VectorMemory,
376 WorkingMemory,
377 WorkingMemoryItem,
378)
380# Memory pyramid
381from agentos.memory.pyramid import (
382 MemoryItem,
383 MemoryLayer,
384 MemoryPyramid,
385 MemoryType,
386)
388# Anthropic Claude Backend (v1.3.5)
389from agentos.models.backends.anthropic import (
390 ClaudeClient,
391 ClaudeConfig,
392)
394# Gemini Backend (v1.3.1)
395from agentos.models.backends.gemini import (
396 GeminiClient,
397 GeminiConfig,
398 GeminiSafetySetting,
399)
401# Ollama Backend (v1.3.5)
402from agentos.models.backends.ollama import (
403 OllamaClient,
404 OllamaConfig,
405)
407# OpenAI Backend (v1.3.5)
408from agentos.models.backends.openai import (
409 OpenAIClient,
410 OpenAIConfig,
411)
413# Models - Resilience (v1.1.5)
414from agentos.models.resilience import (
415 CancellationSource,
416 CancelledError,
417 CircuitBreaker,
418 CircuitBreakerConfig,
419 ResilienceConfig,
420 ResilientCall,
421 RetryConfig,
422 retry_with_backoff,
423 with_fallback,
424 with_timeout,
425)
427# Models - Router (v1.2.7 minimal)
428# Model Route Types (v1.3.1)
429# Model Config (v1.3.2)
430from agentos.models.router import (
431 RECOMMENDED_CONFIG,
432 AllModelsFailed,
433 ModelConfig,
434 ModelResponse,
435 ModelRouter,
436 ModelSpec,
437)
439# Models - Routing Strategy (v1.2.8)
440from agentos.models.routing_strategy import (
441 Budget,
442 Complexity,
443 RoutingStrategy,
444)
446# Monitoring (v1.2.8)
447from agentos.monitoring import (
448 Alert,
449 AlertEvaluator,
450 AlertRule,
451 AlertSeverity,
452 AlertState,
453 MonitoringConfig,
454 WebhookConfig,
455 WebhookDispatcher,
456)
458# Multimodal (v1.2.7)
459from agentos.multimodal import (
460 Modality,
461 MultimodalManager,
462)
464# Observability (v1.2.7)
465from agentos.observability import (
466 BudgetAlert,
467 CostAnalytics,
468 MetricsCollector,
469 NoopTracer,
470 Tracer,
471)
473# Orchestration extensions (v1.2.8)
474from agentos.orchestration import (
475 A2ARouter,
476 AgentGraph,
477 GraphNodeState,
478 GraphRecipe,
479 GraphResult,
480 RouterAgentCard,
481 RouterTask,
482 TaskResult,
483 TaskStatus,
484)
486# Orchestration
487from agentos.orchestration.graph import (
488 GraphEdge,
489 GraphNode,
490 GraphOrchestrator,
491)
493# Plugin Manager (v1.2.9)
494from agentos.plugin_manager import (
495 PluginInfo,
496 PluginManager,
497)
499# Plugins - Plugin System (v1.2.7)
500from agentos.plugins import (
501 DiscoveredPlugin,
502 LifecycleManager,
503 PluginDiscovery,
504 PluginLoader,
505 PluginRegistry,
506 PluginStatus,
507 RegisteredPlugin,
508)
510# Prompts (v1.2.7)
511from agentos.prompts import (
512 PromptRegistry,
513 PromptTemplate,
514)
516# Protocols - Contracts (v1.2.9)
517from agentos.protocols import (
518 AgentCapability,
519 AgentContract,
520 CapabilityDomain,
521 CapabilityMatcher,
522 ContractRegistry,
523 MatchScore,
524 QoSLevel,
525)
527# Protocols - A2A
528from agentos.protocols.a2a import (
529 A2AArtifact,
530 A2AClient,
531 A2AHandoff,
532 A2AMessage,
533 A2AServer,
534 A2ASession,
535 A2ATask,
536 DataPart,
537 FilePart,
538 TaskState,
539 TextPart,
540 new_handoff,
541 new_task,
542)
544# Protocols - Agent Card
545from agentos.protocols.agent_card import (
546 AgentCard,
547 AgentCardDiscovery,
548 AgentCardRegistry,
549 create_card,
550 discover_local,
551)
553# MCP Protocol (v1.2.7)
554from agentos.protocols.mcp import (
555 MCPClient,
556 MCPServerConfig,
557 MCPToolSchema,
558)
560# Protocols - Structured output validation
561from agentos.protocols.output import (
562 OutputValidator,
563 StructuredOutput,
564 validate_output,
565)
567# Queue - Task Queue & Rate Limiter (v1.2.7)
568from agentos.queue import (
569 RateLimitConfig,
570 RateLimiter,
571 RateLimitStrategy,
572 TaskPriority,
573 TaskQueue,
574)
575from agentos.queue import (
576 TaskState as QueueTaskState,
577)
579# RAG Pipeline (v1.3.5)
580from agentos.rag import (
581 ChunkConfig,
582 EmbeddingConfig,
583 RAGPipeline,
584 TextChunker,
585)
587# Security extensions (v1.9.9)
588from agentos.security import (
589 ContentSafetyFilter,
590 GuardAction,
591 GuardChainResult,
592 GuardPipeline,
593 GuardResult,
594 InputGuard,
595 LLMSafetyAnalyzer,
596 OutputGuard,
597 PIIDetector,
598 RiskLevel,
599 SafetyReport,
600 Sandbox,
601 SandboxManager,
602 Severity,
603 create_permissive_guard,
604 create_strict_guard,
605)
607# Security - Auditor (v1.2.7)
608from agentos.security.auditor import (
609 AuditFinding,
610 AuditReport,
611 SecurityAuditor,
612)
614# Content Safety Guardrails (v1.20.0)
615from agentos.security.guardrails import (
616 GuardAction,
617 GuardResult,
618)
620# Security - Sandbox (v1.2.1)
621from agentos.security.sandbox_executor import (
622 DockerSandbox,
623 ProcessSandbox,
624 SandboxExecutor,
625 SandboxMode,
626 SandboxResult,
627)
629# MCP Server (v1.3.0)
630from agentos.server.mcp_server import (
631 MCPPrompt,
632 MCPResource,
633 MCPServer,
634 MCPServerConfig,
635 MCPTool,
636)
638# Storage (v1.2.9)
639from agentos.storage import (
640 CheckpointStore,
641 SqliteStore,
642)
644# SubAgent Manager (v1.2.9) + Parent-Child Communication (v1.3.15)
645from agentos.subagent import (
646 ChildContext,
647 ChildHandle,
648 ChildHeartbeat,
649 ChildInfo,
650 ChildStatus,
651 SharedState,
652 SubAgentManager,
653 SubAgentMode,
654 SubAgentResult,
655 SubAgentSpec,
656)
658# Swarm Patterns (v1.2.8)
659from agentos.swarm import (
660 CollaborationConfig,
661 CollaborationResult,
662 SwarmPatterns,
663 Topology,
664)
666# Code Sandbox (v1.9.5)
667from agentos.swarm.code_sandbox import (
668 CodeFeedbackExtractor,
669 CodeSandbox,
670)
671from agentos.swarm.code_sandbox import (
672 SandboxResult as CodeSandboxResult,
673)
674from agentos.swarm.code_sandbox import (
675 TestCase as CodeTestCase,
676)
678# Swarm coordinator
679# Swarm Coordinator extensions (v1.3.2)
680from agentos.swarm.coordinator import (
681 AgentRole,
682 ExecutionMode,
683 MessageBus,
684 SmartSwarmCoordinator,
685 SwarmCoordinator,
686 SwarmMessage,
687 SwarmResult,
688 SwarmTopology,
689)
691# Human-in-the-Loop (v1.9.5)
692from agentos.swarm.human_loop import (
693 Breakpoint,
694 BreakpointType,
695 HITLConfig,
696 HITLManager,
697 HumanDecision,
698)
700# Testing Fixtures (v1.3.0)
701from agentos.testing.fixtures import (
702 MockLLMClient,
703 MockLLMResponse,
704 mock_model_response,
705 mock_openai_client,
706 sample_config,
707)
709# Tools extensions (v1.2.9)
710from agentos.tools import (
711 BaseTool,
712 BaseToolCall,
713 BaseToolResult,
714 FCToolCall,
715 FCToolRegistry,
716 FCToolResult,
717 GeneratedTool,
718 OpenAPIToolGenerator,
719 PermissionLevel,
720 ToolRegistry,
721 ToolSchema,
722)
724# Concrete Tools (v1.3.0)
725from agentos.tools.code_agent import (
726 CodeAgentTool,
727 ShellTool,
728)
729from agentos.tools.file_tools import (
730 ListDirectoryTool,
731 ReadFileTool,
732 WriteFileTool,
733)
735# Fusion toolkit
736from agentos.tools.fusion import (
737 FusionResult,
738 FusionToolkit,
739 ToolSpec,
740)
742# Tools - Orchestrator (v1.2.7)
743from agentos.tools.orchestrator import (
744 DAGBuilder,
745 DAGSpec,
746 ToolOrchestrator,
747)
749# Tool risk rating (v1.1.4)
750from agentos.tools.risk import (
751 ToolRiskLevel,
752 ToolRiskRating,
753 get_risk_preset,
754 infer_risk_level,
755)
756from agentos.tools.web_tools import (
757 WebFetchTool,
758)
760# Vector Store (v1.2.7)
761from agentos.vectorstore import (
762 BaseVectorStore,
763 ChromaVectorStore,
764 FAISSVectorStore,
765)
767# Workflows (v1.2.7)
768from agentos.workflows import (
769 WorkflowEngine,
770 WorkflowTemplate,
771)
773__all__ = [
774 # Version
775 "__version__",
776 # Core DI
777 "Agent",
778 "RunContext",
779 "Depends",
780 "inject_tool",
781 "requires_context",
782 # Handoff
783 "Handoff",
784 "HandoffResult",
785 "transfer_to",
786 "can_handle",
787 "execute_with_handoff",
788 "HandoffAwareAgent",
789 # CodeAgent
790 "CodeAgent",
791 "CodeResult",
792 "CodeStep",
793 # Tool-Using Agent (v1.3.38)
794 "ToolAgent",
795 "ToolExecutor",
796 "AgentConfig",
797 "AgentStep",
798 "AgentResult",
799 "MockLLMProvider",
800 # Structured output
801 "StructuredOutput",
802 "validate_output",
803 "OutputValidator",
804 # Agent Card
805 "AgentCard",
806 "AgentCardRegistry",
807 "AgentCardDiscovery",
808 "discover_local",
809 "create_card",
810 # A2A
811 "A2ATask",
812 "A2AMessage",
813 "A2AArtifact",
814 "A2AHandoff",
815 "A2ASession",
816 "A2AClient",
817 "A2AServer",
818 "TextPart",
819 "FilePart",
820 "DataPart",
821 "TaskState",
822 "new_task",
823 "new_handoff",
824 # Memory
825 "MemoryPyramid",
826 "MemoryLayer",
827 "MemoryType",
828 "MemoryItem",
829 # Evolution
830 "EvolutionEngine",
831 "EvolutionProposal",
832 "EvolutionStatus",
833 # Fusion
834 "FusionToolkit",
835 "FusionResult",
836 "ToolSpec",
837 # Risk
838 "ToolRiskLevel",
839 "ToolRiskRating",
840 "get_risk_preset",
841 "infer_risk_level",
842 # Swarm
843 "SwarmCoordinator",
844 "SmartSwarmCoordinator",
845 "SwarmTopology",
846 "SwarmMessage",
847 "ExecutionMode",
848 "SwarmResult",
849 # Communication
850 "CommunicationLayer",
851 "Blackboard",
852 "EventBus",
853 "Mailbox",
854 # Orchestration
855 "GraphOrchestrator",
856 "GraphNode",
857 "GraphEdge",
858 # Concurrency
859 "AsyncBatchExecutor",
860 "BatchTaskStatus",
861 "TaskSpec",
862 "BatchTaskResult",
863 "BatchConfig",
864 "BatchResult",
865 "BatchStrategy",
866 # Cost
867 "RunCostSession",
868 "CostTracker",
869 "ModelPricing",
870 "UsageRecord",
871 "PRICING",
872 # Resilience
873 "CancellationSource",
874 "CancelledError",
875 "RetryConfig",
876 "CircuitBreaker",
877 "CircuitBreakerConfig",
878 "ResilienceConfig",
879 "ResilientCall",
880 "retry_with_backoff",
881 "with_timeout",
882 "with_fallback",
883 # Router
884 "ModelRouter",
885 # Sandbox
886 "SandboxExecutor",
887 "SandboxMode",
888 "SandboxResult",
889 "ProcessSandbox",
890 "DockerSandbox",
891 # Middleware (v1.2.7)
892 "MiddlewarePhase",
893 "MiddlewareContext",
894 "MiddlewareDecision",
895 "AgentMiddleware",
896 "MiddlewarePipeline",
897 # Queue (v1.2.7)
898 "TaskQueue",
899 "QueueTaskState",
900 "TaskPriority",
901 "RateLimiter",
902 "RateLimitStrategy",
903 "RateLimitConfig",
904 # Cache (v1.2.7)
905 "LLMCache",
906 "CacheEntry",
907 "BaseEmbedder",
908 "OpenAIEmbedder",
909 "LocalEmbedder",
910 "CohereEmbedder",
911 "ResponseCache",
912 "CacheKeyStrategy",
913 # Plugins (v1.2.7)
914 "PluginRegistry",
915 "RegisteredPlugin",
916 "PluginStatus",
917 "PluginDiscovery",
918 "DiscoveredPlugin",
919 "PluginLoader",
920 "LifecycleManager",
921 # Observability (v1.2.7)
922 "MetricsCollector",
923 "Tracer",
924 "NoopTracer",
925 "CostAnalytics",
926 "BudgetAlert",
927 # Workflows (v1.2.7)
928 "WorkflowEngine",
929 "WorkflowTemplate",
930 # MCP Protocol (v1.2.7)
931 "MCPClient",
932 "MCPServerConfig",
933 "MCPToolSchema",
934 # Config (v1.2.7)
935 "AgentOSConfig",
936 "AgentOSPreset",
937 "ValidationResult",
938 # Evaluation (v1.2.7)
939 "Evaluator",
940 "CompositeScorer",
941 "BenchmarkCase",
942 "EvalResult",
943 # Audit (v1.2.7)
944 "SecurityAuditor",
945 "AuditFinding",
946 "AuditReport",
947 # Orchestrator (v1.2.7)
948 "ToolOrchestrator",
949 "DAGBuilder",
950 "DAGSpec",
951 # Memory (v1.2.7)
952 "SemanticMemoryRetriever",
953 "ConversationMemory",
954 # Prompts (v1.2.7)
955 "PromptTemplate",
956 "PromptRegistry",
957 # Multimodal (v1.2.7)
958 "MultimodalManager",
959 "Modality",
960 # Vector Store (v1.2.7)
961 "BaseVectorStore",
962 "FAISSVectorStore",
963 "ChromaVectorStore",
964 # Errors (v1.2.8)
965 "ErrorCategory",
966 "ErrorContext",
967 "ErrorFormatter",
968 "HumanError",
969 # Deployment (v1.2.8)
970 "DockerConfig",
971 "ComposeService",
972 "ComposeConfig",
973 # Monitoring (v1.2.8)
974 "Alert",
975 "AlertEvaluator",
976 "AlertRule",
977 "AlertSeverity",
978 "AlertState",
979 "MonitoringConfig",
980 "WebhookConfig",
981 "WebhookDispatcher",
982 # Experiments (v1.2.8)
983 "ExperimentRunner",
984 "ExperimentConfig",
985 "ExperimentReport",
986 "PromptVariant",
987 "TrialResult",
988 "ExperimentEvaluator",
989 # Feedback (v1.2.8)
990 "FeedbackCollector",
991 "FeedbackRecord",
992 "FeedbackType",
993 "PreferenceLearner",
994 # Memory extensions (v1.2.8)
995 "MemorySummarizer",
996 "ImportanceScorer",
997 "MemoryChunk",
998 "LongTermMemory",
999 "MemoryStore",
1000 "WorkingMemory",
1001 "WorkingMemoryItem",
1002 "VectorMemory",
1003 # Orchestration extensions (v1.2.8)
1004 "A2ARouter",
1005 "RouterAgentCard",
1006 "RouterTask",
1007 "TaskResult",
1008 "TaskStatus",
1009 "AgentGraph",
1010 "GraphRecipe",
1011 "GraphNodeState",
1012 "GraphResult",
1013 # Models - Routing (v1.2.8)
1014 "RoutingStrategy",
1015 "Complexity",
1016 "Budget",
1017 # Swarm Patterns (v1.2.8)
1018 "SwarmPatterns",
1019 "Topology",
1020 "CollaborationConfig",
1021 "CollaborationResult",
1022 # Code Sandbox (v1.9.5)
1023 "CodeSandbox",
1024 "CodeSandboxResult",
1025 "CodeTestCase",
1026 "CodeFeedbackExtractor",
1027 # Human-in-the-Loop (v1.9.5)
1028 "HITLManager",
1029 "HITLConfig",
1030 "Breakpoint",
1031 "BreakpointType",
1032 "HumanDecision",
1033 # Core extensions (v1.2.9)
1034 "AgentContext",
1035 "ContextManager",
1036 "CoreMessage",
1037 "CoreToolCall",
1038 "CoreToolResult",
1039 "AgentStateMachine",
1040 "AgentState",
1041 "StateTransition",
1042 "TransitionError",
1043 "StateTimeoutError",
1044 "StreamChunk",
1045 "StreamEmitter",
1046 "StreamEvent",
1047 "ResponseCollector",
1048 "Session",
1049 "SessionStore",
1050 "AsyncAgentLoop",
1051 "AsyncLoopConfig",
1052 "AsyncInvocationResult",
1053 "AsyncContextManager",
1054 # Logging (v1.2.9)
1055 "JSONFormatter",
1056 "TraceContext",
1057 # Health (v1.2.9)
1058 "HealthChecker",
1059 "HealthStatus",
1060 "HealthCheck",
1061 "CheckResult",
1062 # Security extensions (v1.9.9)
1063 "GuardPipeline",
1064 "InputGuard",
1065 "OutputGuard",
1066 "PIIDetector",
1067 "ContentSafetyFilter",
1068 "GuardChainResult",
1069 "GuardResult",
1070 "GuardAction",
1071 "Severity",
1072 "create_strict_guard",
1073 "create_permissive_guard",
1074 "SandboxManager",
1075 "Sandbox",
1076 "SafetyReport",
1077 "RiskLevel",
1078 "LLMSafetyAnalyzer",
1079 # Storage (v1.2.9)
1080 "CheckpointStore",
1081 "SqliteStore",
1082 # Plugin Manager (v1.2.9)
1083 "PluginManager",
1084 "PluginInfo",
1085 # Cost - Token Counter (v1.2.9)
1086 "TokenCounter",
1087 "TokenCount",
1088 "CostEstimate",
1089 "ModelFamily",
1090 # Protocols - Contracts (v1.2.9)
1091 "AgentContract",
1092 "AgentCapability",
1093 "CapabilityDomain",
1094 "QoSLevel",
1095 "CapabilityMatcher",
1096 "ContractRegistry",
1097 "MatchScore",
1098 # Memory - Compressor (v1.2.9)
1099 "ContextCompressor",
1100 # Tools extensions (v1.2.9)
1101 "BaseTool",
1102 "PermissionLevel",
1103 "BaseToolCall",
1104 "BaseToolResult",
1105 "ToolRegistry",
1106 "ToolSchema",
1107 "FCToolCall",
1108 "FCToolResult",
1109 "FCToolRegistry",
1110 "OpenAPIToolGenerator",
1111 "GeneratedTool",
1112 # SubAgent Manager (v1.2.9) + Parent-Child (v1.3.15)
1113 "SubAgentManager",
1114 "SubAgentMode",
1115 "SubAgentSpec",
1116 "SubAgentResult",
1117 "ChildStatus",
1118 "ChildHeartbeat",
1119 "ChildInfo",
1120 "SharedState",
1121 "ChildContext",
1122 "ChildHandle",
1123 # Agent Marketplace (v1.3.0)
1124 "AgentMarket",
1125 "AgentSkill",
1126 "AgentCategory",
1127 # API Middleware (v1.3.0)
1128 "CORSConfig",
1129 "CORSMiddleware",
1130 "AuthConfig",
1131 "RequestContext",
1132 "RequestIDMiddleware",
1133 # API Streaming (v1.3.0)
1134 "StreamEvent",
1135 "StreamSession",
1136 "StreamingAgent",
1137 # API Versioning (v1.3.0)
1138 "APIVersion",
1139 "VersionStrategy",
1140 "VersionConfig",
1141 "VersionNegotiator",
1142 # Benchmark Runner (v1.3.0)
1143 "BenchmarkRunner",
1144 "BenchmarkScenario",
1145 "BenchmarkConfig",
1146 "BenchmarkReport",
1147 # Testing Fixtures (v1.3.0)
1148 "MockLLMClient",
1149 "MockLLMResponse",
1150 "mock_openai_client",
1151 "mock_model_response",
1152 "sample_config",
1153 # MCP Server (v1.3.0)
1154 "MCPServer",
1155 "MCPServerConfig",
1156 "MCPTool",
1157 "MCPResource",
1158 "MCPPrompt",
1159 # Concrete Tools (v1.3.0)
1160 "CodeAgentTool",
1161 "ShellTool",
1162 "ReadFileTool",
1163 "WriteFileTool",
1164 "ListDirectoryTool",
1165 "WebFetchTool",
1166 # CLI Serve (v1.3.0)
1167 "ServeConfig",
1168 "start_api_server",
1169 # Model Route Types (v1.3.1)
1170 "ModelResponse",
1171 "ModelSpec",
1172 "AllModelsFailed",
1173 # Gemini Backend (v1.3.1)
1174 "GeminiClient",
1175 "GeminiConfig",
1176 "GeminiSafetySetting",
1177 # Agent Loop (v1.3.2)
1178 "AgentLoop",
1179 "LoopConfig",
1180 "LoopState",
1181 "AgentResult",
1182 "MaxIterationsExceeded",
1183 "HumanInterruptNeeded",
1184 # API Server (v1.3.2)
1185 "AgentAPI",
1186 "RunRequest",
1187 "RunResponse",
1188 # CLI Main (v1.0.0)
1189 "cli_main",
1190 # CLI Init (v1.3.2)
1191 "scaffold",
1192 # Docs Generator (v1.3.2)
1193 "DocConfig",
1194 "generate_api_docs",
1195 "generate_quickstart",
1196 # Swarm Coordinator extensions (v1.3.2)
1197 "AgentRole",
1198 "MessageBus",
1199 # Model Config (v1.3.2)
1200 "ModelConfig",
1201 "RECOMMENDED_CONFIG",
1202 # OpenAI Backend (v1.3.5)
1203 "OpenAIClient",
1204 "OpenAIConfig",
1205 # Anthropic Claude Backend (v1.3.5)
1206 "ClaudeClient",
1207 "ClaudeConfig",
1208 # Ollama Backend (v1.3.5)
1209 "OllamaClient",
1210 "OllamaConfig",
1211 # RAG Pipeline (v1.3.5)
1212 "RAGPipeline",
1213 "TextChunker",
1214 "ChunkConfig",
1215 "EmbeddingConfig",
1216 # MCP Package (v1.3.6)
1217 "MCPFullClient",
1218 "MCPConfig",
1219 "MCPToolInfo",
1220 "MCPResourceInfo",
1221 "MCPPromptInfo",
1222 "MCPError",
1223 "connect_mcp_servers",
1224 # MCP Server (v1.5.2)
1225 "MCPServer",
1226 "MCPToolDef",
1227 "MCPResource",
1228 "MCPPromptDef",
1229 "create_default_server",
1230 "start_mcp_server",
1231 "MCPToolAdapter",
1232 "MCPToolRegistry",
1233 # Built-in MCP Servers
1234 "FilesystemServer",
1235 "WebFetchServer",
1236 "MemoryServer",
1237 "BuiltinMCPRegistry",
1238 "create_default_registry",
1239 # Schema Enforcer (v1.3.9)
1240 "SchemaEnforcer",
1241 "EnforcerConfig",
1242 "EnforcerResult",
1243 "EnforcerStats",
1244 "FixStrategy",
1245 # Conversation Manager (v1.3.10)
1246 "ConversationManager",
1247 "ConversationConfig",
1248 "ConversationStats",
1249 "ConversationSnapshot",
1250 "Message",
1251 "MessageRole",
1252 "TrimStrategy",
1253 # Prompt Optimizer (v1.3.12)
1254 "PromptOptimizer",
1255 "OptimizerConfig",
1256 "OptimizationStrategy",
1257 "OptimizationResult",
1258 "PromptCandidate",
1259 # Few-Shot Selector (v1.3.12)
1260 "FewShotSelector",
1261 "Example",
1262 "SelectionStrategy",
1263 "build_examples",
1264 # SSE Streaming (v1.3.12)
1265 "SSEEvent",
1266 "SSEEventType",
1267 "SSEStream",
1268 "SSEResponse",
1269 # A2A Store (v1.3.13)
1270 "A2ATaskStore",
1271 "InMemoryTaskStore",
1272 "SqliteTaskStore",
1273 # A2A Streaming (v1.3.13)
1274 "A2AStreamEvent",
1275 "TaskProgress",
1276 "A2AStreamSession",
1277 "A2AStreamManager",
1278 # LLM Provider Module (v1.3.36)
1279 "LLMProvider",
1280 "OpenAIProvider",
1281 "DeepSeekProvider",
1282 "AnthropicProvider",
1283 "CompletionResult",
1284 "CompletionChoice",
1285 "CompletionUsage",
1286 "TokenUsage",
1287 "LLMMessage",
1288 "LLMMessageRole",
1289 "LLMStreamChunk",
1290 "LLMTool",
1291 "LLMToolCall",
1292 "LLMToolFunction",
1293 "LLMToolParameter",
1294 "create_llm_provider",
1295 # Enterprise (v1.5.5)
1296 "APIKeyManager",
1297 "APIKey",
1298 "KeyScope",
1299 "KeyCreateRequest",
1300 "KeyCreateResult",
1301 "TenantManager",
1302 "Tenant",
1303 "TenantConfig",
1304 "TenantUsage",
1305 "TenantTier",
1306 "TenantStatus",
1307 "TIER_QUOTAS",
1308 "User",
1309 "Role",
1310 "Permission",
1311 "ROLE_PERMISSIONS",
1312 "RBACEngine",
1313 "EnterpriseSession",
1314 "EnterpriseSessionStore",
1315 "JWTManager",
1316 "SSOProvider",
1317 "OIDCConfig",
1318 "SAMLConfig",
1319 "SSOUser",
1320 "AuditLogger",
1321 "AuditEvent",
1322 "AuditCategory",
1323 "AuditSeverity",
1324 "RetentionPolicy",
1325 # System Layer (v1.6.0) — P0: OS-level operations
1326 "SystemPermissionManager",
1327 "SystemPermission",
1328 "PermissionTier",
1329 "PermissionDenied",
1330 "SAFE_PERMISSIONS",
1331 "DEV_PERMISSIONS",
1332 "FULL_PERMISSIONS",
1333 "FileOperator",
1334 "FileOpResult",
1335 "FileListing",
1336 "ShellExecutor",
1337 "ShellResult",
1338 "ShellSandbox",
1339 "ShellPolicy",
1340 "READONLY_POLICY",
1341 "STANDARD_POLICY",
1342 "FULL_POLICY",
1343 "CDPBrowser",
1344 "BrowserSession",
1345 "BrowserAction",
1346 "BrowserResult",
1347 # Desktop Client (v1.7.0) — P1: One-click desktop
1348 "DesktopServer",
1349 "DesktopConfig",
1350 "launch_desktop",
1351 # v1.10.0: Evaluation (SWE-bench + GAIA)
1352 "EvalMetric",
1353 "EvalSuite",
1354 "EvalCase",
1355 "EvalSample",
1356 "EvalResult",
1357 "EvalReport",
1358 "Scorer",
1359 "ExactMatchScorer",
1360 "F1Scorer",
1361 "ROUGELScorer",
1362 "get_scorer",
1363 "SWEBenchLoader",
1364 "GAIALoader",
1365 "EvalRunner",
1366 "EvalRegistry",
1367 "evaluate_quick",
1368 # v1.10.0: Prompt Hub
1369 "PromptType",
1370 "PromptTag",
1371 "PromptVersion",
1372 "PromptHub",
1373 "BUILTIN_PROMPTS",
1374 "create_default_hub",
1375 # v1.11.0: Background Task Manager
1376 "BackgroundTaskManager",
1377 "BackgroundTask",
1378 "BackgroundTaskStatus",
1379 "BackgroundTaskConfig",
1380 "TaskProgress",
1381 "ProgressPhase",
1382 # v1.11.0: Agent Supervision Tree
1383 "AgentSupervisor",
1384 "SupervisedAgent",
1385 "SupervisorConfig",
1386 "AgentQuota",
1387 "SupervisionEvent",
1388 "SupervisionEventType",
1389 # v1.12.0: Virtual Memory Pager
1390 "MemoryPager",
1391 "SwapStore",
1392 "MemoryPage",
1393 "PagerStats",
1394 "create_paging_callback",
1395 "recall_relevant_memories",
1396 # v1.12.1: Async Parallel Primitives
1397 "ParallelExecutor",
1398 "FanOutExecutor",
1399 "FanOutConfig",
1400 "TaskThrottler",
1401 "ParallelTaskResult",
1402 "ParallelTaskStatus",
1403 "ParallelGatherResult",
1404 "parallel_gather",
1405 "parallel_map",
1406 "create_parallel_agent_gather",
1407]
1409# Enterprise (v1.5.5)
1410# SSE Streaming (v1.3.12)
1411from agentos.api.sse import (
1412 SSEEvent,
1413 SSEEventType,
1414 SSEResponse,
1415 SSEStream,
1416)
1418# ── v1.11.0: Agent Supervision Tree ──
1419from agentos.background.supervisor import (
1420 AgentQuota,
1421 AgentSupervisor,
1422 SupervisedAgent,
1423 SupervisionEvent,
1424 SupervisionEventType,
1425 SupervisorConfig,
1426)
1428# ── v1.11.0: Background Task Manager ──
1429from agentos.background.task_manager import (
1430 BackgroundTask,
1431 BackgroundTaskConfig,
1432 BackgroundTaskManager,
1433 BackgroundTaskStatus,
1434 ProgressPhase,
1435 TaskProgress,
1436)
1438# Conversation Manager (v1.3.10)
1439from agentos.conversation.conversation import (
1440 ConversationConfig,
1441 ConversationManager,
1442 ConversationSnapshot,
1443 ConversationStats,
1444 Message,
1445 MessageRole,
1446 TrimStrategy,
1447)
1449# Desktop Client (v1.7.0) — P1: One-click web desktop (AutoClaw-inspired)
1450from agentos.desktop.server import (
1451 DesktopConfig,
1452 DesktopServer,
1453 launch_desktop,
1454)
1455from agentos.enterprise import (
1456 ROLE_PERMISSIONS,
1457 TIER_QUOTAS,
1458 APIKey,
1459 APIKeyManager,
1460 AuditCategory,
1461 AuditEvent,
1462 AuditLogger,
1463 AuditSeverity,
1464 JWTManager,
1465 KeyCreateRequest,
1466 KeyCreateResult,
1467 KeyScope,
1468 OIDCConfig,
1469 Permission,
1470 RBACEngine,
1471 RetentionPolicy,
1472 Role,
1473 SAMLConfig,
1474 SSOProvider,
1475 SSOUser,
1476 Tenant,
1477 TenantConfig,
1478 TenantManager,
1479 TenantStatus,
1480 TenantTier,
1481 TenantUsage,
1482 User,
1483)
1484from agentos.enterprise import (
1485 Session as EnterpriseSession,
1486)
1487from agentos.enterprise import (
1488 SessionStore as EnterpriseSessionStore,
1489)
1491# ── v1.10.0: Evaluation Framework (SWE-bench + GAIA) ──
1492from agentos.eval.benchmark import (
1493 EvalCase,
1494 EvalMetric,
1495 EvalRegistry,
1496 EvalReport,
1497 EvalResult,
1498 EvalRunner,
1499 EvalSample,
1500 EvalSuite,
1501 ExactMatchScorer,
1502 F1Scorer,
1503 GAIALoader,
1504 ROUGELScorer,
1505 Scorer,
1506 SWEBenchLoader,
1507 evaluate_quick,
1508 get_scorer,
1509)
1511# Guardrails (v1.3.11)
1512# HITL (v1.3.11)
1513from agentos.hitl.approver import (
1514 RiskLevel,
1515)
1517# LLM Provider Module (v1.3.36)
1518from agentos.llm import (
1519 AnthropicProvider,
1520 CompletionChoice,
1521 CompletionResult,
1522 CompletionUsage,
1523 DeepSeekProvider,
1524 LLMProvider,
1525 OpenAIProvider,
1526 TokenUsage,
1527)
1528from agentos.llm import (
1529 Message as LLMMessage,
1530)
1531from agentos.llm import (
1532 MessageRole as LLMMessageRole,
1533)
1534from agentos.llm import (
1535 StreamChunk as LLMStreamChunk,
1536)
1537from agentos.llm import (
1538 Tool as LLMTool,
1539)
1540from agentos.llm import (
1541 ToolCall as LLMToolCall,
1542)
1543from agentos.llm import (
1544 ToolFunction as LLMToolFunction,
1545)
1546from agentos.llm import (
1547 ToolParameter as LLMToolParameter,
1548)
1549from agentos.llm import (
1550 create_provider as create_llm_provider,
1551)
1553# ── v1.12.0: Virtual Memory Pager ──
1554from agentos.memory.pager import (
1555 MemoryPage,
1556 MemoryPager,
1557 PagerStats,
1558 SwapStore,
1559 create_paging_callback,
1560 recall_relevant_memories,
1561)
1563# ── v1.10.0: Prompt Hub (versioned templates) ──
1564from agentos.prompt.hub import (
1565 BUILTIN_PROMPTS,
1566 PromptHub,
1567 PromptTag,
1568 PromptType,
1569 PromptVersion,
1570 create_default_hub,
1571)
1573# Few-Shot Selector (v1.3.12)
1574from agentos.prompts.few_shot import (
1575 Example,
1576 FewShotSelector,
1577 SelectionStrategy,
1578 build_examples,
1579)
1581# Prompt Optimizer (v1.3.12)
1582from agentos.prompts.optimizer import (
1583 OptimizationResult,
1584 OptimizationStrategy,
1585 OptimizerConfig,
1586 PromptCandidate,
1587 PromptOptimizer,
1588)
1590# A2A Store (v1.3.13)
1591from agentos.protocols.a2a_store import (
1592 A2ATaskStore,
1593 InMemoryTaskStore,
1594 SqliteTaskStore,
1595)
1597# A2A Streaming (v1.3.13)
1598from agentos.protocols.a2a_streaming import (
1599 A2AStreamEvent,
1600 A2AStreamManager,
1601 A2AStreamSession,
1602 TaskProgress,
1603)
1604from agentos.system.browser import (
1605 BrowserAction,
1606 BrowserResult,
1607 BrowserSession,
1608 CDPBrowser,
1609)
1610from agentos.system.file_ops import (
1611 FileListing,
1612 FileOperator,
1613 FileOpResult,
1614)
1616# System Layer (v1.6.0) — P0: OS-level operations with tiered permissions
1617from agentos.system.permissions import (
1618 DEV_PERMISSIONS,
1619 FULL_PERMISSIONS,
1620 SAFE_PERMISSIONS,
1621 PermissionDenied,
1622 PermissionTier,
1623 SystemPermission,
1624 SystemPermissionManager,
1625)
1626from agentos.system.shell_exec import (
1627 FULL_POLICY,
1628 READONLY_POLICY,
1629 STANDARD_POLICY,
1630 ShellExecutor,
1631 ShellPolicy,
1632 ShellResult,
1633 ShellSandbox,
1634)
1636# Schema Enforcer (v1.3.9)
1637from agentos.validation.schema_enforcer import (
1638 EnforcerConfig,
1639 EnforcerResult,
1640 EnforcerStats,
1641 FixStrategy,
1642 SchemaEnforcer,
1643)