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