Coverage for agentos/__init__.py: 100%
112 statements
« prev ^ index » next coverage.py v7.14.3, created at 2026-07-03 07:07 +0800
« prev ^ index » next coverage.py v7.14.3, created at 2026-07-03 07:07 +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.15.9"
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
51from agentos.core.di import (
52 Agent,
53 RunContext,
54 Depends,
55 inject_tool,
56 requires_context,
57)
59# Core - Handoff protocol
60from agentos.core.handoff import (
61 Handoff,
62 HandoffResult,
63 transfer_to,
64 can_handle,
65 execute_with_handoff,
66 HandoffAwareAgent,
67)
69# Core - CodeAgent
70from agentos.core.code_agent import (
71 CodeAgent,
72 CodeResult,
73 CodeStep,
74)
76# Protocols - Structured output validation
77from agentos.protocols.output import (
78 StructuredOutput,
79 validate_output,
80 OutputValidator,
81)
83# Protocols - Agent Card
84from agentos.protocols.agent_card import (
85 AgentCard,
86 AgentCardRegistry,
87 AgentCardDiscovery,
88 discover_local,
89 create_card,
90)
92# Protocols - A2A
93from agentos.protocols.a2a import (
94 A2ATask,
95 A2AMessage,
96 A2AArtifact,
97 A2AHandoff,
98 A2ASession,
99 A2AClient,
100 A2AServer,
101 TextPart,
102 FilePart,
103 DataPart,
104 TaskState,
105 new_task,
106 new_handoff,
107)
109# Memory pyramid
110from agentos.memory.pyramid import (
111 MemoryPyramid,
112 MemoryLayer,
113 MemoryType,
114 MemoryItem,
115)
117# Evolution engine
118from agentos.evolution.engine import (
119 EvolutionEngine,
120 EvolutionProposal,
121 EvolutionStatus,
122)
124# Fusion toolkit
125from agentos.tools.fusion import (
126 FusionToolkit,
127 FusionResult,
128 ToolSpec,
129)
131# Tool risk rating (v1.1.4)
132from agentos.tools.risk import (
133 ToolRiskLevel,
134 ToolRiskRating,
135 get_risk_preset,
136 infer_risk_level,
137)
139# Swarm coordinator
140from agentos.swarm.coordinator import (
141 SwarmCoordinator,
142 SmartSwarmCoordinator,
143 SwarmTopology,
144 SwarmMessage,
145 ExecutionMode,
146 SwarmResult,
147)
149# Communication layer
150from agentos.comm.layer import (
151 CommunicationLayer,
152 Blackboard,
153 EventBus,
154 Mailbox,
155)
157# Orchestration
158from agentos.orchestration.graph import (
159 GraphOrchestrator,
160 GraphNode,
161 GraphEdge,
162)
164# Concurrency (v1.1.3)
165from agentos.concurrency.batch import (
166 AsyncBatchExecutor,
167 TaskStatus as BatchTaskStatus,
168 TaskSpec,
169 TaskResult as BatchTaskResult,
170 BatchConfig,
171 BatchResult,
172 BatchStrategy,
173)
175# Cost tracking (v1.1.4)
176from agentos.cost.tracker import (
177 RunCostSession,
178 CostTracker,
179 ModelPricing,
180 UsageRecord,
181 PRICING,
182)
184# Models - Resilience (v1.1.5)
185from agentos.models.resilience import (
186 CancellationSource,
187 CancelledError,
188 RetryConfig,
189 CircuitBreaker,
190 CircuitBreakerConfig,
191 ResilienceConfig,
192 ResilientCall,
193 retry_with_backoff,
194 with_timeout,
195 with_fallback,
196)
198# Models - Router (v1.2.7 minimal)
199from agentos.models.router import ModelRouter
201# Security - Sandbox (v1.2.1)
202from agentos.security.sandbox_executor import (
203 SandboxExecutor,
204 SandboxMode,
205 SandboxResult,
206 ProcessSandbox,
207 DockerSandbox,
208)
210# Core - Middleware Pipeline (v1.2.7)
211from agentos.core.middleware import (
212 MiddlewarePhase,
213 MiddlewareContext,
214 MiddlewareDecision,
215 AgentMiddleware,
216 MiddlewarePipeline,
217)
219# Queue - Task Queue & Rate Limiter (v1.2.7)
220from agentos.queue import (
221 TaskQueue,
222 TaskState as QueueTaskState,
223 TaskPriority,
224 RateLimiter,
225 RateLimitStrategy,
226 RateLimitConfig,
227)
229# Cache - LLM Response Cache (v1.2.7)
230from agentos.cache import (
231 LLMCache,
232 CacheEntry,
233 BaseEmbedder,
234 OpenAIEmbedder,
235 LocalEmbedder,
236 CohereEmbedder,
237 ResponseCache,
238 CacheKeyStrategy,
239)
241# Plugins - Plugin System (v1.2.7)
242from agentos.plugins import (
243 PluginRegistry,
244 RegisteredPlugin,
245 PluginStatus,
246 PluginDiscovery,
247 DiscoveredPlugin,
248 PluginLoader,
249 LifecycleManager,
250)
252# Observability (v1.2.7)
253from agentos.observability import (
254 MetricsCollector,
255 Tracer,
256 NoopTracer,
257 CostAnalytics,
258 BudgetAlert,
259)
261# Workflows (v1.2.7)
262from agentos.workflows import (
263 WorkflowEngine,
264 WorkflowTemplate,
265)
267# MCP Protocol (v1.2.7)
268from agentos.protocols.mcp import (
269 MCPClient,
270 MCPServerConfig,
271 MCPToolSchema,
272)
274# Config System (v1.2.7)
275from agentos.config import (
276 AgentOSConfig,
277 AgentOSPreset,
278 ValidationResult,
279)
281# Evaluation Framework (v1.3.18)
282from agentos.evaluation import (
283 GoldenDataset,
284 GoldenCase,
285 Evaluator,
286 EvalConfig,
287 EvalReport,
288 ScoreDetail,
289 Scorer,
290 load_dataset,
291 save_dataset,
292 quick_eval,
293)
294from agentos.evaluation.regression import (
295 RegressionRunner,
296 RegressionReport,
297 RegressionCheck,
298 StatisticalRunner,
299 StatResult,
300 to_junit_xml,
301 to_json,
302 save_report,
303)
304from agentos.evaluation.scorers import (
305 CompositeScorer,
306 ScoringStrategy,
307 ScoreResult,
308 rouge_l,
309 bleu,
310 semantic_similarity,
311 exact_match,
312 contains_match,
313 STRATEGY_CODE_GEN,
314 STRATEGY_QA,
315 STRATEGY_SUMMARY,
316 STRATEGY_TRANSLATION,
317)
319# Security - Auditor (v1.2.7)
320from agentos.security.auditor import (
321 SecurityAuditor,
322 AuditFinding,
323 AuditReport,
324)
326# Tools - Orchestrator (v1.2.7)
327from agentos.tools.orchestrator import (
328 ToolOrchestrator,
329 DAGBuilder,
330 DAGSpec,
331)
333# Memory - Retriever + Conversation (v1.2.7)
334from agentos.memory import (
335 SemanticMemoryRetriever,
336 ConversationMemory,
337)
339# Prompts (v1.2.7)
340from agentos.prompts import (
341 PromptTemplate,
342 PromptRegistry,
343)
345# Multimodal (v1.2.7)
346from agentos.multimodal import (
347 MultimodalManager,
348 Modality,
349)
351# Vector Store (v1.2.7)
352from agentos.vectorstore import (
353 BaseVectorStore,
354 FAISSVectorStore,
355 ChromaVectorStore,
356)
358# Errors (v1.2.8)
359from agentos.errors import (
360 ErrorCategory,
361 ErrorContext,
362 ErrorFormatter,
363 HumanError,
364)
366# Deployment (v1.2.8)
367from agentos.deployment import (
368 DockerConfig,
369 ComposeService,
370 ComposeConfig,
371)
373# Monitoring (v1.2.8)
374from agentos.monitoring import (
375 Alert,
376 AlertEvaluator,
377 AlertRule,
378 AlertSeverity,
379 AlertState,
380 MonitoringConfig,
381 WebhookConfig,
382 WebhookDispatcher,
383)
385# Experiments (v1.2.8)
386from agentos.experiments import (
387 ExperimentRunner,
388 ExperimentConfig,
389 ExperimentReport,
390 PromptVariant,
391 TrialResult,
392 Evaluator as ExperimentEvaluator,
393)
395# Feedback (v1.2.8)
396from agentos.feedback import (
397 FeedbackCollector,
398 FeedbackRecord,
399 FeedbackType,
400 PreferenceLearner,
401)
403# Memory extensions (v1.2.8)
404from agentos.memory import (
405 MemorySummarizer,
406 ImportanceScorer,
407 MemoryChunk,
408 LongTermMemory,
409 MemoryStore,
410 WorkingMemory,
411 WorkingMemoryItem,
412 VectorMemory,
413)
415# Orchestration extensions (v1.2.8)
416from agentos.orchestration import (
417 A2ARouter,
418 RouterAgentCard,
419 RouterTask,
420 TaskResult,
421 TaskStatus,
422 AgentGraph,
423 GraphRecipe,
424 GraphNodeState,
425 GraphResult,
426)
428# Models - Routing Strategy (v1.2.8)
429from agentos.models.routing_strategy import (
430 RoutingStrategy,
431 Complexity,
432 Budget,
433)
435# Swarm Patterns (v1.2.8)
436from agentos.swarm import (
437 SwarmPatterns,
438 Topology,
439 CollaborationConfig,
440 CollaborationResult,
441)
443# Code Sandbox (v1.9.5)
444from agentos.swarm.code_sandbox import (
445 CodeSandbox,
446 SandboxResult as CodeSandboxResult,
447 TestCase as CodeTestCase,
448 CodeFeedbackExtractor,
449)
451# Human-in-the-Loop (v1.9.5)
452from agentos.swarm.human_loop import (
453 HITLManager,
454 HITLConfig,
455 Breakpoint,
456 BreakpointType,
457 HumanDecision,
458)
460# Core extensions (v1.2.9)
461from agentos.core import (
462 AgentContext,
463 ContextManager,
464 CoreMessage,
465 CoreToolCall,
466 CoreToolResult,
467 AgentStateMachine,
468 AgentState,
469 StateTransition,
470 TransitionError,
471 StateTimeoutError,
472 StreamChunk,
473 StreamEmitter,
474 StreamEvent,
475 ResponseCollector,
476 Session,
477 SessionStore,
478 AsyncAgentLoop,
479 AsyncLoopConfig,
480 AsyncInvocationResult,
481 AsyncContextManager,
482)
484# Logging (v1.2.9)
485from agentos.log import (
486 JSONFormatter,
487 TraceContext,
488)
490# Health (v1.2.9)
491from agentos.health import (
492 HealthChecker,
493 HealthStatus,
494 HealthCheck,
495 CheckResult,
496)
498# Security extensions (v1.9.9)
499from agentos.security import (
500 GuardPipeline,
501 InputGuard,
502 OutputGuard,
503 PIIDetector,
504 ContentSafetyFilter,
505 GuardChainResult,
506 GuardResult,
507 GuardAction,
508 Severity,
509 create_strict_guard,
510 create_permissive_guard,
511 SandboxManager,
512 Sandbox,
513 SafetyReport,
514 RiskLevel,
515 LLMSafetyAnalyzer,
516)
518# Storage (v1.2.9)
519from agentos.storage import (
520 CheckpointStore,
521 SqliteStore,
522)
524# Plugin Manager (v1.2.9)
525from agentos.plugin_manager import (
526 PluginManager,
527 PluginInfo,
528)
530# Cost - Token Counter (v1.2.9)
531from agentos.cost import (
532 TokenCounter,
533 TokenCount,
534 CostEstimate,
535 ModelFamily,
536)
538# Protocols - Contracts (v1.2.9)
539from agentos.protocols import (
540 AgentContract,
541 AgentCapability,
542 CapabilityDomain,
543 QoSLevel,
544 CapabilityMatcher,
545 ContractRegistry,
546 MatchScore,
547)
549# Memory - Compressor (v1.2.9)
550from agentos.memory import (
551 ContextCompressor,
552)
554# Tools extensions (v1.2.9)
555from agentos.tools import (
556 BaseTool,
557 PermissionLevel,
558 BaseToolCall,
559 BaseToolResult,
560 ToolRegistry,
561 ToolSchema,
562 FCToolCall,
563 FCToolResult,
564 FCToolRegistry,
565 OpenAPIToolGenerator,
566 GeneratedTool,
567)
569# SubAgent Manager (v1.2.9) + Parent-Child Communication (v1.3.15)
570from agentos.subagent import (
571 SubAgentManager,
572 SubAgentMode,
573 SubAgentSpec,
574 SubAgentResult,
575 ChildStatus,
576 ChildHeartbeat,
577 ChildInfo,
578 SharedState,
579 ChildContext,
580 ChildHandle,
581)
583# Agent Marketplace (v1.3.0)
584from agentos.agents.market import (
585 AgentMarket,
586 AgentSkill,
587 AgentCategory,
588)
590# Tool-Using Agent (v1.3.38)
591from agentos.agent import (
592 ToolAgent,
593 ToolExecutor,
594 AgentConfig,
595 AgentStep,
596 AgentResult,
597 MockLLMProvider,
598)
600# API Middleware (v1.3.0)
601from agentos.api.middleware import (
602 CORSConfig,
603 CORSMiddleware,
604 AuthConfig,
605 RequestContext,
606 RequestIDMiddleware,
607)
609# API Streaming (v1.3.0)
610from agentos.api.streaming import (
611 StreamEvent,
612 StreamSession,
613 StreamingAgent,
614)
616# API Versioning (v1.3.0)
617from agentos.api.versioning import (
618 APIVersion,
619 VersionStrategy,
620 VersionConfig,
621 VersionNegotiator,
622)
624# Benchmark Runner (v1.3.0)
625from agentos.benchmarks.runner import (
626 BenchmarkRunner,
627 BenchmarkScenario,
628 BenchmarkConfig,
629 BenchmarkReport,
630)
632# Testing Fixtures (v1.3.0)
633from agentos.testing.fixtures import (
634 MockLLMClient,
635 MockLLMResponse,
636 mock_openai_client,
637 mock_model_response,
638 sample_config,
639)
641# MCP Server (v1.3.0)
642from agentos.server.mcp_server import (
643 MCPServer,
644 MCPServerConfig,
645 MCPTool,
646 MCPResource,
647 MCPPrompt,
648)
650# Concrete Tools (v1.3.0)
651from agentos.tools.code_agent import (
652 CodeAgentTool,
653 ShellTool,
654)
655from agentos.tools.file_tools import (
656 ReadFileTool,
657 WriteFileTool,
658 ListDirectoryTool,
659)
660from agentos.tools.web_tools import (
661 WebFetchTool,
662)
664# CLI Serve (v1.3.0)
665from agentos.cli.serve import (
666 ServeConfig,
667 start_api_server,
668)
670# Model Route Types (v1.3.1)
671from agentos.models.router import (
672 ModelResponse,
673 ModelSpec,
674 AllModelsFailed,
675)
677# Gemini Backend (v1.3.1)
678from agentos.models.backends.gemini import (
679 GeminiClient,
680 GeminiConfig,
681 GeminiSafetySetting,
682)
684# Agent Loop (v1.3.2)
685from agentos.core.loop import (
686 AgentLoop,
687 LoopConfig,
688 LoopState,
689 AgentResult,
690 MaxIterationsExceeded,
691 HumanInterruptNeeded,
692)
694# API Server (v1.3.2)
695from agentos.api.server import (
696 AgentAPI,
697 RunRequest,
698 RunResponse,
699)
701# CLI Main (v1.0.0)
702from agentos.cli.main import main as cli_main
704# CLI Init (v1.4.1)
705from agentos.cli.init import (
706 init_cli,
707 scaffold,
708 load_config,
709 config_status_text,
710)
712# Docs Generator (v1.3.2)
713from agentos.docs.generator import (
714 DocConfig,
715 generate_api_docs,
716 generate_quickstart,
717)
719# Swarm Coordinator extensions (v1.3.2)
720from agentos.swarm.coordinator import (
721 AgentRole,
722 MessageBus,
723)
725# Model Config (v1.3.2)
726from agentos.models.router import (
727 ModelConfig,
728 RECOMMENDED_CONFIG,
729)
731# OpenAI Backend (v1.3.5)
732from agentos.models.backends.openai import (
733 OpenAIClient,
734 OpenAIConfig,
735)
737# Anthropic Claude Backend (v1.3.5)
738from agentos.models.backends.anthropic import (
739 ClaudeClient,
740 ClaudeConfig,
741)
743# Ollama Backend (v1.3.5)
744from agentos.models.backends.ollama import (
745 OllamaClient,
746 OllamaConfig,
747)
749# RAG Pipeline (v1.3.5)
750from agentos.rag import (
751 RAGPipeline,
752 TextChunker,
753 ChunkConfig,
754 EmbeddingConfig,
755)
757# MCP Package (v1.3.6)
758from agentos.mcp import (
759 MCPClient as MCPFullClient,
760 MCPServerConfig as MCPConfig,
761 MCPToolInfo,
762 MCPResourceInfo,
763 MCPPromptInfo,
764 MCPError,
765 connect_mcp_servers,
766 # MCP Server (v1.5.2)
767 MCPServer,
768 MCPToolDef,
769 MCPResource,
770 MCPPromptDef,
771 create_default_server,
772 start_mcp_server,
773)
774from agentos.mcp.adapter import (
775 MCPToolAdapter,
776 MCPToolRegistry,
777)
778# Built-in MCP Servers (v1.7.8)
779from agentos.mcp.builtin_servers import (
780 FilesystemServer,
781 WebFetchServer,
782 MemoryServer,
783 BuiltinMCPRegistry,
784 create_default_registry,
785)
787__all__ = [
788 # Version
789 "__version__",
790 # Core DI
791 "Agent",
792 "RunContext",
793 "Depends",
794 "inject_tool",
795 "requires_context",
796 # Handoff
797 "Handoff",
798 "HandoffResult",
799 "transfer_to",
800 "can_handle",
801 "execute_with_handoff",
802 "HandoffAwareAgent",
803 # CodeAgent
804 "CodeAgent",
805 "CodeResult",
806 "CodeStep",
807 # Tool-Using Agent (v1.3.38)
808 "ToolAgent",
809 "ToolExecutor",
810 "AgentConfig",
811 "AgentStep",
812 "AgentResult",
813 "MockLLMProvider",
814 # Structured output
815 "StructuredOutput",
816 "validate_output",
817 "OutputValidator",
818 # Agent Card
819 "AgentCard",
820 "AgentCardRegistry",
821 "AgentCardDiscovery",
822 "discover_local",
823 "create_card",
824 # A2A
825 "A2ATask",
826 "A2AMessage",
827 "A2AArtifact",
828 "A2AHandoff",
829 "A2ASession",
830 "A2AClient",
831 "A2AServer",
832 "TextPart",
833 "FilePart",
834 "DataPart",
835 "TaskState",
836 "new_task",
837 "new_handoff",
838 # Memory
839 "MemoryPyramid",
840 "MemoryLayer",
841 "MemoryType",
842 "MemoryItem",
843 # Evolution
844 "EvolutionEngine",
845 "EvolutionProposal",
846 "EvolutionStatus",
847 # Fusion
848 "FusionToolkit",
849 "FusionResult",
850 "ToolSpec",
851 # Risk
852 "ToolRiskLevel",
853 "ToolRiskRating",
854 "get_risk_preset",
855 "infer_risk_level",
856 # Swarm
857 "SwarmCoordinator",
858 "SmartSwarmCoordinator",
859 "SwarmTopology",
860 "SwarmMessage",
861 "ExecutionMode",
862 "SwarmResult",
863 # Communication
864 "CommunicationLayer",
865 "Blackboard",
866 "EventBus",
867 "Mailbox",
868 # Orchestration
869 "GraphOrchestrator",
870 "GraphNode",
871 "GraphEdge",
872 # Concurrency
873 "AsyncBatchExecutor",
874 "BatchTaskStatus",
875 "TaskSpec",
876 "BatchTaskResult",
877 "BatchConfig",
878 "BatchResult",
879 "BatchStrategy",
880 # Cost
881 "RunCostSession",
882 "CostTracker",
883 "ModelPricing",
884 "UsageRecord",
885 "PRICING",
886 # Resilience
887 "CancellationSource",
888 "CancelledError",
889 "RetryConfig",
890 "CircuitBreaker",
891 "CircuitBreakerConfig",
892 "ResilienceConfig",
893 "ResilientCall",
894 "retry_with_backoff",
895 "with_timeout",
896 "with_fallback",
897 # Router
898 "ModelRouter",
899 # Sandbox
900 "SandboxExecutor",
901 "SandboxMode",
902 "SandboxResult",
903 "ProcessSandbox",
904 "DockerSandbox",
905 # Middleware (v1.2.7)
906 "MiddlewarePhase",
907 "MiddlewareContext",
908 "MiddlewareDecision",
909 "AgentMiddleware",
910 "MiddlewarePipeline",
911 # Queue (v1.2.7)
912 "TaskQueue",
913 "QueueTaskState",
914 "TaskPriority",
915 "RateLimiter",
916 "RateLimitStrategy",
917 "RateLimitConfig",
918 # Cache (v1.2.7)
919 "LLMCache",
920 "CacheEntry",
921 "BaseEmbedder",
922 "OpenAIEmbedder",
923 "LocalEmbedder",
924 "CohereEmbedder",
925 "ResponseCache",
926 "CacheKeyStrategy",
927 # Plugins (v1.2.7)
928 "PluginRegistry",
929 "RegisteredPlugin",
930 "PluginStatus",
931 "PluginDiscovery",
932 "DiscoveredPlugin",
933 "PluginLoader",
934 "LifecycleManager",
935 # Observability (v1.2.7)
936 "MetricsCollector",
937 "Tracer",
938 "NoopTracer",
939 "CostAnalytics",
940 "BudgetAlert",
941 # Workflows (v1.2.7)
942 "WorkflowEngine",
943 "WorkflowTemplate",
944 # MCP Protocol (v1.2.7)
945 "MCPClient",
946 "MCPServerConfig",
947 "MCPToolSchema",
948 # Config (v1.2.7)
949 "AgentOSConfig",
950 "AgentOSPreset",
951 "ValidationResult",
952 # Evaluation (v1.2.7)
953 "Evaluator",
954 "CompositeScorer",
955 "BenchmarkCase",
956 "EvalResult",
957 # Audit (v1.2.7)
958 "SecurityAuditor",
959 "AuditFinding",
960 "AuditReport",
961 # Orchestrator (v1.2.7)
962 "ToolOrchestrator",
963 "DAGBuilder",
964 "DAGSpec",
965 # Memory (v1.2.7)
966 "SemanticMemoryRetriever",
967 "ConversationMemory",
968 # Prompts (v1.2.7)
969 "PromptTemplate",
970 "PromptRegistry",
971 # Multimodal (v1.2.7)
972 "MultimodalManager",
973 "Modality",
974 # Vector Store (v1.2.7)
975 "BaseVectorStore",
976 "FAISSVectorStore",
977 "ChromaVectorStore",
978 # Errors (v1.2.8)
979 "ErrorCategory",
980 "ErrorContext",
981 "ErrorFormatter",
982 "HumanError",
983 # Deployment (v1.2.8)
984 "DockerConfig",
985 "ComposeService",
986 "ComposeConfig",
987 # Monitoring (v1.2.8)
988 "Alert",
989 "AlertEvaluator",
990 "AlertRule",
991 "AlertSeverity",
992 "AlertState",
993 "MonitoringConfig",
994 "WebhookConfig",
995 "WebhookDispatcher",
996 # Experiments (v1.2.8)
997 "ExperimentRunner",
998 "ExperimentConfig",
999 "ExperimentReport",
1000 "PromptVariant",
1001 "TrialResult",
1002 "ExperimentEvaluator",
1003 # Feedback (v1.2.8)
1004 "FeedbackCollector",
1005 "FeedbackRecord",
1006 "FeedbackType",
1007 "PreferenceLearner",
1008 # Memory extensions (v1.2.8)
1009 "MemorySummarizer",
1010 "ImportanceScorer",
1011 "MemoryChunk",
1012 "LongTermMemory",
1013 "MemoryStore",
1014 "WorkingMemory",
1015 "WorkingMemoryItem",
1016 "VectorMemory",
1017 # Orchestration extensions (v1.2.8)
1018 "A2ARouter",
1019 "RouterAgentCard",
1020 "RouterTask",
1021 "TaskResult",
1022 "TaskStatus",
1023 "AgentGraph",
1024 "GraphRecipe",
1025 "GraphNodeState",
1026 "GraphResult",
1027 # Models - Routing (v1.2.8)
1028 "RoutingStrategy",
1029 "Complexity",
1030 "Budget",
1031 # Swarm Patterns (v1.2.8)
1032 "SwarmPatterns",
1033 "Topology",
1034 "CollaborationConfig",
1035 "CollaborationResult",
1036 # Code Sandbox (v1.9.5)
1037 "CodeSandbox",
1038 "CodeSandboxResult",
1039 "CodeTestCase",
1040 "CodeFeedbackExtractor",
1041 # Human-in-the-Loop (v1.9.5)
1042 "HITLManager",
1043 "HITLConfig",
1044 "Breakpoint",
1045 "BreakpointType",
1046 "HumanDecision",
1047 # Core extensions (v1.2.9)
1048 "AgentContext",
1049 "ContextManager",
1050 "CoreMessage",
1051 "CoreToolCall",
1052 "CoreToolResult",
1053 "AgentStateMachine",
1054 "AgentState",
1055 "StateTransition",
1056 "TransitionError",
1057 "StateTimeoutError",
1058 "StreamChunk",
1059 "StreamEmitter",
1060 "StreamEvent",
1061 "ResponseCollector",
1062 "Session",
1063 "SessionStore",
1064 "AsyncAgentLoop",
1065 "AsyncLoopConfig",
1066 "AsyncInvocationResult",
1067 "AsyncContextManager",
1068 # Logging (v1.2.9)
1069 "JSONFormatter",
1070 "TraceContext",
1071 # Health (v1.2.9)
1072 "HealthChecker",
1073 "HealthStatus",
1074 "HealthCheck",
1075 "CheckResult",
1076 # Security extensions (v1.9.9)
1077 "GuardPipeline",
1078 "InputGuard",
1079 "OutputGuard",
1080 "PIIDetector",
1081 "ContentSafetyFilter",
1082 "GuardChainResult",
1083 "GuardResult",
1084 "GuardAction",
1085 "Severity",
1086 "create_strict_guard",
1087 "create_permissive_guard",
1088 "SandboxManager",
1089 "Sandbox",
1090 "SafetyReport",
1091 "RiskLevel",
1092 "LLMSafetyAnalyzer",
1093 # Storage (v1.2.9)
1094 "CheckpointStore",
1095 "SqliteStore",
1096 # Plugin Manager (v1.2.9)
1097 "PluginManager",
1098 "PluginInfo",
1099 # Cost - Token Counter (v1.2.9)
1100 "TokenCounter",
1101 "TokenCount",
1102 "CostEstimate",
1103 "ModelFamily",
1104 # Protocols - Contracts (v1.2.9)
1105 "AgentContract",
1106 "AgentCapability",
1107 "CapabilityDomain",
1108 "QoSLevel",
1109 "CapabilityMatcher",
1110 "ContractRegistry",
1111 "MatchScore",
1112 # Memory - Compressor (v1.2.9)
1113 "ContextCompressor",
1114 # Tools extensions (v1.2.9)
1115 "BaseTool",
1116 "PermissionLevel",
1117 "BaseToolCall",
1118 "BaseToolResult",
1119 "ToolRegistry",
1120 "ToolSchema",
1121 "FCToolCall",
1122 "FCToolResult",
1123 "FCToolRegistry",
1124 "OpenAPIToolGenerator",
1125 "GeneratedTool",
1126 # SubAgent Manager (v1.2.9) + Parent-Child (v1.3.15)
1127 "SubAgentManager",
1128 "SubAgentMode",
1129 "SubAgentSpec",
1130 "SubAgentResult",
1131 "ChildStatus",
1132 "ChildHeartbeat",
1133 "ChildInfo",
1134 "SharedState",
1135 "ChildContext",
1136 "ChildHandle",
1137 # Agent Marketplace (v1.3.0)
1138 "AgentMarket",
1139 "AgentSkill",
1140 "AgentCategory",
1141 # API Middleware (v1.3.0)
1142 "CORSConfig",
1143 "CORSMiddleware",
1144 "AuthConfig",
1145 "RequestContext",
1146 "RequestIDMiddleware",
1147 # API Streaming (v1.3.0)
1148 "StreamEvent",
1149 "StreamSession",
1150 "StreamingAgent",
1151 # API Versioning (v1.3.0)
1152 "APIVersion",
1153 "VersionStrategy",
1154 "VersionConfig",
1155 "VersionNegotiator",
1156 # Benchmark Runner (v1.3.0)
1157 "BenchmarkRunner",
1158 "BenchmarkScenario",
1159 "BenchmarkConfig",
1160 "BenchmarkReport",
1161 # Testing Fixtures (v1.3.0)
1162 "MockLLMClient",
1163 "MockLLMResponse",
1164 "mock_openai_client",
1165 "mock_model_response",
1166 "sample_config",
1167 # MCP Server (v1.3.0)
1168 "MCPServer",
1169 "MCPServerConfig",
1170 "MCPTool",
1171 "MCPResource",
1172 "MCPPrompt",
1173 # Concrete Tools (v1.3.0)
1174 "CodeAgentTool",
1175 "ShellTool",
1176 "ReadFileTool",
1177 "WriteFileTool",
1178 "ListDirectoryTool",
1179 "WebFetchTool",
1180 # CLI Serve (v1.3.0)
1181 "ServeConfig",
1182 "start_api_server",
1183 # Model Route Types (v1.3.1)
1184 "ModelResponse",
1185 "ModelSpec",
1186 "AllModelsFailed",
1187 # Gemini Backend (v1.3.1)
1188 "GeminiClient",
1189 "GeminiConfig",
1190 "GeminiSafetySetting",
1191 # Agent Loop (v1.3.2)
1192 "AgentLoop",
1193 "LoopConfig",
1194 "LoopState",
1195 "AgentResult",
1196 "MaxIterationsExceeded",
1197 "HumanInterruptNeeded",
1198 # API Server (v1.3.2)
1199 "AgentAPI",
1200 "RunRequest",
1201 "RunResponse",
1202 # CLI Main (v1.0.0)
1203 "cli_main",
1204 # CLI Init (v1.3.2)
1205 "scaffold",
1206 # Docs Generator (v1.3.2)
1207 "DocConfig",
1208 "generate_api_docs",
1209 "generate_quickstart",
1210 # Swarm Coordinator extensions (v1.3.2)
1211 "AgentRole",
1212 "MessageBus",
1213 # Model Config (v1.3.2)
1214 "ModelConfig",
1215 "RECOMMENDED_CONFIG",
1216 # OpenAI Backend (v1.3.5)
1217 "OpenAIClient",
1218 "OpenAIConfig",
1219 # Anthropic Claude Backend (v1.3.5)
1220 "ClaudeClient",
1221 "ClaudeConfig",
1222 # Ollama Backend (v1.3.5)
1223 "OllamaClient",
1224 "OllamaConfig",
1225 # RAG Pipeline (v1.3.5)
1226 "RAGPipeline",
1227 "TextChunker",
1228 "ChunkConfig",
1229 "EmbeddingConfig",
1230 # MCP Package (v1.3.6)
1231 "MCPFullClient",
1232 "MCPConfig",
1233 "MCPToolInfo",
1234 "MCPResourceInfo",
1235 "MCPPromptInfo",
1236 "MCPError",
1237 "connect_mcp_servers",
1238 # MCP Server (v1.5.2)
1239 "MCPServer",
1240 "MCPToolDef",
1241 "MCPResource",
1242 "MCPPromptDef",
1243 "create_default_server",
1244 "start_mcp_server",
1245 "MCPToolAdapter",
1246 "MCPToolRegistry",
1247 # Built-in MCP Servers
1248 "FilesystemServer",
1249 "WebFetchServer",
1250 "MemoryServer",
1251 "BuiltinMCPRegistry",
1252 "create_default_registry",
1253 # Schema Enforcer (v1.3.9)
1254 "SchemaEnforcer",
1255 "EnforcerConfig",
1256 "EnforcerResult",
1257 "EnforcerStats",
1258 "FixStrategy",
1259 # Conversation Manager (v1.3.10)
1260 "ConversationManager",
1261 "ConversationConfig",
1262 "ConversationStats",
1263 "ConversationSnapshot",
1264 "Message",
1265 "MessageRole",
1266 "TrimStrategy",
1267 # Prompt Optimizer (v1.3.12)
1268 "PromptOptimizer",
1269 "OptimizerConfig",
1270 "OptimizationStrategy",
1271 "OptimizationResult",
1272 "PromptCandidate",
1273 # Few-Shot Selector (v1.3.12)
1274 "FewShotSelector",
1275 "Example",
1276 "SelectionStrategy",
1277 "build_examples",
1278 # SSE Streaming (v1.3.12)
1279 "SSEEvent",
1280 "SSEEventType",
1281 "SSEStream",
1282 "SSEResponse",
1283 # A2A Store (v1.3.13)
1284 "A2ATaskStore",
1285 "InMemoryTaskStore",
1286 "SqliteTaskStore",
1287 # A2A Streaming (v1.3.13)
1288 "A2AStreamEvent",
1289 "TaskProgress",
1290 "A2AStreamSession",
1291 "A2AStreamManager",
1292 # LLM Provider Module (v1.3.36)
1293 "LLMProvider",
1294 "OpenAIProvider",
1295 "DeepSeekProvider",
1296 "AnthropicProvider",
1297 "CompletionResult",
1298 "CompletionChoice",
1299 "CompletionUsage",
1300 "TokenUsage",
1301 "LLMMessage",
1302 "LLMMessageRole",
1303 "LLMStreamChunk",
1304 "LLMTool",
1305 "LLMToolCall",
1306 "LLMToolFunction",
1307 "LLMToolParameter",
1308 "create_llm_provider",
1309 # Enterprise (v1.5.5)
1310 "APIKeyManager", "APIKey", "KeyScope", "KeyCreateRequest", "KeyCreateResult",
1311 "TenantManager", "Tenant", "TenantConfig", "TenantUsage", "TenantTier", "TenantStatus", "TIER_QUOTAS",
1312 "User", "Role", "Permission", "ROLE_PERMISSIONS", "RBACEngine",
1313 "EnterpriseSession", "EnterpriseSessionStore",
1314 "JWTManager", "SSOProvider", "OIDCConfig", "SAMLConfig", "SSOUser",
1315 "AuditLogger", "AuditEvent", "AuditCategory", "AuditSeverity", "RetentionPolicy",
1316 # System Layer (v1.6.0) — P0: OS-level operations
1317 "SystemPermissionManager", "SystemPermission", "PermissionTier", "PermissionDenied",
1318 "SAFE_PERMISSIONS", "DEV_PERMISSIONS", "FULL_PERMISSIONS",
1319 "FileOperator", "FileOpResult", "FileListing",
1320 "ShellExecutor", "ShellResult", "ShellSandbox", "ShellPolicy",
1321 "READONLY_POLICY", "STANDARD_POLICY", "FULL_POLICY",
1322 "CDPBrowser", "BrowserSession", "BrowserAction", "BrowserResult",
1323 # Desktop Client (v1.7.0) — P1: One-click desktop
1324 "DesktopServer", "DesktopConfig", "launch_desktop",
1325 # v1.10.0: Evaluation (SWE-bench + GAIA)
1326 "EvalMetric", "EvalSuite", "EvalCase", "EvalSample", "EvalResult", "EvalReport",
1327 "Scorer", "ExactMatchScorer", "F1Scorer", "ROUGELScorer", "get_scorer",
1328 "SWEBenchLoader", "GAIALoader",
1329 "EvalRunner", "EvalRegistry", "evaluate_quick",
1330 # v1.10.0: Prompt Hub
1331 "PromptType", "PromptTag", "PromptVersion", "PromptHub",
1332 "BUILTIN_PROMPTS", "create_default_hub",
1333 # v1.11.0: Background Task Manager
1334 "BackgroundTaskManager", "BackgroundTask", "BackgroundTaskStatus", "BackgroundTaskConfig",
1335 "TaskProgress", "ProgressPhase",
1336 # v1.11.0: Agent Supervision Tree
1337 "AgentSupervisor", "SupervisedAgent", "SupervisorConfig",
1338 "AgentQuota", "SupervisionEvent", "SupervisionEventType",
1339 # v1.12.0: Virtual Memory Pager
1340 "MemoryPager", "SwapStore", "MemoryPage", "PagerStats",
1341 "create_paging_callback", "recall_relevant_memories",
1342 # v1.12.1: Async Parallel Primitives
1343 "ParallelExecutor", "FanOutExecutor", "FanOutConfig",
1344 "TaskThrottler", "ParallelTaskResult", "ParallelTaskStatus", "ParallelGatherResult",
1345 "parallel_gather", "parallel_map", "create_parallel_agent_gather",
1346]
1348# Enterprise (v1.5.5)
1349from agentos.enterprise import (
1350 APIKeyManager, APIKey, KeyScope, KeyCreateRequest, KeyCreateResult,
1351 TenantManager, Tenant, TenantConfig, TenantUsage, TenantTier, TenantStatus, TIER_QUOTAS,
1352 User, Role, Permission, ROLE_PERMISSIONS, RBACEngine,
1353 Session as EnterpriseSession, SessionStore as EnterpriseSessionStore,
1354 JWTManager, SSOProvider, OIDCConfig, SAMLConfig, SSOUser,
1355 AuditLogger, AuditEvent, AuditCategory, AuditSeverity, RetentionPolicy,
1356)
1358# System Layer (v1.6.0) — P0: OS-level operations with tiered permissions
1359from agentos.system.permissions import (
1360 SystemPermissionManager, SystemPermission, PermissionTier, PermissionDenied,
1361 SAFE_PERMISSIONS, DEV_PERMISSIONS, FULL_PERMISSIONS,
1362)
1363from agentos.system.file_ops import (
1364 FileOperator, FileOpResult, FileListing,
1365)
1366from agentos.system.shell_exec import (
1367 ShellExecutor, ShellResult, ShellSandbox, ShellPolicy,
1368 READONLY_POLICY, STANDARD_POLICY, FULL_POLICY,
1369)
1370from agentos.system.browser import (
1371 CDPBrowser, BrowserSession, BrowserAction, BrowserResult,
1372)
1374# Desktop Client (v1.7.0) — P1: One-click web desktop (AutoClaw-inspired)
1375from agentos.desktop.server import (
1376 DesktopServer, DesktopConfig,
1377)
1378from agentos.desktop.server import launch_desktop
1380# ── v1.10.0: Evaluation Framework (SWE-bench + GAIA) ──
1381from agentos.eval.benchmark import (
1382 EvalMetric, EvalSuite, EvalCase, EvalSample, EvalResult, EvalReport,
1383 Scorer, ExactMatchScorer, F1Scorer, ROUGELScorer, get_scorer,
1384 SWEBenchLoader, GAIALoader,
1385 EvalRunner, EvalRegistry, evaluate_quick,
1386)
1388# ── v1.10.0: Prompt Hub (versioned templates) ──
1389from agentos.prompt.hub import (
1390 PromptType, PromptTag, PromptVersion, PromptHub,
1391 BUILTIN_PROMPTS, create_default_hub,
1392)
1394# ── v1.12.0: Virtual Memory Pager ──
1395from agentos.memory.pager import (
1396 MemoryPager, SwapStore, MemoryPage, PagerStats,
1397 create_paging_callback, recall_relevant_memories,
1398)
1400# ── v1.11.0: Background Task Manager ──
1401from agentos.background.task_manager import (
1402 BackgroundTaskManager, BackgroundTask, BackgroundTaskStatus, BackgroundTaskConfig,
1403 TaskProgress, ProgressPhase,
1404)
1406# ── v1.11.0: Agent Supervision Tree ──
1407from agentos.background.supervisor import (
1408 AgentSupervisor, SupervisedAgent, SupervisorConfig,
1409 AgentQuota, SupervisionEvent, SupervisionEventType,
1410)
1412# Schema Enforcer (v1.3.9)
1413from agentos.validation.schema_enforcer import (
1414 SchemaEnforcer,
1415 EnforcerConfig,
1416 EnforcerResult,
1417 EnforcerStats,
1418 FixStrategy,
1419)
1421# Conversation Manager (v1.3.10)
1422from agentos.conversation.conversation import (
1423 ConversationManager,
1424 ConversationConfig,
1425 ConversationStats,
1426 ConversationSnapshot,
1427 Message,
1428 MessageRole,
1429 TrimStrategy,
1430)
1432# Guardrails (v1.3.11)
1433from agentos.guardrails.engine import (
1434 GuardrailEngine,
1435 GuardrailResult,
1436 GuardrailAction,
1437 GuardrailRule,
1438 GuardrailCategory,
1439 InputGuardrail,
1440 OutputGuardrail,
1441)
1442from agentos.guardrails.rules import (
1443 PIIRule,
1444 KeywordBlockRule,
1445 LengthLimitRule,
1446 RegexRule,
1447 ToxicityRule,
1448 CodeInjectionRule,
1449 build_default_rules,
1450)
1451from agentos.guardrails.policy import (
1452 GuardrailPolicy,
1453 PolicyEnforcer,
1454 PolicyViolation,
1455)
1457# HITL (v1.3.11)
1458from agentos.hitl.approver import (
1459 HumanInTheLoop,
1460 ApprovalRequest,
1461 ApprovalDecision,
1462 ApprovalStatus,
1463 RiskLevel,
1464 ApprovalPolicy,
1465 ApprovalCallback,
1466)
1467from agentos.hitl.presets import (
1468 default_approval_policy,
1469 permissive_approval_policy,
1470 strict_approval_policy,
1471)
1473# Prompt Optimizer (v1.3.12)
1474from agentos.prompts.optimizer import (
1475 PromptOptimizer,
1476 OptimizerConfig,
1477 OptimizationStrategy,
1478 OptimizationResult,
1479 PromptCandidate,
1480)
1482# Few-Shot Selector (v1.3.12)
1483from agentos.prompts.few_shot import (
1484 FewShotSelector,
1485 Example,
1486 SelectionStrategy,
1487 build_examples,
1488)
1490# SSE Streaming (v1.3.12)
1491from agentos.api.sse import (
1492 SSEEvent,
1493 SSEEventType,
1494 SSEStream,
1495 SSEResponse,
1496)
1498# A2A Store (v1.3.13)
1499from agentos.protocols.a2a_store import (
1500 A2ATaskStore,
1501 InMemoryTaskStore,
1502 SqliteTaskStore,
1503)
1505# A2A Streaming (v1.3.13)
1506from agentos.protocols.a2a_streaming import (
1507 A2AStreamEvent,
1508 TaskProgress,
1509 A2AStreamSession,
1510 A2AStreamManager,
1511)
1513# LLM Provider Module (v1.3.36)
1514from agentos.llm import (
1515 LLMProvider,
1516 OpenAIProvider,
1517 DeepSeekProvider,
1518 AnthropicProvider,
1519 CompletionResult,
1520 CompletionChoice,
1521 CompletionUsage,
1522 TokenUsage,
1523 Message as LLMMessage,
1524 MessageRole as LLMMessageRole,
1525 StreamChunk as LLMStreamChunk,
1526 Tool as LLMTool,
1527 ToolCall as LLMToolCall,
1528 ToolFunction as LLMToolFunction,
1529 ToolParameter as LLMToolParameter,
1530 create_provider as create_llm_provider,
1531)