Coverage for agentos/__init__.py: 100%

1"""NexusAgent - Production-grade Agent Framework SDK 

2 

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. 

16 

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). 

38 

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""" 

43 

44__version__ = "1.16.1" 

45 

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). 

49 

50# Core - DI system 

51from agentos.core.di import ( 

52 Agent, 

53 RunContext, 

54 Depends, 

55 inject_tool, 

56 requires_context, 

57) 

58 

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) 

68 

69# Core - CodeAgent 

70from agentos.core.code_agent import ( 

71 CodeAgent, 

72 CodeResult, 

73 CodeStep, 

74) 

75 

76# Protocols - Structured output validation 

77from agentos.protocols.output import ( 

78 StructuredOutput, 

79 validate_output, 

80 OutputValidator, 

81) 

82 

83# Protocols - Agent Card 

84from agentos.protocols.agent_card import ( 

85 AgentCard, 

86 AgentCardRegistry, 

87 AgentCardDiscovery, 

88 discover_local, 

89 create_card, 

90) 

91 

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) 

108 

109# Memory pyramid 

110from agentos.memory.pyramid import ( 

111 MemoryPyramid, 

112 MemoryLayer, 

113 MemoryType, 

114 MemoryItem, 

115) 

116 

117# Evolution engine 

118from agentos.evolution.engine import ( 

119 EvolutionEngine, 

120 EvolutionProposal, 

121 EvolutionStatus, 

122) 

123 

124# Fusion toolkit 

125from agentos.tools.fusion import ( 

126 FusionToolkit, 

127 FusionResult, 

128 ToolSpec, 

129) 

130 

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) 

138 

139# Swarm coordinator 

140from agentos.swarm.coordinator import ( 

141 SwarmCoordinator, 

142 SmartSwarmCoordinator, 

143 SwarmTopology, 

144 SwarmMessage, 

145 ExecutionMode, 

146 SwarmResult, 

147) 

148 

149# Communication layer 

150from agentos.comm.layer import ( 

151 CommunicationLayer, 

152 Blackboard, 

153 EventBus, 

154 Mailbox, 

155) 

156 

157# Orchestration 

158from agentos.orchestration.graph import ( 

159 GraphOrchestrator, 

160 GraphNode, 

161 GraphEdge, 

162) 

163 

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) 

174 

175# Cost tracking (v1.1.4) 

176from agentos.cost.tracker import ( 

177 RunCostSession, 

178 CostTracker, 

179 ModelPricing, 

180 UsageRecord, 

181 PRICING, 

182) 

183 

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) 

197 

198# Models - Router (v1.2.7 minimal) 

199from agentos.models.router import ModelRouter 

200 

201# Security - Sandbox (v1.2.1) 

202from agentos.security.sandbox_executor import ( 

203 SandboxExecutor, 

204 SandboxMode, 

205 SandboxResult, 

206 ProcessSandbox, 

207 DockerSandbox, 

208) 

209 

210# Core - Middleware Pipeline (v1.2.7) 

211from agentos.core.middleware import ( 

212 MiddlewarePhase, 

213 MiddlewareContext, 

214 MiddlewareDecision, 

215 AgentMiddleware, 

216 MiddlewarePipeline, 

217) 

218 

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) 

228 

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) 

240 

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) 

251 

252# Observability (v1.2.7) 

253from agentos.observability import ( 

254 MetricsCollector, 

255 Tracer, 

256 NoopTracer, 

257 CostAnalytics, 

258 BudgetAlert, 

259) 

260 

261# Workflows (v1.2.7) 

262from agentos.workflows import ( 

263 WorkflowEngine, 

264 WorkflowTemplate, 

265) 

266 

267# MCP Protocol (v1.2.7) 

268from agentos.protocols.mcp import ( 

269 MCPClient, 

270 MCPServerConfig, 

271 MCPToolSchema, 

272) 

273 

274# Config System (v1.2.7) 

275from agentos.config import ( 

276 AgentOSConfig, 

277 AgentOSPreset, 

278 ValidationResult, 

279) 

280 

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) 

318 

319# Security - Auditor (v1.2.7) 

