alysis-code 0.13.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- alysis_code/__init__.py +3 -0
- alysis_code/__main__.py +4 -0
- alysis_code/_build_info.py +14 -0
- alysis_code/account_login.py +468 -0
- alysis_code/agent/README.md +35 -0
- alysis_code/agent/__init__.py +11 -0
- alysis_code/agent/acceptance_contract.py +2217 -0
- alysis_code/agent/blast_radius.py +1403 -0
- alysis_code/agent/cache_keepalive.py +227 -0
- alysis_code/agent/completion_certificate.py +366 -0
- alysis_code/agent/completion_gate.py +306 -0
- alysis_code/agent/empty_response_stall.py +403 -0
- alysis_code/agent/errors.py +28 -0
- alysis_code/agent/llm_calls.py +475 -0
- alysis_code/agent/mutation_classification.py +227 -0
- alysis_code/agent/prompt_context.py +2508 -0
- alysis_code/agent/read_ledger.py +253 -0
- alysis_code/agent/regression_baseline.py +642 -0
- alysis_code/agent/reproduction_first.py +610 -0
- alysis_code/agent/sensitive_output.py +629 -0
- alysis_code/agent/session.py +3218 -0
- alysis_code/agent/steering.py +191 -0
- alysis_code/agent/subagent_execution.py +5177 -0
- alysis_code/agent/subagent_workspace.py +666 -0
- alysis_code/agent/tools_assembly.py +4728 -0
- alysis_code/agent/turn/__init__.py +101 -0
- alysis_code/agent/turn/core.py +8483 -0
- alysis_code/agent/turn/events.py +113 -0
- alysis_code/agent/turn/exploration.py +590 -0
- alysis_code/agent/turn/interventions.py +65 -0
- alysis_code/agent/turn/read_cache.py +420 -0
- alysis_code/agent/turn/snapshot.py +179 -0
- alysis_code/agent/turn_contract.py +661 -0
- alysis_code/agent/turn_path.py +129 -0
- alysis_code/agent/verification.py +2885 -0
- alysis_code/agent/verification_commands.py +512 -0
- alysis_code/agent/verification_evidence.py +738 -0
- alysis_code/agent_loop.py +693 -0
- alysis_code/agent_runtimes/__init__.py +51 -0
- alysis_code/agent_runtimes/base.py +114 -0
- alysis_code/agent_runtimes/builtins.py +129 -0
- alysis_code/agent_runtimes/codex_cli.py +664 -0
- alysis_code/agent_runtimes/host.py +263 -0
- alysis_code/agent_runtimes/registry.py +64 -0
- alysis_code/agent_runtimes/service.py +150 -0
- alysis_code/agentbox_client.py +416 -0
- alysis_code/agentbox_integration.py +310 -0
- alysis_code/alysis_cloud.py +152 -0
- alysis_code/approval_scope.py +276 -0
- alysis_code/assets/README.md +33 -0
- alysis_code/assets/__init__.py +126 -0
- alysis_code/assets/asset_read_core.py +281 -0
- alysis_code/assets/budget_allocator.py +456 -0
- alysis_code/assets/comprehender.py +759 -0
- alysis_code/assets/index.py +654 -0
- alysis_code/assets/ingestion.py +275 -0
- alysis_code/assets/legacy_migration.py +413 -0
- alysis_code/assets/models.py +263 -0
- alysis_code/assets/ocr.py +239 -0
- alysis_code/assets/owl/ascii/f-000.txt +13 -0
- alysis_code/assets/owl/ascii/f-001.txt +13 -0
- alysis_code/assets/owl/ascii/f-002.txt +13 -0
- alysis_code/assets/owl/ascii/f-003.txt +13 -0
- alysis_code/assets/owl/ascii/f-004.txt +13 -0
- alysis_code/assets/owl/ascii/f-005.txt +13 -0
- alysis_code/assets/owl/ascii/f-006.txt +13 -0
- alysis_code/assets/owl/ascii/f-007.txt +13 -0
- alysis_code/assets/owl/ascii/f-008.txt +13 -0
- alysis_code/assets/owl/ascii/f-009.txt +13 -0
- alysis_code/assets/owl/ascii/f-010.txt +13 -0
- alysis_code/assets/owl/ascii/f-011.txt +13 -0
- alysis_code/assets/owl/ascii/f-012.txt +13 -0
- alysis_code/assets/owl/ascii/f-013.txt +13 -0
- alysis_code/assets/owl/ascii/f-014.txt +13 -0
- alysis_code/assets/owl/ascii/f-015.txt +13 -0
- alysis_code/assets/owl/ascii/f-016.txt +13 -0
- alysis_code/assets/owl/ascii/f-017.txt +13 -0
- alysis_code/assets/owl/ascii/f-018.txt +13 -0
- alysis_code/assets/owl/ascii/f-019.txt +13 -0
- alysis_code/assets/owl/ascii/f-020.txt +13 -0
- alysis_code/assets/owl/index.html +98 -0
- alysis_code/assets/owl/show-owl.sh +761 -0
- alysis_code/assets/paths.py +49 -0
- alysis_code/assets/plan_binding.py +326 -0
- alysis_code/assets/planner_context.py +466 -0
- alysis_code/assets/planner_tools.py +184 -0
- alysis_code/assets/prompts.py +101 -0
- alysis_code/assets/replanner_context.py +239 -0
- alysis_code/assets/surface.py +521 -0
- alysis_code/assets/untrusted_content.py +48 -0
- alysis_code/assets/usage_logger.py +94 -0
- alysis_code/assets/worker_mirror.py +428 -0
- alysis_code/assets/worker_section.py +303 -0
- alysis_code/assets/worker_tools.py +468 -0
- alysis_code/atomic_io.py +83 -0
- alysis_code/auth_diagnostics.py +272 -0
- alysis_code/background_runner.py +366 -0
- alysis_code/branding.py +270 -0
- alysis_code/budget_policy.py +390 -0
- alysis_code/build_identity.py +465 -0
- alysis_code/builtin_hooks/__init__.py +7 -0
- alysis_code/builtin_hooks/notify_done_windows.py +65 -0
- alysis_code/bwrap_etc.py +76 -0
- alysis_code/cancellation.py +41 -0
- alysis_code/capabilities.py +137 -0
- alysis_code/chatgpt_codex_static_provider.py +133 -0
- alysis_code/cli.py +51 -0
- alysis_code/cli_impl/__init__.py +1 -0
- alysis_code/cli_impl/assets_cli.py +537 -0
- alysis_code/cli_impl/assets_modal.py +412 -0
- alysis_code/cli_impl/chat/__init__.py +156 -0
- alysis_code/cli_impl/chat/commands.py +2616 -0
- alysis_code/cli_impl/chat/loop.py +4508 -0
- alysis_code/cli_impl/chat/mid_turn_policy.py +125 -0
- alysis_code/cli_impl/chat/rendering.py +444 -0
- alysis_code/cli_impl/chat/state.py +124 -0
- alysis_code/cli_impl/chat_resume.py +830 -0
- alysis_code/cli_impl/chat_slash_completer.py +258 -0
- alysis_code/cli_impl/commands/__init__.py +11 -0
- alysis_code/cli_impl/commands/_shared.py +89 -0
- alysis_code/cli_impl/commands/auth.py +623 -0
- alysis_code/cli_impl/commands/chat_resume_helpers.py +1531 -0
- alysis_code/cli_impl/commands/chat_state.py +158 -0
- alysis_code/cli_impl/commands/chat_status.py +1248 -0
- alysis_code/cli_impl/commands/chat_terminal.py +942 -0
- alysis_code/cli_impl/commands/chat_tui_panels.py +1018 -0
- alysis_code/cli_impl/commands/cli_common.py +1223 -0
- alysis_code/cli_impl/commands/cli_surface.py +77 -0
- alysis_code/cli_impl/commands/config.py +131 -0
- alysis_code/cli_impl/commands/conventions.py +85 -0
- alysis_code/cli_impl/commands/execution_helpers.py +350 -0
- alysis_code/cli_impl/commands/extensions.py +401 -0
- alysis_code/cli_impl/commands/forge.py +1282 -0
- alysis_code/cli_impl/commands/forge_asset_view.py +121 -0
- alysis_code/cli_impl/commands/forge_helpers.py +1215 -0
- alysis_code/cli_impl/commands/hooks.py +737 -0
- alysis_code/cli_impl/commands/ide_bridge.py +31 -0
- alysis_code/cli_impl/commands/mcp.py +700 -0
- alysis_code/cli_impl/commands/profile.py +453 -0
- alysis_code/cli_impl/commands/prompt_helpers.py +307 -0
- alysis_code/cli_impl/commands/report.py +88 -0
- alysis_code/cli_impl/commands/root.py +1118 -0
- alysis_code/cli_impl/commands/sandbox.py +184 -0
- alysis_code/cli_impl/commands/server.py +54 -0
- alysis_code/cli_impl/commands/sessions.py +252 -0
- alysis_code/cli_impl/commands/skills.py +404 -0
- alysis_code/cli_impl/commands/startup.py +946 -0
- alysis_code/cli_impl/commands/tools.py +335 -0
- alysis_code/cli_impl/commands/update.py +364 -0
- alysis_code/cli_impl/commands/welcome.py +972 -0
- alysis_code/cli_impl/config_menu.py +3882 -0
- alysis_code/cli_impl/forge.py +4509 -0
- alysis_code/cli_impl/forge_recovery.py +485 -0
- alysis_code/cli_impl/setup_wizard.py +2409 -0
- alysis_code/cli_impl/tui/__init__.py +58 -0
- alysis_code/cli_impl/tui/app.py +4551 -0
- alysis_code/cli_impl/tui/config.py +32 -0
- alysis_code/cli_impl/tui/config_flow.py +2754 -0
- alysis_code/cli_impl/tui/config_overlay.py +566 -0
- alysis_code/cli_impl/tui/content.py +78 -0
- alysis_code/cli_impl/tui/footer.py +218 -0
- alysis_code/cli_impl/tui/forge_status.py +136 -0
- alysis_code/cli_impl/tui/markdown.py +244 -0
- alysis_code/cli_impl/tui/owl.py +109 -0
- alysis_code/cli_impl/tui/plan_meta.py +477 -0
- alysis_code/cli_impl/tui/setup_app.py +519 -0
- alysis_code/cli_impl/tui/setup_flow.py +1622 -0
- alysis_code/cli_impl/tui/state.py +101 -0
- alysis_code/cli_impl/tui/subagent_identity.py +66 -0
- alysis_code/cli_impl/tui/subagent_panel.py +186 -0
- alysis_code/cli_impl/tui/surface.py +796 -0
- alysis_code/cli_impl/tui/transcript.py +514 -0
- alysis_code/cli_impl/tui/update_prompt.py +79 -0
- alysis_code/cli_impl/tui/workspace_guard.py +384 -0
- alysis_code/clipboard.py +172 -0
- alysis_code/code_review.py +1211 -0
- alysis_code/compaction/__init__.py +28 -0
- alysis_code/compaction/conversation_compactor.py +2932 -0
- alysis_code/compaction/importance.py +177 -0
- alysis_code/compaction/settings.py +297 -0
- alysis_code/compaction/tool_output_offload.py +447 -0
- alysis_code/config.py +3509 -0
- alysis_code/conflict_auto_resolver.py +895 -0
- alysis_code/context/__init__.py +1 -0
- alysis_code/context/tool_schema_budgeter.py +220 -0
- alysis_code/crash_diagnostics.py +282 -0
- alysis_code/custom_tools/README.md +34 -0
- alysis_code/custom_tools/__init__.py +43 -0
- alysis_code/custom_tools/discovery.py +903 -0
- alysis_code/custom_tools/runtime.py +1516 -0
- alysis_code/custom_tools/session.py +227 -0
- alysis_code/custom_tools/trust.py +232 -0
- alysis_code/diff_paths.py +113 -0
- alysis_code/direction_change.py +293 -0
- alysis_code/dispatch_timing.py +306 -0
- alysis_code/durable_service_manager.py +1236 -0
- alysis_code/edit_discipline.py +659 -0
- alysis_code/error_text.py +73 -0
- alysis_code/execution_budget.py +411 -0
- alysis_code/execution_context.py +915 -0
- alysis_code/execution_deadline.py +1065 -0
- alysis_code/execution_shared.py +1904 -0
- alysis_code/extensions/README.md +30 -0
- alysis_code/extensions/__init__.py +93 -0
- alysis_code/extensions/activation.py +138 -0
- alysis_code/extensions/install.py +1436 -0
- alysis_code/extensions/manifest.py +487 -0
- alysis_code/extensions/models.py +74 -0
- alysis_code/extensions/paths.py +56 -0
- alysis_code/extensions/registry.json +4 -0
- alysis_code/extensions/registry.py +52 -0
- alysis_code/extensions/state.py +83 -0
- alysis_code/extensions/workspace_trust.py +101 -0
- alysis_code/failed_task_evidence.py +369 -0
- alysis_code/failure_category.py +315 -0
- alysis_code/feedback_report.py +1647 -0
- alysis_code/file_classification.py +485 -0
- alysis_code/forge.py +2064 -0
- alysis_code/forge_completion.py +362 -0
- alysis_code/forge_events.py +475 -0
- alysis_code/frontmatter_utils.py +95 -0
- alysis_code/git_evidence.py +1181 -0
- alysis_code/git_ops.py +560 -0
- alysis_code/git_safe.py +62 -0
- alysis_code/git_worktrees.py +190 -0
- alysis_code/hooks/README.md +33 -0
- alysis_code/hooks/__init__.py +67 -0
- alysis_code/hooks/audit.py +171 -0
- alysis_code/hooks/config.py +225 -0
- alysis_code/hooks/dispatcher.py +1110 -0
- alysis_code/hooks/models.py +447 -0
- alysis_code/hooks/trust.py +202 -0
- alysis_code/host_actions.py +543 -0
- alysis_code/host_browser.py +103 -0
- alysis_code/ide/__init__.py +5 -0
- alysis_code/ide/activity_events.py +399 -0
- alysis_code/ide/approvals.py +337 -0
- alysis_code/ide/artifacts.py +153 -0
- alysis_code/ide/browser_egress_proxy.py +1076 -0
- alysis_code/ide/cdp_websocket_transport.py +1192 -0
- alysis_code/ide/change_ledger.py +1721 -0
- alysis_code/ide/context_blocks.py +979 -0
- alysis_code/ide/event_stream.py +531 -0
- alysis_code/ide/forge_protocol.py +3112 -0
- alysis_code/ide/forge_request_ledger.py +737 -0
- alysis_code/ide/health.py +965 -0
- alysis_code/ide/managed_browser.py +2251 -0
- alysis_code/ide/management_protocol.py +3414 -0
- alysis_code/ide/mcp_oauth_coordinator.py +744 -0
- alysis_code/ide/mcp_oauth_lifecycle.py +1504 -0
- alysis_code/ide/prompt_queue.py +1070 -0
- alysis_code/ide/protocol.py +191 -0
- alysis_code/ide/resumable_swarm.py +1543 -0
- alysis_code/ide/session_search.py +295 -0
- alysis_code/ide/stdio_bridge.py +9935 -0
- alysis_code/ide/structured_state.py +1579 -0
- alysis_code/ide/swarm_protocol.py +816 -0
- alysis_code/integration_gate.py +506 -0
- alysis_code/interactive_input_guard.py +39 -0
- alysis_code/interactive_plan_mode.py +26 -0
- alysis_code/internal_artifacts.py +179 -0
- alysis_code/knowledge_base.py +1409 -0
- alysis_code/knowledge_capture.py +1190 -0
- alysis_code/knowledge_librarian.py +605 -0
- alysis_code/language_policy.py +34 -0
- alysis_code/litellm_static_provider.py +535 -0
- alysis_code/llm/__init__.py +1 -0
- alysis_code/llm/anthropic_messages.py +2288 -0
- alysis_code/llm/base.py +71 -0
- alysis_code/llm/cache_capabilities.py +985 -0
- alysis_code/llm/cache_control_blocks.py +244 -0
- alysis_code/llm/cache_policy.py +388 -0
- alysis_code/llm/factory.py +373 -0
- alysis_code/llm/gemini_generate_content.py +2652 -0
- alysis_code/llm/gemini_interactions.py +739 -0
- alysis_code/llm/metadata.py +450 -0
- alysis_code/llm/openai_compat.py +2947 -0
- alysis_code/llm/openai_responses.py +2604 -0
- alysis_code/llm/protocols.py +609 -0
- alysis_code/llm/provider_limits.py +525 -0
- alysis_code/llm/request_plan.py +389 -0
- alysis_code/llm/request_shape.py +238 -0
- alysis_code/llm/streaming.py +108 -0
- alysis_code/llm/temperature_compat.py +78 -0
- alysis_code/llm/types.py +195 -0
- alysis_code/llm/usage_normalization.py +222 -0
- alysis_code/llm_error_display.py +315 -0
- alysis_code/logging_redaction.py +326 -0
- alysis_code/managed_host_deadline.py +196 -0
- alysis_code/mcp/README.md +33 -0
- alysis_code/mcp/__init__.py +24 -0
- alysis_code/mcp/client.py +1137 -0
- alysis_code/mcp/config.py +597 -0
- alysis_code/mcp/errors.py +113 -0
- alysis_code/mcp/forge_scope.py +154 -0
- alysis_code/mcp/jsonrpc.py +214 -0
- alysis_code/mcp/manager.py +2308 -0
- alysis_code/mcp/models.py +666 -0
- alysis_code/mcp/oauth.py +972 -0
- alysis_code/mcp/oauth_runtime.py +310 -0
- alysis_code/mcp/oauth_store.py +276 -0
- alysis_code/mcp/prompts.py +329 -0
- alysis_code/mcp/resources.py +295 -0
- alysis_code/mcp/roots.py +106 -0
- alysis_code/mcp/server_requests.py +75 -0
- alysis_code/mcp/token_store.py +859 -0
- alysis_code/mcp/transport_http.py +1338 -0
- alysis_code/mcp/transport_stdio.py +1267 -0
- alysis_code/mcp/untrusted_content.py +119 -0
- alysis_code/merge_conflict_reviewer.py +729 -0
- alysis_code/model_catalog/__init__.py +1 -0
- alysis_code/model_catalog/chatgpt_codex_subscription_snapshot.json +186 -0
- alysis_code/model_catalog/litellm_model_prices_snapshot.json +44715 -0
- alysis_code/model_catalog/litellm_model_prices_snapshot.meta.json +17 -0
- alysis_code/model_metadata_policy.py +223 -0
- alysis_code/model_metadata_utils.py +103 -0
- alysis_code/model_registry.py +1420 -0
- alysis_code/model_router.py +147 -0
- alysis_code/permission_policy.py +1016 -0
- alysis_code/personas.py +451 -0
- alysis_code/pipeline_facts.py +233 -0
- alysis_code/plan_assistant.py +4763 -0
- alysis_code/plan_mode.py +393 -0
- alysis_code/plan_reconciliation.py +1228 -0
- alysis_code/plan_repair.py +652 -0
- alysis_code/plan_validation.py +1099 -0
- alysis_code/planning_constraints.py +904 -0
- alysis_code/policy.py +95 -0
- alysis_code/preview_server.py +457 -0
- alysis_code/process_reaping.py +566 -0
- alysis_code/profile_presets.py +1834 -0
- alysis_code/profiles.py +666 -0
- alysis_code/provider_auth/__init__.py +29 -0
- alysis_code/provider_auth/base.py +99 -0
- alysis_code/provider_auth/openai_codex.py +951 -0
- alysis_code/provider_auth/registry.py +76 -0
- alysis_code/provider_auth/store.py +125 -0
- alysis_code/provider_diagnostics.py +1209 -0
- alysis_code/provider_model_catalog.py +685 -0
- alysis_code/provider_telemetry.py +1699 -0
- alysis_code/provider_url.py +75 -0
- alysis_code/reasoning_contracts.py +911 -0
- alysis_code/remote_sync.py +350 -0
- alysis_code/replanning.py +1195 -0
- alysis_code/repo_scan.py +1152 -0
- alysis_code/request_estimation.py +296 -0
- alysis_code/review_gate.py +617 -0
- alysis_code/run_lock.py +1141 -0
- alysis_code/run_outcome.py +58 -0
- alysis_code/run_provenance.py +774 -0
- alysis_code/run_state.py +445 -0
- alysis_code/runtime_artifacts.py +116 -0
- alysis_code/runtime_context_features.py +78 -0
- alysis_code/runtime_kind.py +52 -0
- alysis_code/safety/__init__.py +11 -0
- alysis_code/safety/mcp_sanitize.py +29 -0
- alysis_code/safety/safe_http.py +297 -0
- alysis_code/safety/subagent_report.py +184 -0
- alysis_code/sandbox_doctor.py +682 -0
- alysis_code/sandbox_runner.py +1025 -0
- alysis_code/sandbox_settings.py +423 -0
- alysis_code/serialized_paths.py +355 -0
- alysis_code/server/__init__.py +3 -0
- alysis_code/server/app.py +367 -0
- alysis_code/server/auth.py +34 -0
- alysis_code/server/job_config.py +30 -0
- alysis_code/server/settings.py +215 -0
- alysis_code/server/store.py +193 -0
- alysis_code/server/worker_runner.py +657 -0
- alysis_code/service_persistence.py +355 -0
- alysis_code/session_artifacts.py +108 -0
- alysis_code/session_metrics.py +331 -0
- alysis_code/session_store.py +624 -0
- alysis_code/skills/README.md +34 -0
- alysis_code/skills/__init__.py +104 -0
- alysis_code/skills/conventions.py +84 -0
- alysis_code/skills/discovery.py +176 -0
- alysis_code/skills/eval_models.py +232 -0
- alysis_code/skills/eval_runner.py +372 -0
- alysis_code/skills/evals.py +1344 -0
- alysis_code/skills/install.py +293 -0
- alysis_code/skills/loader.py +118 -0
- alysis_code/skills/matching.py +103 -0
- alysis_code/skills/models.py +71 -0
- alysis_code/skills/paths.py +56 -0
- alysis_code/skills/prompting.py +500 -0
- alysis_code/skills/scaffold.py +142 -0
- alysis_code/skills/state.py +441 -0
- alysis_code/skills/transactions.py +125 -0
- alysis_code/skills/validation.py +304 -0
- alysis_code/step_budget.py +238 -0
- alysis_code/subagent_labels.py +49 -0
- alysis_code/subagents.py +1072 -0
- alysis_code/surface/__init__.py +80 -0
- alysis_code/surface/base.py +305 -0
- alysis_code/surface/console.py +387 -0
- alysis_code/surface/events.py +372 -0
- alysis_code/surface/hidden_surface.py +529 -0
- alysis_code/surface/noop_surface.py +219 -0
- alysis_code/surface/rich_surface.py +1555 -0
- alysis_code/surface/styles.py +67 -0
- alysis_code/surface/theme.py +455 -0
- alysis_code/surface/types.py +100 -0
- alysis_code/swarm_backend.py +926 -0
- alysis_code/swarm_orchestrator.py +4020 -0
- alysis_code/swarm_scheduler.py +441 -0
- alysis_code/swarm_trace.py +429 -0
- alysis_code/swarm_worker.py +2119 -0
- alysis_code/swarm_write_guard.py +348 -0
- alysis_code/task_dependencies.py +170 -0
- alysis_code/task_readiness.py +992 -0
- alysis_code/task_scope.py +2148 -0
- alysis_code/terminal_manager.py +762 -0
- alysis_code/terminal_ownership.py +460 -0
- alysis_code/text_normalization.py +30 -0
- alysis_code/token_budget.py +97 -0
- alysis_code/tools/README.md +34 -0
- alysis_code/tools/__init__.py +1 -0
- alysis_code/tools/artifacts.py +127 -0
- alysis_code/tools/availability.py +188 -0
- alysis_code/tools/fs.py +1456 -0
- alysis_code/tools/git.py +461 -0
- alysis_code/tools/history.py +229 -0
- alysis_code/tools/http_timeout.py +78 -0
- alysis_code/tools/image_generation.py +552 -0
- alysis_code/tools/registry.py +2936 -0
- alysis_code/tools/repo_map.py +476 -0
- alysis_code/tools/search.py +563 -0
- alysis_code/tools/shell.py +135 -0
- alysis_code/tools/symbols.py +1350 -0
- alysis_code/tools/test_discovery.py +643 -0
- alysis_code/tools/web.py +482 -0
- alysis_code/tools/web_search.py +2012 -0
- alysis_code/tools/web_search_dashscope.py +557 -0
- alysis_code/tools/web_search_ddgs.py +221 -0
- alysis_code/tools/web_search_provider_adapters.py +1429 -0
- alysis_code/tools/web_search_tavily.py +194 -0
- alysis_code/updates.py +933 -0
- alysis_code/usage_tracker.py +1990 -0
- alysis_code/verification_command_analysis.py +1004 -0
- alysis_code/verification_contract.py +574 -0
- alysis_code/verification_failure_summary.py +273 -0
- alysis_code/verification_repair.py +385 -0
- alysis_code/verify_gate.py +3129 -0
- alysis_code/web_research.py +1872 -0
- alysis_code/web_search_adapters.py +66 -0
- alysis_code/web_search_policy.py +27 -0
- alysis_code/workspace_binding.py +389 -0
- alysis_code/workspace_binding_ui.py +408 -0
- alysis_code/workspace_context.py +273 -0
- alysis_code/workspace_isolation.py +138 -0
- alysis_code/workspace_provisioning.py +455 -0
- alysis_code-0.13.0.dist-info/METADATA +507 -0
- alysis_code-0.13.0.dist-info/RECORD +458 -0
- alysis_code-0.13.0.dist-info/WHEEL +4 -0
- alysis_code-0.13.0.dist-info/entry_points.txt +3 -0
- alysis_code-0.13.0.dist-info/licenses/LICENSE +176 -0
- alysis_code-0.13.0.dist-info/licenses/NOTICE +4 -0
|
@@ -0,0 +1,389 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
import copy
|
|
4
|
+
import hashlib
|
|
5
|
+
import json
|
|
6
|
+
from collections.abc import Mapping
|
|
7
|
+
from dataclasses import dataclass, field, replace
|
|
8
|
+
from typing import Any
|
|
9
|
+
|
|
10
|
+
from ..request_estimation import (
|
|
11
|
+
estimate_message_tokens,
|
|
12
|
+
estimate_provider_payload_tokens,
|
|
13
|
+
estimate_request_token_breakdown,
|
|
14
|
+
estimate_tool_schema_tokens,
|
|
15
|
+
sanitize_messages_for_estimation,
|
|
16
|
+
)
|
|
17
|
+
from .cache_control_blocks import cacheable_prefix_message_count
|
|
18
|
+
|
|
19
|
+
|
|
20
|
+
def _stable_payload_hash(value: Any) -> str:
|
|
21
|
+
serialized = json.dumps(
|
|
22
|
+
value,
|
|
23
|
+
ensure_ascii=False,
|
|
24
|
+
sort_keys=True,
|
|
25
|
+
separators=(",", ":"),
|
|
26
|
+
)
|
|
27
|
+
return hashlib.sha256(serialized.encode("utf-8", errors="surrogatepass")).hexdigest()
|
|
28
|
+
|
|
29
|
+
|
|
30
|
+
def _optional_payload_token_estimate(payload: Any) -> int | None:
|
|
31
|
+
if payload is None:
|
|
32
|
+
return None
|
|
33
|
+
return estimate_provider_payload_tokens(payload)
|
|
34
|
+
|
|
35
|
+
|
|
36
|
+
def _safe_int(value: Any) -> int | None:
|
|
37
|
+
try:
|
|
38
|
+
parsed = int(value)
|
|
39
|
+
except (TypeError, ValueError):
|
|
40
|
+
return None
|
|
41
|
+
if parsed < 0:
|
|
42
|
+
return None
|
|
43
|
+
return parsed
|
|
44
|
+
|
|
45
|
+
|
|
46
|
+
def _safe_label(value: Any, default: str = "") -> str:
|
|
47
|
+
text = str(value or "").strip()
|
|
48
|
+
return text or default
|
|
49
|
+
|
|
50
|
+
|
|
51
|
+
@dataclass(frozen=True)
|
|
52
|
+
class RequestLayout:
|
|
53
|
+
schema_version: int = 1
|
|
54
|
+
input_mode: str = "full"
|
|
55
|
+
continuation_strategy: str = "full_replay"
|
|
56
|
+
cache_strategy: str = "none"
|
|
57
|
+
cache_mode: str = "manual"
|
|
58
|
+
message_count: int = 0
|
|
59
|
+
request_message_count: int = 0
|
|
60
|
+
tool_count: int = 0
|
|
61
|
+
stable_prefix_message_count: int = 0
|
|
62
|
+
dynamic_suffix_message_count: int = 0
|
|
63
|
+
provider_metadata_message_count: int = 0
|
|
64
|
+
stable_prefix_estimated_tokens: int = 0
|
|
65
|
+
dynamic_suffix_estimated_tokens: int = 0
|
|
66
|
+
tool_schema_tokens: int = 0
|
|
67
|
+
total_estimated_tokens: int = 0
|
|
68
|
+
serialized_request_estimate_tokens: int | None = None
|
|
69
|
+
sent_serialized_request_estimate_tokens: int | None = None
|
|
70
|
+
cacheable_prefix_hash: str = ""
|
|
71
|
+
request_messages_signature: str = ""
|
|
72
|
+
tool_schema_hash: str = ""
|
|
73
|
+
metadata: dict[str, Any] = field(default_factory=dict)
|
|
74
|
+
|
|
75
|
+
def to_payload(self) -> dict[str, Any]:
|
|
76
|
+
payload: dict[str, Any] = {
|
|
77
|
+
"schema_version": self.schema_version,
|
|
78
|
+
"input_mode": self.input_mode,
|
|
79
|
+
"continuation_strategy": self.continuation_strategy,
|
|
80
|
+
"cache_strategy": self.cache_strategy,
|
|
81
|
+
"cache_mode": self.cache_mode,
|
|
82
|
+
"message_count": self.message_count,
|
|
83
|
+
"request_message_count": self.request_message_count,
|
|
84
|
+
"tool_count": self.tool_count,
|
|
85
|
+
"stable_prefix_message_count": self.stable_prefix_message_count,
|
|
86
|
+
"dynamic_suffix_message_count": self.dynamic_suffix_message_count,
|
|
87
|
+
"provider_metadata_message_count": self.provider_metadata_message_count,
|
|
88
|
+
"stable_prefix_estimated_tokens": self.stable_prefix_estimated_tokens,
|
|
89
|
+
"dynamic_suffix_estimated_tokens": self.dynamic_suffix_estimated_tokens,
|
|
90
|
+
"tool_schema_tokens": self.tool_schema_tokens,
|
|
91
|
+
"total_estimated_tokens": self.total_estimated_tokens,
|
|
92
|
+
"cacheable_prefix_hash": self.cacheable_prefix_hash,
|
|
93
|
+
"request_messages_signature": self.request_messages_signature,
|
|
94
|
+
"tool_schema_hash": self.tool_schema_hash,
|
|
95
|
+
}
|
|
96
|
+
if self.serialized_request_estimate_tokens is not None:
|
|
97
|
+
payload["serialized_request_estimate_tokens"] = self.serialized_request_estimate_tokens
|
|
98
|
+
if self.sent_serialized_request_estimate_tokens is not None:
|
|
99
|
+
payload["sent_serialized_request_estimate_tokens"] = (
|
|
100
|
+
self.sent_serialized_request_estimate_tokens
|
|
101
|
+
)
|
|
102
|
+
for key, value in self.metadata.items():
|
|
103
|
+
if key in payload:
|
|
104
|
+
continue
|
|
105
|
+
if isinstance(value, bool):
|
|
106
|
+
payload[key] = value
|
|
107
|
+
continue
|
|
108
|
+
parsed_int = _safe_int(value)
|
|
109
|
+
if parsed_int is not None:
|
|
110
|
+
payload[key] = parsed_int
|
|
111
|
+
continue
|
|
112
|
+
if value is not None:
|
|
113
|
+
payload[key] = _safe_label(value)
|
|
114
|
+
return payload
|
|
115
|
+
|
|
116
|
+
@classmethod
|
|
117
|
+
def from_payload(cls, payload: Any) -> RequestLayout | None:
|
|
118
|
+
if not isinstance(payload, Mapping):
|
|
119
|
+
return None
|
|
120
|
+
|
|
121
|
+
metadata: dict[str, Any] = {}
|
|
122
|
+
known = set(cls.__dataclass_fields__)
|
|
123
|
+
for key, value in payload.items():
|
|
124
|
+
if key not in known:
|
|
125
|
+
metadata[str(key)] = copy.deepcopy(value)
|
|
126
|
+
|
|
127
|
+
return cls(
|
|
128
|
+
schema_version=_safe_int(payload.get("schema_version")) or 1,
|
|
129
|
+
input_mode=_safe_label(payload.get("input_mode"), "full"),
|
|
130
|
+
continuation_strategy=_safe_label(
|
|
131
|
+
payload.get("continuation_strategy"),
|
|
132
|
+
"full_replay",
|
|
133
|
+
),
|
|
134
|
+
cache_strategy=_safe_label(payload.get("cache_strategy"), "none"),
|
|
135
|
+
cache_mode=_safe_label(payload.get("cache_mode"), "manual"),
|
|
136
|
+
message_count=_safe_int(payload.get("message_count")) or 0,
|
|
137
|
+
request_message_count=_safe_int(payload.get("request_message_count")) or 0,
|
|
138
|
+
tool_count=_safe_int(payload.get("tool_count")) or 0,
|
|
139
|
+
stable_prefix_message_count=(
|
|
140
|
+
_safe_int(payload.get("stable_prefix_message_count")) or 0
|
|
141
|
+
),
|
|
142
|
+
dynamic_suffix_message_count=(
|
|
143
|
+
_safe_int(payload.get("dynamic_suffix_message_count")) or 0
|
|
144
|
+
),
|
|
145
|
+
provider_metadata_message_count=(
|
|
146
|
+
_safe_int(payload.get("provider_metadata_message_count")) or 0
|
|
147
|
+
),
|
|
148
|
+
stable_prefix_estimated_tokens=(
|
|
149
|
+
_safe_int(payload.get("stable_prefix_estimated_tokens")) or 0
|
|
150
|
+
),
|
|
151
|
+
dynamic_suffix_estimated_tokens=(
|
|
152
|
+
_safe_int(payload.get("dynamic_suffix_estimated_tokens")) or 0
|
|
153
|
+
),
|
|
154
|
+
tool_schema_tokens=_safe_int(payload.get("tool_schema_tokens")) or 0,
|
|
155
|
+
total_estimated_tokens=_safe_int(payload.get("total_estimated_tokens")) or 0,
|
|
156
|
+
serialized_request_estimate_tokens=_safe_int(
|
|
157
|
+
payload.get("serialized_request_estimate_tokens")
|
|
158
|
+
),
|
|
159
|
+
sent_serialized_request_estimate_tokens=_safe_int(
|
|
160
|
+
payload.get("sent_serialized_request_estimate_tokens")
|
|
161
|
+
),
|
|
162
|
+
cacheable_prefix_hash=_safe_label(payload.get("cacheable_prefix_hash")),
|
|
163
|
+
request_messages_signature=_safe_label(payload.get("request_messages_signature")),
|
|
164
|
+
tool_schema_hash=_safe_label(payload.get("tool_schema_hash")),
|
|
165
|
+
metadata=metadata,
|
|
166
|
+
)
|
|
167
|
+
|
|
168
|
+
|
|
169
|
+
def _provider_metadata_message_count(messages: list[dict[str, Any]]) -> int:
|
|
170
|
+
return sum(1 for message in messages if "_alysis_provider_metadata" in message)
|
|
171
|
+
|
|
172
|
+
|
|
173
|
+
def _policy_label(policy: Mapping[str, Any] | None, key: str) -> str | None:
|
|
174
|
+
if not isinstance(policy, Mapping):
|
|
175
|
+
return None
|
|
176
|
+
value = policy.get(key)
|
|
177
|
+
return str(value).strip() if value is not None else None
|
|
178
|
+
|
|
179
|
+
|
|
180
|
+
def _merge_metadata(*items: Mapping[str, Any] | None) -> dict[str, Any]:
|
|
181
|
+
merged: dict[str, Any] = {}
|
|
182
|
+
for item in items:
|
|
183
|
+
if not isinstance(item, Mapping):
|
|
184
|
+
continue
|
|
185
|
+
for key, value in item.items():
|
|
186
|
+
merged[str(key)] = copy.deepcopy(value)
|
|
187
|
+
return merged
|
|
188
|
+
|
|
189
|
+
|
|
190
|
+
@dataclass(frozen=True)
|
|
191
|
+
class RequestCachePlan:
|
|
192
|
+
strategy: str = "none"
|
|
193
|
+
mode: str = "manual"
|
|
194
|
+
prompt_cache_key: str | None = None
|
|
195
|
+
prompt_cache_retention: str | None = None
|
|
196
|
+
anthropic_cache_control_enabled: bool = False
|
|
197
|
+
anthropic_cache_control_ttl: str = "5m"
|
|
198
|
+
|
|
199
|
+
def __post_init__(self) -> None:
|
|
200
|
+
mode = str(self.mode or "manual").strip().lower()
|
|
201
|
+
if mode not in {"manual", "automatic", "auto", "off"}:
|
|
202
|
+
mode = "manual"
|
|
203
|
+
if mode == "auto":
|
|
204
|
+
mode = "automatic"
|
|
205
|
+
ttl = str(self.anthropic_cache_control_ttl or "5m").strip().lower()
|
|
206
|
+
if ttl not in {"5m", "1h"}:
|
|
207
|
+
ttl = "5m"
|
|
208
|
+
object.__setattr__(self, "strategy", str(self.strategy or "none").strip() or "none")
|
|
209
|
+
object.__setattr__(self, "mode", mode)
|
|
210
|
+
object.__setattr__(
|
|
211
|
+
self,
|
|
212
|
+
"prompt_cache_key",
|
|
213
|
+
None if mode == "off" else str(self.prompt_cache_key or "").strip() or None,
|
|
214
|
+
)
|
|
215
|
+
object.__setattr__(
|
|
216
|
+
self,
|
|
217
|
+
"prompt_cache_retention",
|
|
218
|
+
None if mode == "off" else str(self.prompt_cache_retention or "").strip() or None,
|
|
219
|
+
)
|
|
220
|
+
object.__setattr__(
|
|
221
|
+
self,
|
|
222
|
+
"anthropic_cache_control_enabled",
|
|
223
|
+
False if mode == "off" else bool(self.anthropic_cache_control_enabled),
|
|
224
|
+
)
|
|
225
|
+
object.__setattr__(self, "anthropic_cache_control_ttl", ttl)
|
|
226
|
+
|
|
227
|
+
def anthropic_cache_control_payload(self) -> dict[str, str] | None:
|
|
228
|
+
if not self.anthropic_cache_control_enabled:
|
|
229
|
+
return None
|
|
230
|
+
payload = {"type": "ephemeral"}
|
|
231
|
+
if self.anthropic_cache_control_ttl == "1h":
|
|
232
|
+
payload["ttl"] = "1h"
|
|
233
|
+
return payload
|
|
234
|
+
|
|
235
|
+
def anthropic_cache_policy_metadata(self) -> dict[str, Any] | None:
|
|
236
|
+
if not self.anthropic_cache_control_enabled:
|
|
237
|
+
return None
|
|
238
|
+
return {
|
|
239
|
+
"strategy": "anthropic_cache_control",
|
|
240
|
+
"enabled": True,
|
|
241
|
+
"ttl": self.anthropic_cache_control_ttl,
|
|
242
|
+
"mode": self.mode if self.mode != "off" else "manual",
|
|
243
|
+
}
|
|
244
|
+
|
|
245
|
+
def openai_prompt_cache_policy_metadata(self) -> dict[str, Any] | None:
|
|
246
|
+
if (
|
|
247
|
+
self.mode == "off"
|
|
248
|
+
or self.strategy != "openai_prompt_cache"
|
|
249
|
+
or (not self.prompt_cache_key and not self.prompt_cache_retention)
|
|
250
|
+
):
|
|
251
|
+
return None
|
|
252
|
+
metadata: dict[str, Any] = {
|
|
253
|
+
"strategy": "openai_prompt_cache",
|
|
254
|
+
"enabled": True,
|
|
255
|
+
"mode": self.mode if self.mode != "off" else "manual",
|
|
256
|
+
}
|
|
257
|
+
if self.prompt_cache_retention:
|
|
258
|
+
metadata["retention"] = self.prompt_cache_retention
|
|
259
|
+
return metadata
|
|
260
|
+
|
|
261
|
+
|
|
262
|
+
@dataclass(frozen=True)
|
|
263
|
+
class LLMRequestPlan:
|
|
264
|
+
messages: tuple[dict[str, Any], ...]
|
|
265
|
+
tools: tuple[dict[str, Any], ...] = ()
|
|
266
|
+
tool_choice: Any | None = None
|
|
267
|
+
response_format: dict[str, Any] | None = None
|
|
268
|
+
stream: bool = False
|
|
269
|
+
temperature: float | None = None
|
|
270
|
+
max_tokens: int | None = None
|
|
271
|
+
cache: RequestCachePlan = field(default_factory=RequestCachePlan)
|
|
272
|
+
metadata: dict[str, Any] = field(default_factory=dict)
|
|
273
|
+
|
|
274
|
+
@classmethod
|
|
275
|
+
def from_chat_args(
|
|
276
|
+
cls,
|
|
277
|
+
*,
|
|
278
|
+
messages: list[dict[str, Any]] | tuple[dict[str, Any], ...] | None,
|
|
279
|
+
tools: list[dict[str, Any]] | tuple[dict[str, Any], ...] | None = None,
|
|
280
|
+
tool_choice: Any | None = None,
|
|
281
|
+
response_format: dict[str, Any] | None = None,
|
|
282
|
+
stream: bool = False,
|
|
283
|
+
temperature: float | None = None,
|
|
284
|
+
max_tokens: int | None = None,
|
|
285
|
+
cache: RequestCachePlan | None = None,
|
|
286
|
+
metadata: dict[str, Any] | None = None,
|
|
287
|
+
) -> LLMRequestPlan:
|
|
288
|
+
return cls(
|
|
289
|
+
messages=tuple(
|
|
290
|
+
copy.deepcopy(message) for message in (messages or []) if isinstance(message, dict)
|
|
291
|
+
),
|
|
292
|
+
tools=tuple(copy.deepcopy(tool) for tool in (tools or []) if isinstance(tool, dict)),
|
|
293
|
+
tool_choice=copy.deepcopy(tool_choice),
|
|
294
|
+
response_format=(
|
|
295
|
+
copy.deepcopy(response_format) if isinstance(response_format, dict) else None
|
|
296
|
+
),
|
|
297
|
+
stream=bool(stream),
|
|
298
|
+
temperature=temperature,
|
|
299
|
+
max_tokens=max_tokens,
|
|
300
|
+
cache=cache or RequestCachePlan(),
|
|
301
|
+
metadata=copy.deepcopy(metadata or {}),
|
|
302
|
+
)
|
|
303
|
+
|
|
304
|
+
def message_list(self) -> list[dict[str, Any]]:
|
|
305
|
+
return [copy.deepcopy(message) for message in self.messages]
|
|
306
|
+
|
|
307
|
+
def tool_list(self) -> list[dict[str, Any]] | None:
|
|
308
|
+
tools = [copy.deepcopy(tool) for tool in self.tools]
|
|
309
|
+
return tools or None
|
|
310
|
+
|
|
311
|
+
def with_cache(self, cache: RequestCachePlan) -> LLMRequestPlan:
|
|
312
|
+
return replace(self, cache=cache)
|
|
313
|
+
|
|
314
|
+
def layout(
|
|
315
|
+
self,
|
|
316
|
+
*,
|
|
317
|
+
input_mode: str = "full",
|
|
318
|
+
continuation_strategy: str | None = None,
|
|
319
|
+
provider_payload: Mapping[str, Any] | None = None,
|
|
320
|
+
sent_provider_payload: Mapping[str, Any] | None = None,
|
|
321
|
+
cache_policy_metadata: Mapping[str, Any] | None = None,
|
|
322
|
+
extra: Mapping[str, Any] | None = None,
|
|
323
|
+
) -> RequestLayout:
|
|
324
|
+
messages = self.message_list()
|
|
325
|
+
tools = self.tool_list() or []
|
|
326
|
+
sanitized_messages = sanitize_messages_for_estimation(messages)
|
|
327
|
+
stable_prefix_count = min(
|
|
328
|
+
cacheable_prefix_message_count(sanitized_messages),
|
|
329
|
+
len(sanitized_messages),
|
|
330
|
+
)
|
|
331
|
+
stable_prefix = sanitized_messages[:stable_prefix_count]
|
|
332
|
+
dynamic_suffix = sanitized_messages[stable_prefix_count:]
|
|
333
|
+
tool_schema_tokens = estimate_tool_schema_tokens(tools)
|
|
334
|
+
stable_prefix_tokens = estimate_message_tokens(stable_prefix)
|
|
335
|
+
dynamic_suffix_tokens = estimate_message_tokens(dynamic_suffix)
|
|
336
|
+
breakdown = estimate_request_token_breakdown(
|
|
337
|
+
messages=messages,
|
|
338
|
+
tool_list=tools,
|
|
339
|
+
pinned_prefix_len=stable_prefix_count,
|
|
340
|
+
)
|
|
341
|
+
policy_cache_strategy = _policy_label(cache_policy_metadata, "strategy")
|
|
342
|
+
policy_cache_mode = _policy_label(cache_policy_metadata, "mode")
|
|
343
|
+
metadata = _merge_metadata(self.metadata, extra)
|
|
344
|
+
return RequestLayout(
|
|
345
|
+
input_mode=_safe_label(input_mode, "full"),
|
|
346
|
+
continuation_strategy=_safe_label(
|
|
347
|
+
continuation_strategy,
|
|
348
|
+
"previous_response_id" if input_mode == "previous_response_id" else "full_replay",
|
|
349
|
+
),
|
|
350
|
+
cache_strategy=_safe_label(policy_cache_strategy, self.cache.strategy),
|
|
351
|
+
cache_mode=_safe_label(policy_cache_mode, self.cache.mode),
|
|
352
|
+
message_count=len(sanitized_messages),
|
|
353
|
+
request_message_count=len(sanitized_messages),
|
|
354
|
+
tool_count=len(tools),
|
|
355
|
+
stable_prefix_message_count=stable_prefix_count,
|
|
356
|
+
dynamic_suffix_message_count=len(dynamic_suffix),
|
|
357
|
+
provider_metadata_message_count=_provider_metadata_message_count(messages),
|
|
358
|
+
stable_prefix_estimated_tokens=stable_prefix_tokens,
|
|
359
|
+
dynamic_suffix_estimated_tokens=dynamic_suffix_tokens,
|
|
360
|
+
tool_schema_tokens=tool_schema_tokens,
|
|
361
|
+
total_estimated_tokens=breakdown.total_tokens,
|
|
362
|
+
serialized_request_estimate_tokens=_optional_payload_token_estimate(provider_payload),
|
|
363
|
+
sent_serialized_request_estimate_tokens=_optional_payload_token_estimate(
|
|
364
|
+
sent_provider_payload
|
|
365
|
+
),
|
|
366
|
+
cacheable_prefix_hash=_stable_payload_hash(stable_prefix) if stable_prefix else "",
|
|
367
|
+
request_messages_signature=_stable_payload_hash(sanitized_messages),
|
|
368
|
+
tool_schema_hash=_stable_payload_hash(tools) if tools else "",
|
|
369
|
+
metadata=metadata,
|
|
370
|
+
)
|
|
371
|
+
|
|
372
|
+
def request_plan_metadata(
|
|
373
|
+
self,
|
|
374
|
+
*,
|
|
375
|
+
input_mode: str = "full",
|
|
376
|
+
continuation_strategy: str | None = None,
|
|
377
|
+
provider_payload: Mapping[str, Any] | None = None,
|
|
378
|
+
sent_provider_payload: Mapping[str, Any] | None = None,
|
|
379
|
+
cache_policy_metadata: Mapping[str, Any] | None = None,
|
|
380
|
+
extra: Mapping[str, Any] | None = None,
|
|
381
|
+
) -> dict[str, Any]:
|
|
382
|
+
return self.layout(
|
|
383
|
+
input_mode=input_mode,
|
|
384
|
+
continuation_strategy=continuation_strategy,
|
|
385
|
+
provider_payload=provider_payload,
|
|
386
|
+
sent_provider_payload=sent_provider_payload,
|
|
387
|
+
cache_policy_metadata=cache_policy_metadata,
|
|
388
|
+
extra=extra,
|
|
389
|
+
).to_payload()
|
|
@@ -0,0 +1,238 @@
|
|
|
1
|
+
from __future__ import annotations
|
|
2
|
+
|
|
3
|
+
from collections.abc import Mapping
|
|
4
|
+
from typing import Any
|
|
5
|
+
|
|
6
|
+
from ..request_estimation import (
|
|
7
|
+
estimate_message_tokens,
|
|
8
|
+
estimate_request_token_breakdown,
|
|
9
|
+
estimate_tool_schema_tokens,
|
|
10
|
+
)
|
|
11
|
+
from .cache_capabilities import (
|
|
12
|
+
CACHE_CONTROL_FIELD,
|
|
13
|
+
CACHED_CONTENT_FIELD,
|
|
14
|
+
OPENROUTER_SESSION_ID_FIELD,
|
|
15
|
+
OPENROUTER_SESSION_ID_HEADER_FIELD,
|
|
16
|
+
PROMPT_CACHE_KEY_FIELD,
|
|
17
|
+
XAI_CONVERSATION_ID_HEADER_FIELD,
|
|
18
|
+
)
|
|
19
|
+
from .cache_control_blocks import (
|
|
20
|
+
cacheable_prefix_message_count,
|
|
21
|
+
count_explicit_cache_control_blocks,
|
|
22
|
+
)
|
|
23
|
+
|
|
24
|
+
_AFFINITY_CACHE_FIELDS = frozenset(
|
|
25
|
+
{
|
|
26
|
+
PROMPT_CACHE_KEY_FIELD,
|
|
27
|
+
OPENROUTER_SESSION_ID_FIELD,
|
|
28
|
+
OPENROUTER_SESSION_ID_HEADER_FIELD,
|
|
29
|
+
XAI_CONVERSATION_ID_HEADER_FIELD,
|
|
30
|
+
}
|
|
31
|
+
)
|
|
32
|
+
|
|
33
|
+
|
|
34
|
+
def build_request_shape_report(
|
|
35
|
+
*,
|
|
36
|
+
messages: list[dict[str, Any]],
|
|
37
|
+
tools: list[dict[str, Any]] | None,
|
|
38
|
+
cache_policy: Mapping[str, Any] | None,
|
|
39
|
+
provider_payload: Mapping[str, Any] | None = None,
|
|
40
|
+
input_mode: str = "full",
|
|
41
|
+
) -> dict[str, Any]:
|
|
42
|
+
clean_messages = [message for message in messages if isinstance(message, dict)]
|
|
43
|
+
clean_tools = [tool for tool in tools or [] if isinstance(tool, dict)]
|
|
44
|
+
prefix_count = cacheable_prefix_message_count(clean_messages)
|
|
45
|
+
prefix_messages = clean_messages[:prefix_count]
|
|
46
|
+
prefix_tokens = estimate_message_tokens(prefix_messages)
|
|
47
|
+
tool_schema_tokens = estimate_tool_schema_tokens(clean_tools)
|
|
48
|
+
breakdown = estimate_request_token_breakdown(
|
|
49
|
+
messages=clean_messages,
|
|
50
|
+
tool_list=clean_tools,
|
|
51
|
+
pinned_prefix_len=prefix_count,
|
|
52
|
+
)
|
|
53
|
+
total_tokens = breakdown.total_tokens
|
|
54
|
+
min_tokens = _policy_min_tokens(cache_policy)
|
|
55
|
+
emitted_fields = _policy_list(cache_policy, "emitted_fields")
|
|
56
|
+
cache_enabled = (
|
|
57
|
+
bool(cache_policy.get("enabled")) if isinstance(cache_policy, Mapping) else False
|
|
58
|
+
)
|
|
59
|
+
explicit_cache_control_blocks = count_explicit_cache_control_blocks(
|
|
60
|
+
provider_payload or clean_messages
|
|
61
|
+
)
|
|
62
|
+
top_level_cache_control_present = (
|
|
63
|
+
isinstance(provider_payload, Mapping) and CACHE_CONTROL_FIELD in provider_payload
|
|
64
|
+
)
|
|
65
|
+
cached_content_attached = _cached_content_attached(provider_payload)
|
|
66
|
+
affinity_field_emitted = any(field in _AFFINITY_CACHE_FIELDS for field in emitted_fields)
|
|
67
|
+
cache_material_attached = (
|
|
68
|
+
top_level_cache_control_present
|
|
69
|
+
or explicit_cache_control_blocks > 0
|
|
70
|
+
or cached_content_attached
|
|
71
|
+
)
|
|
72
|
+
cache_fields_emitted = bool(cache_material_attached or affinity_field_emitted)
|
|
73
|
+
cache_eligible = _cache_eligible(
|
|
74
|
+
cache_enabled=cache_enabled,
|
|
75
|
+
emitted_fields=emitted_fields,
|
|
76
|
+
prefix_tokens=prefix_tokens,
|
|
77
|
+
min_tokens=min_tokens,
|
|
78
|
+
)
|
|
79
|
+
token_breakdown = breakdown.to_payload()
|
|
80
|
+
token_breakdown.pop("tool_schema_budget", None)
|
|
81
|
+
return {
|
|
82
|
+
"schema_version": 1,
|
|
83
|
+
"input_mode": str(input_mode or "full").strip() or "full",
|
|
84
|
+
"message_count": len(clean_messages),
|
|
85
|
+
"tool_count": len(clean_tools),
|
|
86
|
+
"system_message_count": _role_count(clean_messages, "system"),
|
|
87
|
+
"developer_message_count": _role_count(clean_messages, "developer"),
|
|
88
|
+
"user_message_count": _role_count(clean_messages, "user"),
|
|
89
|
+
"assistant_message_count": _role_count(clean_messages, "assistant"),
|
|
90
|
+
"tool_message_count": _role_count(clean_messages, "tool"),
|
|
91
|
+
"tool_call_message_count": sum(
|
|
92
|
+
1 for message in clean_messages if bool(message.get("tool_calls"))
|
|
93
|
+
),
|
|
94
|
+
"content_block_count": sum(
|
|
95
|
+
_content_block_count(message.get("content")) for message in clean_messages
|
|
96
|
+
),
|
|
97
|
+
"cache_enabled": cache_enabled,
|
|
98
|
+
"cache_eligible": cache_eligible,
|
|
99
|
+
"cache_used": bool(cache_material_attached),
|
|
100
|
+
"cache_fields_emitted": cache_fields_emitted,
|
|
101
|
+
"top_level_cache_control_present": top_level_cache_control_present,
|
|
102
|
+
"cached_content_attached": cached_content_attached,
|
|
103
|
+
"affinity_field_emitted": affinity_field_emitted,
|
|
104
|
+
"cache_strategy": _policy_label(cache_policy, "strategy"),
|
|
105
|
+
"cache_status": _policy_label(cache_policy, "status"),
|
|
106
|
+
"emitted_cache_fields": emitted_fields,
|
|
107
|
+
"cache_control_block_count": explicit_cache_control_blocks,
|
|
108
|
+
"explicit_cache_control_block_count": explicit_cache_control_blocks,
|
|
109
|
+
"cacheable_prefix_present": prefix_count > 0,
|
|
110
|
+
"cacheable_prefix_message_count": prefix_count,
|
|
111
|
+
"cacheable_prefix_estimated_tokens": prefix_tokens,
|
|
112
|
+
"cacheable_surface_estimated_tokens": prefix_tokens + tool_schema_tokens,
|
|
113
|
+
"min_cacheable_tokens": min_tokens,
|
|
114
|
+
"total_estimated_tokens": total_tokens,
|
|
115
|
+
"tool_schema_share": _ratio(tool_schema_tokens, total_tokens),
|
|
116
|
+
"inline_tool_transcript_share": _ratio(
|
|
117
|
+
breakdown.inline_tool_transcript_tokens,
|
|
118
|
+
total_tokens,
|
|
119
|
+
),
|
|
120
|
+
"token_breakdown": token_breakdown,
|
|
121
|
+
"risk_reasons": _risk_reasons(
|
|
122
|
+
cache_enabled=cache_enabled,
|
|
123
|
+
cache_eligible=cache_eligible,
|
|
124
|
+
emitted_fields=emitted_fields,
|
|
125
|
+
explicit_cache_control_blocks=explicit_cache_control_blocks,
|
|
126
|
+
top_level_cache_control_present=top_level_cache_control_present,
|
|
127
|
+
cached_content_attached=cached_content_attached,
|
|
128
|
+
prefix_count=prefix_count,
|
|
129
|
+
prefix_tokens=prefix_tokens,
|
|
130
|
+
min_tokens=min_tokens,
|
|
131
|
+
breakdown=token_breakdown,
|
|
132
|
+
total_tokens=total_tokens,
|
|
133
|
+
),
|
|
134
|
+
}
|
|
135
|
+
|
|
136
|
+
|
|
137
|
+
def _policy_label(policy: Mapping[str, Any] | None, key: str) -> str:
|
|
138
|
+
if not isinstance(policy, Mapping):
|
|
139
|
+
return ""
|
|
140
|
+
return str(policy.get(key) or "").strip()
|
|
141
|
+
|
|
142
|
+
|
|
143
|
+
def _policy_list(policy: Mapping[str, Any] | None, key: str) -> list[str]:
|
|
144
|
+
if not isinstance(policy, Mapping):
|
|
145
|
+
return []
|
|
146
|
+
raw = policy.get(key)
|
|
147
|
+
if not isinstance(raw, (list, tuple)):
|
|
148
|
+
return []
|
|
149
|
+
return [str(item).strip() for item in raw if str(item).strip()]
|
|
150
|
+
|
|
151
|
+
|
|
152
|
+
def _policy_min_tokens(policy: Mapping[str, Any] | None) -> int:
|
|
153
|
+
if not isinstance(policy, Mapping):
|
|
154
|
+
return 0
|
|
155
|
+
for key in ("min_tokens", "min_cacheable_tokens"):
|
|
156
|
+
try:
|
|
157
|
+
value = int(policy.get(key))
|
|
158
|
+
except (TypeError, ValueError):
|
|
159
|
+
continue
|
|
160
|
+
if value >= 0:
|
|
161
|
+
return value
|
|
162
|
+
return 0
|
|
163
|
+
|
|
164
|
+
|
|
165
|
+
def _cached_content_attached(provider_payload: Mapping[str, Any] | None) -> bool:
|
|
166
|
+
if not isinstance(provider_payload, Mapping):
|
|
167
|
+
return False
|
|
168
|
+
return "cachedContent" in provider_payload or CACHED_CONTENT_FIELD in provider_payload
|
|
169
|
+
|
|
170
|
+
|
|
171
|
+
def _cache_eligible(
|
|
172
|
+
*,
|
|
173
|
+
cache_enabled: bool,
|
|
174
|
+
emitted_fields: list[str],
|
|
175
|
+
prefix_tokens: int,
|
|
176
|
+
min_tokens: int,
|
|
177
|
+
) -> bool:
|
|
178
|
+
if not cache_enabled:
|
|
179
|
+
return False
|
|
180
|
+
if CACHE_CONTROL_FIELD in emitted_fields or CACHED_CONTENT_FIELD in emitted_fields:
|
|
181
|
+
return prefix_tokens >= min_tokens and prefix_tokens > 0
|
|
182
|
+
return True
|
|
183
|
+
|
|
184
|
+
|
|
185
|
+
def _role_count(messages: list[dict[str, Any]], role: str) -> int:
|
|
186
|
+
return sum(1 for message in messages if str(message.get("role") or "").strip().lower() == role)
|
|
187
|
+
|
|
188
|
+
|
|
189
|
+
def _content_block_count(content: Any) -> int:
|
|
190
|
+
if isinstance(content, list):
|
|
191
|
+
return len(content)
|
|
192
|
+
return 1 if content is not None else 0
|
|
193
|
+
|
|
194
|
+
|
|
195
|
+
def _ratio(numerator: int, denominator: int) -> float:
|
|
196
|
+
if denominator <= 0:
|
|
197
|
+
return 0.0
|
|
198
|
+
return round(max(0, numerator) / denominator, 4)
|
|
199
|
+
|
|
200
|
+
|
|
201
|
+
def _risk_reasons(
|
|
202
|
+
*,
|
|
203
|
+
cache_enabled: bool,
|
|
204
|
+
cache_eligible: bool,
|
|
205
|
+
emitted_fields: list[str],
|
|
206
|
+
explicit_cache_control_blocks: int,
|
|
207
|
+
top_level_cache_control_present: bool,
|
|
208
|
+
cached_content_attached: bool,
|
|
209
|
+
prefix_count: int,
|
|
210
|
+
prefix_tokens: int,
|
|
211
|
+
min_tokens: int,
|
|
212
|
+
breakdown: Mapping[str, Any],
|
|
213
|
+
total_tokens: int,
|
|
214
|
+
) -> list[str]:
|
|
215
|
+
reasons: list[str] = []
|
|
216
|
+
if not cache_enabled:
|
|
217
|
+
reasons.append("cache_disabled_or_unavailable")
|
|
218
|
+
if cache_enabled and not emitted_fields:
|
|
219
|
+
reasons.append("no_emitted_cache_fields")
|
|
220
|
+
if (
|
|
221
|
+
CACHE_CONTROL_FIELD in emitted_fields
|
|
222
|
+
and explicit_cache_control_blocks <= 0
|
|
223
|
+
and not top_level_cache_control_present
|
|
224
|
+
):
|
|
225
|
+
reasons.append("cache_control_blocks_absent")
|
|
226
|
+
if CACHED_CONTENT_FIELD in emitted_fields and cache_eligible and not cached_content_attached:
|
|
227
|
+
reasons.append("cached_content_absent")
|
|
228
|
+
if prefix_count <= 0:
|
|
229
|
+
reasons.append("no_cacheable_prefix")
|
|
230
|
+
if prefix_tokens < min_tokens:
|
|
231
|
+
reasons.append("prefix_below_min_tokens")
|
|
232
|
+
if cache_enabled and not cache_eligible:
|
|
233
|
+
reasons.append("cache_not_eligible")
|
|
234
|
+
if _ratio(int(breakdown.get("tool_schema_tokens") or 0), total_tokens) >= 0.25:
|
|
235
|
+
reasons.append("large_tool_schema_share")
|
|
236
|
+
if _ratio(int(breakdown.get("inline_tool_transcript_tokens") or 0), total_tokens) >= 0.25:
|
|
237
|
+
reasons.append("large_inline_tool_transcript_share")
|
|
238
|
+
return list(dict.fromkeys(reasons))
|