320from agentos.security.auditor import ( 

321 SecurityAuditor, 

322 AuditFinding, 

323 AuditReport, 

324) 

325 

326# Tools - Orchestrator (v1.2.7) 

327from agentos.tools.orchestrator import ( 

328 ToolOrchestrator, 

329 DAGBuilder, 

330 DAGSpec, 

331) 

332 

333# Memory - Retriever + Conversation (v1.2.7) 

334from agentos.memory import ( 

335 SemanticMemoryRetriever, 

336 ConversationMemory, 

337) 

338 

339# Prompts (v1.2.7) 

340from agentos.prompts import ( 

341 PromptTemplate, 

342 PromptRegistry, 

343) 

344 

345# Multimodal (v1.2.7) 

346from agentos.multimodal import ( 

347 MultimodalManager, 

348 Modality, 

349) 

350 

351# Vector Store (v1.2.7) 

352from agentos.vectorstore import ( 

353 BaseVectorStore, 

354 FAISSVectorStore, 

355 ChromaVectorStore, 

356) 

357 

358# Errors (v1.2.8) 

359from agentos.errors import ( 

360 ErrorCategory, 

361 ErrorContext, 

362 ErrorFormatter, 

363 HumanError, 

364) 

365 

366# Deployment (v1.2.8) 

367from agentos.deployment import ( 

368 DockerConfig, 

369 ComposeService, 

370 ComposeConfig, 

371) 

372 

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) 

384 

385# Experiments (v1.2.8) 

386from agentos.experiments import ( 

387 ExperimentRunner, 

388 ExperimentConfig, 

389 ExperimentReport, 

390 PromptVariant, 

391 TrialResult, 

392 Evaluator as ExperimentEvaluator, 

393) 

394 

395# Feedback (v1.2.8) 

396from agentos.feedback import ( 

397 FeedbackCollector, 

398 FeedbackRecord, 

399 FeedbackType, 

400 PreferenceLearner, 

401) 

402 

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) 

414 

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) 

427 

428# Models - Routing Strategy (v1.2.8) 

429from agentos.models.routing_strategy import ( 

430 RoutingStrategy, 

431 Complexity, 

432 Budget, 

433) 

434 

435# Swarm Patterns (v1.2.8) 

436from agentos.swarm import ( 

437 SwarmPatterns, 

438 Topology, 

439 CollaborationConfig, 

440 CollaborationResult, 

441) 

442 

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) 

450 

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) 

459 

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) 

483 

484# Logging (v1.2.9) 

485from agentos.log import ( 

486 JSONFormatter, 

487 TraceContext, 

488) 

489 

490# Health (v1.2.9) 

491from agentos.health import ( 

492 HealthChecker, 

493 HealthStatus, 

494 HealthCheck, 

495 CheckResult, 

496) 

497 

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) 

517 

518# Storage (v1.2.9) 

519from agentos.storage import ( 

520 CheckpointStore, 

521 SqliteStore, 

522) 

523 

524# Plugin Manager (v1.2.9) 

525from agentos.plugin_manager import ( 

526 PluginManager, 

527 PluginInfo, 

528) 

529 

530# Cost - Token Counter (v1.2.9) 

531from agentos.cost import ( 

532 TokenCounter, 

533 TokenCount, 

534 CostEstimate, 

535 ModelFamily, 

536) 

537 

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) 

548 

549# Memory - Compressor (v1.2.9) 

550from agentos.memory import ( 

551 ContextCompressor, 

552) 

553 

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) 

568 

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) 

582 

583# Agent Marketplace (v1.3.0) 

584from agentos.agents.market import ( 

585 AgentMarket, 

586 AgentSkill, 

587 AgentCategory, 

588) 

589 

590# Tool-Using Agent (v1.3.38) 

591from agentos.agent import ( 

592 ToolAgent, 

593 ToolExecutor, 

594 AgentConfig, 

595 AgentStep, 

596 AgentResult, 

597 MockLLMProvider, 

598) 

599 

600# API Middleware (v1.3.0) 

601from agentos.api.middleware import ( 

602 CORSConfig, 

603 CORSMiddleware, 

604 AuthConfig, 

605 RequestContext, 

606 RequestIDMiddleware, 

607) 

608 

609# API Streaming (v1.3.0) 

610from agentos.api.streaming import ( 

611 StreamEvent, 

612 StreamSession, 

613 StreamingAgent, 

614) 

615 

616# API Versioning (v1.3.0) 

617from agentos.api.versioning import ( 

618 APIVersion, 

619 SemVer, 

620 VersionStrategy, 

621 VersionConfig, 

622 VersionNegotiator, 

623 DeprecationPolicy, 

624 VersionedRouter, 

625 APIVersioningMiddleware, 

626 VersionExtractor, 

627) 

628 

629# Content Safety Guardrails (v1.20.0) 

630from agentos.security.guardrails import ( 

631 GuardrailsPipeline, 

632 RegexGuard, 

633 RegexRule, 

634 GuardAction, 

635 GuardResult, 

636 GuardViolation, 

637 GuardViolationError, 

638 Category, 

639 ViolationSeverity, 

640 create_default_pipeline, 

641) 

642 

643# Feature Flags (v1.20.0) 

644from agentos.core.feature_flags import ( 

645 FeatureFlagManager, 

646 FlagRule, 

647 FlagType, 

648 FlagContext, 

649 FlagEvaluation, 

650 InMemoryFlagStore, 

651 create_flag_manager, 

652) 

653 

654# Cost Tracking (v1.20.0) 

655from agentos.core.cost_tracker import ( 

656 CostTracker, 

657 PricingRegistry, 

658 ModelPricing, 

659 BudgetLimit, 

660 BudgetAction, 

661 BudgetExceededError, 

662 UsageRecord, 

663) 

664 

665# Benchmark Runner (v1.3.0) 

666from agentos.benchmarks.runner import ( 

667 BenchmarkRunner, 

668 BenchmarkScenario, 

669 BenchmarkConfig, 

670 BenchmarkReport, 

671) 

672 

673# Testing Fixtures (v1.3.0) 

674from agentos.testing.fixtures import ( 

675 MockLLMClient, 

676 MockLLMResponse, 

677 mock_openai_client, 

678 mock_model_response, 

679 sample_config, 

680) 

681 

682# MCP Server (v1.3.0) 

683from agentos.server.mcp_server import ( 

684 MCPServer, 

685 MCPServerConfig, 

686 MCPTool, 

687 MCPResource, 

688 MCPPrompt, 

689) 

690 

691# Concrete Tools (v1.3.0) 

692from agentos.tools.code_agent import ( 

693 CodeAgentTool, 

694 ShellTool, 

695) 

696from agentos.tools.file_tools import ( 

697 ReadFileTool, 

698 WriteFileTool, 

699 ListDirectoryTool, 

700) 

701from agentos.tools.web_tools import ( 

702 WebFetchTool, 

703) 

704 

705# CLI Serve (v1.3.0) 

706from agentos.cli.serve import ( 

707 ServeConfig, 

708 start_api_server, 

709) 

710 

711# Model Route Types (v1.3.1) 

712from agentos.models.router import ( 

713 ModelResponse, 

714 ModelSpec, 

715 AllModelsFailed, 

716) 

717 

718# Gemini Backend (v1.3.1) 

719from agentos.models.backends.gemini import ( 

720 GeminiClient, 

721 GeminiConfig, 

722 GeminiSafetySetting, 

723) 

724 

725# Agent Loop (v1.3.2) 

726from agentos.core.loop import ( 

727 AgentLoop, 

728 LoopConfig, 

729 LoopState, 

730 AgentResult, 

731 MaxIterationsExceeded, 

732 HumanInterruptNeeded, 

733) 

734 

735# API Server (v1.3.2) 

736from agentos.api.server import ( 

737 AgentAPI, 

738 RunRequest, 

739 RunResponse, 

740) 

741 

742# CLI Main (v1.0.0) 

743from agentos.cli.main import main as cli_main 

744 

745# CLI Init (v1.4.1) 

746from agentos.cli.init import ( 

747 init_cli, 

748 scaffold, 

749 load_config, 

750 config_status_text, 

751) 

752 

753# Docs Generator (v1.3.2) 

754from agentos.docs.generator import ( 

755 DocConfig, 

756 generate_api_docs, 

757 generate_quickstart, 

758) 

759 

760# Swarm Coordinator extensions (v1.3.2) 

761from agentos.swarm.coordinator import ( 

762 AgentRole, 

763 MessageBus, 

764) 

765 

766# Model Config (v1.3.2) 

767from agentos.models.router import ( 

768 ModelConfig, 

769 RECOMMENDED_CONFIG, 

770) 

771 

772# OpenAI Backend (v1.3.5) 

773from agentos.models.backends.openai import ( 

774 OpenAIClient, 

775 OpenAIConfig, 

776) 

777 

778# Anthropic Claude Backend (v1.3.5) 

779from agentos.models.backends.anthropic import ( 

780 ClaudeClient, 

781 ClaudeConfig, 

782) 

783 

784# Ollama Backend (v1.3.5) 

785from agentos.models.backends.ollama import ( 

786 OllamaClient, 

787 OllamaConfig, 

788) 

789 

790# RAG Pipeline (v1.3.5) 

791from agentos.rag import ( 

792 RAGPipeline, 

793 TextChunker, 

794 ChunkConfig, 

795 EmbeddingConfig, 

796) 

797 

798# MCP Package (v1.3.6) 

799from agentos.mcp import ( 

800 MCPClient as MCPFullClient, 

801 MCPServerConfig as MCPConfig, 

802 MCPToolInfo, 

803 MCPResourceInfo, 

804 MCPPromptInfo, 

805 MCPError, 

806 connect_mcp_servers, 

807 # MCP Server (v1.5.2) 

808 MCPServer, 

809 MCPToolDef, 

810 MCPResource, 

811 MCPPromptDef, 

812 create_default_server, 

813 start_mcp_server, 

814) 

815from agentos.mcp.adapter import ( 

816 MCPToolAdapter, 

817 MCPToolRegistry, 

818) 

819# Built-in MCP Servers (v1.7.8) 

820from agentos.mcp.builtin_servers import ( 

821 FilesystemServer, 

822 WebFetchServer, 

823 MemoryServer, 

824 BuiltinMCPRegistry, 

825 create_default_registry, 

826) 

827 

828__all__ = [ 

829 # Version 

830 "__version__", 

831 # Core DI 

832 "Agent", 

833 "RunContext", 

834 "Depends", 

835 "inject_tool", 

836 "requires_context", 

837 # Handoff 

838 "Handoff", 

839 "HandoffResult", 

840 "transfer_to", 

841 "can_handle", 

842 "execute_with_handoff", 

843 "HandoffAwareAgent", 

844 # CodeAgent 

845 "CodeAgent", 

846 "CodeResult", 

847 "CodeStep", 

848 # Tool-Using Agent (v1.3.38) 

849 "ToolAgent", 

850 "ToolExecutor", 

851 "AgentConfig", 

852 "AgentStep", 

853 "AgentResult", 

854 "MockLLMProvider", 

855 # Structured output 

856 "StructuredOutput", 

857 "validate_output", 

858 "OutputValidator", 

859 # Agent Card 

860 "AgentCard", 

861 "AgentCardRegistry", 

862 "AgentCardDiscovery", 

863 "discover_local", 

864 "create_card", 

865 # A2A 

866 "A2ATask", 

867 "A2AMessage", 

868 "A2AArtifact", 

869 "A2AHandoff", 

870 "A2ASession", 

871 "A2AClient", 

872 "A2AServer", 

873 "TextPart", 

874 "FilePart", 

875 "DataPart", 

876 "TaskState", 

877 "new_task", 

878 "new_handoff", 

879 # Memory 

880 "MemoryPyramid", 

881 "MemoryLayer", 

882 "MemoryType", 

883 "MemoryItem", 

884 # Evolution 

885 "EvolutionEngine", 

886 "EvolutionProposal", 

887 "EvolutionStatus", 

888 # Fusion 

889 "FusionToolkit", 

890 "FusionResult", 

891 "ToolSpec", 

892 # Risk 

893 "ToolRiskLevel", 

894 "ToolRiskRating", 

895 "get_risk_preset", 

896 "infer_risk_level", 

897 # Swarm 

898 "SwarmCoordinator", 

899 "SmartSwarmCoordinator", 

900 "SwarmTopology", 

901 "SwarmMessage", 

902 "ExecutionMode", 

903 "SwarmResult", 

904 # Communication 

905 "CommunicationLayer", 

906 "Blackboard", 

907 "EventBus", 

908 "Mailbox", 

909 # Orchestration 

910 "GraphOrchestrator", 

911 "GraphNode", 

912 "GraphEdge", 

913 # Concurrency 

914 "AsyncBatchExecutor", 

915 "BatchTaskStatus", 

916 "TaskSpec", 

917 "BatchTaskResult", 

918 "BatchConfig", 

919 "BatchResult", 

920 "BatchStrategy", 

921 # Cost 

922 "RunCostSession", 

923 "CostTracker", 

924 "ModelPricing", 

925 "UsageRecord", 

926 "PRICING", 

927 # Resilience 

928 "CancellationSource", 

929 "CancelledError", 

930 "RetryConfig", 

931 "CircuitBreaker", 

932 "CircuitBreakerConfig", 

933 "ResilienceConfig", 

934 "ResilientCall", 

935 "retry_with_backoff", 

936 "with_timeout", 

937 "with_fallback", 

938 # Router 

939 "ModelRouter", 

940 # Sandbox 

941 "SandboxExecutor", 

942 "SandboxMode", 

943 "SandboxResult", 

944 "ProcessSandbox", 

945 "DockerSandbox", 

946 # Middleware (v1.2.7) 

947 "MiddlewarePhase", 

948 "MiddlewareContext", 

949 "MiddlewareDecision", 

950 "AgentMiddleware", 

951 "MiddlewarePipeline", 

952 # Queue (v1.2.7) 

953 "TaskQueue", 

954 "QueueTaskState", 

955 "TaskPriority", 

956 "RateLimiter", 

957 "RateLimitStrategy", 

958 "RateLimitConfig", 

959 # Cache (v1.2.7) 

960 "LLMCache", 

961 "CacheEntry", 

962 "BaseEmbedder", 

963 "OpenAIEmbedder", 

964 "LocalEmbedder", 

965 "CohereEmbedder", 

966 "ResponseCache", 

967 "CacheKeyStrategy", 

968 # Plugins (v1.2.7) 

969 "PluginRegistry", 

970 "RegisteredPlugin", 

971 "PluginStatus", 

972 "PluginDiscovery", 

973 "DiscoveredPlugin", 

974 "PluginLoader", 

975 "LifecycleManager", 

976 # Observability (v1.2.7) 

977 "MetricsCollector", 

978 "Tracer", 

979 "NoopTracer", 

980 "CostAnalytics", 

981 "BudgetAlert", 

982 # Workflows (v1.2.7) 

983 "WorkflowEngine", 

984 "WorkflowTemplate", 

985 # MCP Protocol (v1.2.7) 

986 "MCPClient", 

987 "MCPServerConfig", 

988 "MCPToolSchema", 

989 # Config (v1.2.7) 

990 "AgentOSConfig", 

991 "AgentOSPreset", 

992 "ValidationResult", 

993 # Evaluation (v1.2.7) 

994 "Evaluator", 

995 "CompositeScorer", 

996 "BenchmarkCase", 

997 "EvalResult", 

998 # Audit (v1.2.7) 

999 "SecurityAuditor", 

1000 "AuditFinding", 

1001 "AuditReport", 

1002 # Orchestrator (v1.2.7) 

1003 "ToolOrchestrator", 

1004 "DAGBuilder", 

1005 "DAGSpec", 

1006 # Memory (v1.2.7) 

1007 "SemanticMemoryRetriever", 

1008 "ConversationMemory", 

1009 # Prompts (v1.2.7) 

1010 "PromptTemplate", 

1011 "PromptRegistry", 

1012 # Multimodal (v1.2.7) 

1013 "MultimodalManager", 

1014 "Modality", 

1015 # Vector Store (v1.2.7) 

1016 "BaseVectorStore", 

1017 "FAISSVectorStore", 

1018 "ChromaVectorStore", 

1019 # Errors (v1.2.8) 

1020 "ErrorCategory", 

1021 "ErrorContext", 

1022 "ErrorFormatter", 

1023 "HumanError", 

1024 # Deployment (v1.2.8) 

1025 "DockerConfig", 

1026 "ComposeService", 

1027 "ComposeConfig", 

1028 # Monitoring (v1.2.8) 

1029 "Alert", 

1030 "AlertEvaluator", 

1031 "AlertRule", 

1032 "AlertSeverity", 

1033 "AlertState", 

1034 "MonitoringConfig", 

1035 "WebhookConfig", 

1036 "WebhookDispatcher", 

1037 # Experiments (v1.2.8) 

1038 "ExperimentRunner", 

1039 "ExperimentConfig", 

1040 "ExperimentReport", 

1041 "PromptVariant", 

1042 "TrialResult", 

1043 "ExperimentEvaluator", 

1044 # Feedback (v1.2.8) 

1045 "FeedbackCollector", 

1046 "FeedbackRecord", 

1047 "FeedbackType", 

1048 "PreferenceLearner", 

1049 # Memory extensions (v1.2.8) 

1050 "MemorySummarizer", 

1051 "ImportanceScorer", 

1052 "MemoryChunk", 

1053 "LongTermMemory", 

1054 "MemoryStore", 

1055 "WorkingMemory", 

1056 "WorkingMemoryItem", 

1057 "VectorMemory", 

1058 # Orchestration extensions (v1.2.8) 

1059 "A2ARouter", 

1060 "RouterAgentCard", 

1061 "RouterTask", 

1062 "TaskResult", 

1063 "TaskStatus", 

1064 "AgentGraph", 

1065 "GraphRecipe", 

1066 "GraphNodeState", 

1067 "GraphResult", 

1068 # Models - Routing (v1.2.8) 

1069 "RoutingStrategy", 

1070 "Complexity", 

1071 "Budget", 

1072 # Swarm Patterns (v1.2.8) 

1073 "SwarmPatterns", 

1074 "Topology", 

1075 "CollaborationConfig", 

1076 "CollaborationResult", 

1077 # Code Sandbox (v1.9.5) 

1078 "CodeSandbox", 

1079 "CodeSandboxResult", 

1080 "CodeTestCase", 

1081 "CodeFeedbackExtractor", 

1082 # Human-in-the-Loop (v1.9.5) 

1083 "HITLManager", 

1084 "HITLConfig", 

1085 "Breakpoint", 

1086 "BreakpointType", 

1087 "HumanDecision", 

1088 # Core extensions (v1.2.9) 

1089 "AgentContext", 

1090 "ContextManager", 

1091 "CoreMessage", 

1092 "CoreToolCall", 

1093 "CoreToolResult", 

1094 "AgentStateMachine", 

1095 "AgentState", 

1096 "StateTransition", 

1097 "TransitionError", 

1098 "StateTimeoutError", 

1099 "StreamChunk", 

1100 "StreamEmitter", 

1101 "StreamEvent", 

1102 "ResponseCollector", 

1103 "Session", 

1104 "SessionStore", 

1105 "AsyncAgentLoop", 

1106 "AsyncLoopConfig", 

1107 "AsyncInvocationResult", 

1108 "AsyncContextManager", 

1109 # Logging (v1.2.9) 

1110 "JSONFormatter", 

1111 "TraceContext", 

1112 # Health (v1.2.9) 

1113 "HealthChecker", 

1114 "HealthStatus", 

1115 "HealthCheck", 

1116 "CheckResult", 

1117 # Security extensions (v1.9.9) 

1118 "GuardPipeline", 

1119 "InputGuard", 

1120 "OutputGuard", 

1121 "PIIDetector", 

1122 "ContentSafetyFilter", 

1123 "GuardChainResult", 

1124 "GuardResult", 

1125 "GuardAction", 

1126 "Severity", 

1127 "create_strict_guard", 

1128 "create_permissive_guard", 

1129 "SandboxManager", 

1130 "Sandbox", 

1131 "SafetyReport", 

1132 "RiskLevel", 

1133 "LLMSafetyAnalyzer", 

1134 # Storage (v1.2.9) 

1135 "CheckpointStore", 

1136 "SqliteStore",