CoreRasurae a8f68b8262 fix(graph): enforce USD budget at each agent invocation with pre-flight gate ...

Enforces the USD budget (max_cost_usd) as a hard pre-flight gate before
every node execution and at each main-agent ainvoke round inside the
multi-turn tool loop, with pruning passes silently skipped when the
budget is exhausted.

Pre-flight gate (pure_graph.py):
  _check_budget_pre_flight() raises ExecutionError(kind='cost',
  reason='budget_exhausted') immediately before node execution in both
  _execute_from_node and all three branches of _stream_from_node.

In-node budget gating (llm.py):
  _check_budget_before_invoke() gates every _retry_ainvoke() call
  inside _execute_tool_loop, process_message, and stream_message.
  _should_skip_pruning() skips pruning passes when budget exhausted.
  _accumulate_cost_after_invoke() computes USD cost from token counts
  using the pricing table and advances current_accumulated_cost so
  subsequent budget checks see the updated cost.

ContextVar plumbing (retry.py, pure_graph.py):
  current_accumulated_cost, current_max_cost_usd, and current_pricing
  ContextVars carry per-node budget state from PureLangGraph into
  LLMAgent. Set before node.execute() via _set_budget_context_for_node,
  reset in finally blocks via _reset_budget_context.

Reviewer fixes (rui.hu #80):
  - Move cost accumulation inside try block in terminal streaming branch
    to prevent ContextVar leak (was set after finally reset).
  - Standardize log precision to $%.6f across all budget messages.
  - Remove orphaned 'already exhausted' step (criterion unreachable).
  - Add dedicated streaming budget enforcement scenario.
  - Remove redundant = None default on _reset_budget_context pricing_token.

ISSUES CLOSED: #76

2026-07-12 19:51:39 +01:00

..

__init__.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

_coverage_utils.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

actor_result_token_counting_steps.py

test(agents): disable retry in error-path tests to avoid unnecessary delays

2026-07-03 16:00:20 +01:00

agent_modules_steps.py

feat(credentials): refactor LLMAgent/AgentFactory for per-request credential injection and extended provider routing

2026-06-08 11:10:46 +00:00

agent_templates_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

agents_base_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

agents_base_final_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

application_core_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

application_coverage_gaps_steps.py

feat(agents): add tool_agent_class parameter to AgentFactory and create_executor

2026-07-08 09:23:34 +00:00

application_direct_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

application_interactive_coverage_steps.py

feat(ActorResult): implement ActorResult and NodeUsage types; capture per-node token counts from LangChain responses

2026-06-11 03:05:20 +00:00

application_missing_lines_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

application_utilities_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

bridge_coverage_gaps_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

bridge_final_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

bridge_remaining_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

bridge_source_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

cache_coverage_steps.py

test(coverage): add coverage scenarios for pre-existing code paths

2026-06-23 10:49:16 +01:00

chain_agent_comprehensive_steps.py

fix: resolve all bandit, semgrep, and vulture security findings

2026-05-26 23:12:05 +00:00

complex_template_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

composite_agent_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

config_core_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

config_parser_comprehensive_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

config_parser_coverage_gaps_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

config_parser_missing_lines_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

config_specific_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

config_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

context_manager_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

coverage_final_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

coverage_gaps_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

coverage_gaps_v2_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

credential_agent_state_steps.py

feat(credentials): refactor LLMAgent/AgentFactory for per-request credential injection and extended provider routing

2026-06-08 11:10:46 +00:00

credential_cleanup_steps.py

fix(llmagent): do not close shared cached httpx clients in cleanup()

2026-06-17 17:19:44 +00:00

credential_composite_steps.py

feat(credentials): refactor LLMAgent/AgentFactory for per-request credential injection and extended provider routing

2026-06-08 11:10:46 +00:00

credential_domain_warning_steps.py

feat(credentials): refactor LLMAgent/AgentFactory for per-request credential injection and extended provider routing

2026-06-08 11:10:46 +00:00

credential_envvar_steps.py

feat(credentials): refactor LLMAgent/AgentFactory for per-request credential injection and extended provider routing

2026-06-08 11:10:46 +00:00

credential_executor_paths_steps.py

feat(agents): add tool_agent_class parameter to AgentFactory and create_executor

2026-07-08 09:23:34 +00:00

credential_executor_steps.py

feat(ActorResult): implement ActorResult and NodeUsage types; capture per-node token counts from LangChain responses

2026-06-11 03:05:20 +00:00

credential_executor_validation_steps.py

feat(create_executor): implement create_executor() factory and Executor.execute() returning ActorResult

2026-06-10 09:41:39 +00:00

credential_factory_steps.py

feat(create_executor): implement create_executor() factory and Executor.execute() returning ActorResult

2026-06-10 09:41:39 +00:00

credential_graph_cleanup_steps.py

feat(ActorResult): implement ActorResult and NodeUsage types; capture per-node token counts from LangChain responses

2026-06-11 03:05:20 +00:00

credential_graph_error_steps.py

feat(credentials): refactor LLMAgent/AgentFactory for per-request credential injection and extended provider routing

2026-06-08 11:10:46 +00:00

credential_injection_steps.py

feat(credentials): refactor LLMAgent/AgentFactory for per-request credential injection and extended provider routing

2026-06-08 11:10:46 +00:00

credential_llm_error_steps.py

test(agents): disable retry in error-path tests to avoid unnecessary delays

2026-07-03 16:00:20 +01:00

credential_native_provider_steps.py

feat(credentials): refactor LLMAgent/AgentFactory for per-request credential injection and extended provider routing

2026-06-08 11:10:46 +00:00

credential_ssrf_steps.py

feat(credentials): refactor LLMAgent/AgentFactory for per-request credential injection and extended provider routing

2026-06-08 11:10:46 +00:00

decorators_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

deferred_template_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

dynamic_router_coverage_steps.py

test(coverage): add BDD unit tests to close coverage gap toward 97%

2026-06-09 16:06:32 +01:00

enhanced_registry_dedicated_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

error_verbosity_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

execute_stream_steps.py

fix(graph): enforce USD budget at each agent invocation with pre-flight gate

2026-07-12 19:51:39 +01:00

execution_limits_steps.py

feat(execution-limits): add structured ExecutionError kind/reason fields; enforce all 5 execution limits in PureLangGraph

2026-06-11 09:37:37 +00:00

executor_error_steps.py

feat(create_executor): implement create_executor() factory and Executor.execute() returning ActorResult

2026-06-10 09:41:39 +00:00

final_coverage_push_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

graph_templates_direct_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

graph_templates_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

graph_tool_node_execution_steps.py

feat(langgraph): implement real type:tool graph node execution via ToolAgent dispatch

2026-07-09 14:06:59 +00:00

inline_jinja_handler_direct_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

inline_yaml_jinja_complete_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

inline_yaml_jinja_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

inline_yaml_jinja_enhanced_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

isolated_unique_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

jinja_yaml_preprocessor_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

jinja_yaml_preprocessor_focused_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

langgraph_bridge_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

langgraph_bridge_direct_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

langgraph_graph_comprehensive_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

langgraph_graph_edge_cases_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

langgraph_graph_final_coverage_steps.py

fix: resolve all bandit, semgrep, and vulture security findings

2026-05-26 23:12:05 +00:00

langgraph_isolated_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

langgraph_nodes_optimized_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

langgraph_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

langgraph_visualization_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

llm_agent_config_validation_steps.py

test(unit): increase BDD coverage to 97% for registry, LLM, YAML template error branches

2026-06-15 23:13:46 +01:00

llm_agent_retry_steps.py

feat(agents): add LLM agent retry mechanisms with exponential backoff

2026-07-03 16:00:11 +01:00

llm_agent_steps.py

test(agents): disable retry in error-path tests to avoid unnecessary delays

2026-07-03 16:00:20 +01:00

llm_agent_stream_tool_calls_steps.py

feat(agents): add tool-call support to LLMAgent.stream_message

2026-07-02 07:47:39 +00:00

llm_agent_tool_calling_steps.py

fix(actors): llm token consumption undercounted

2026-07-01 14:55:58 +01:00

llm_agent_tool_loop_steps.py

fix(graph): enforce USD budget at each agent invocation with pre-flight gate

2026-07-12 19:51:39 +01:00

llm_cleanup_shared_client_steps.py

fix(llmagent): do not close shared cached httpx clients in cleanup()

2026-06-17 17:19:44 +00:00

llm_imports_coverage_steps.py

feat(credentials): refactor LLMAgent/AgentFactory for per-request credential injection and extended provider routing

2026-06-08 11:10:46 +00:00

llm_missing_coverage_steps.py

feat(agents): implement multi-turn tool-call loop and sandbox improvements

2026-06-23 12:39:20 +01:00

llm_provider_url_canonicalization_steps.py

test(unit): increase BDD coverage to 97% for registry, LLM, YAML template error branches

2026-06-15 23:13:46 +01:00

llm_system_prompt_context_rendering_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

llm_tools_coverage_steps.py

feat(agents): implement multi-turn tool-call loop and sandbox improvements

2026-06-23 12:39:20 +01:00

massive_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

merge_configs_steps.py

feat(merge_configs): expose public merge_configs(*dicts) API per §3.1 deep-merge algorithm

2026-06-03 14:44:18 +00:00

missing_steps.py

fix: resolve all bandit, semgrep, and vulture security findings

2026-05-26 23:12:05 +00:00

network_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

nodes_coverage_gaps_steps.py

fix(nodes): propagate ExecutionError from retry mechanism instead of swallowing it

2026-07-07 13:47:02 +01:00

nodes_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

progress_coverage_steps.py

test(coverage): add BDD unit tests to close coverage gap toward 97%

2026-06-09 16:06:32 +01:00

pure_graph_coverage_gaps_v2_steps.py

feat(execution-limits): add structured ExecutionError kind/reason fields; enforce all 5 execution limits in PureLangGraph

2026-06-11 09:37:37 +00:00

pure_graph_coverage_steps.py

feat(execution-limits): add structured ExecutionError kind/reason fields; enforce all 5 execution limits in PureLangGraph

2026-06-11 09:37:37 +00:00

reactive_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

registry_cache_steps.py

feat(registry): implement RegistryCache with LRU eviction and TTL

2026-06-11 19:26:04 +01:00

registry_canonicalization_steps.py

feat(registry): implement Canonicalizer with NFC normalization and SHA-1 hashing

2026-06-10 11:33:50 +01:00

registry_core_types_steps.py

feat(registry): add PackageType, PackageId, PackageReference, PackageContent core types

2026-06-06 19:21:22 +01:00

registry_exceptions_steps.py

feat(registry): implement RegistryError hierarchy with typed exceptions

2026-06-11 20:44:54 +01:00

registry_http_client_steps.py

feat(registry): implement RegistryError hierarchy with typed exceptions

2026-06-11 20:44:54 +01:00

registry_local_store_steps.py

feat(registry): implement local namespace reference resolution

2026-06-12 17:11:35 +01:00

registry_reference_resolver_coverage_steps.py

feat(registry): implement local namespace reference resolution

2026-06-12 17:11:35 +01:00

registry_reference_resolver_steps.py

feat(registry): implement ReferenceResolver with version alias resolution

2026-06-10 12:04:00 +01:00

registry_resolver_errors_steps.py

test(coverage): add coverage scenarios for pre-existing code paths

2026-06-23 10:49:16 +01:00

registry_steps.py

feat(registry): extend TemplateType and integrate PackageReference into template system

2026-06-10 12:50:38 +01:00

registry_template_integration_steps.py

feat(registry): implement local namespace reference resolution

2026-06-12 17:11:35 +01:00

road_to_97_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

route_bridge_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

route_coverage_comprehensive_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

route_graph_config_coverage_steps.py

test(coverage): add BDD unit tests to close coverage gap toward 97%

2026-06-09 16:06:32 +01:00

route_missing_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

routes_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

routing_prefix_stripping_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

runtime_config_steps.py

feat(create_executor): implement create_executor() factory and Executor.execute() returning ActorResult

2026-06-10 09:41:39 +00:00

runtime_coverage_gaps_steps.py

feat(ActorResult): implement ActorResult and NodeUsage types; capture per-node token counts from LangChain responses

2026-06-11 03:05:20 +00:00

runtime_coverage_steps.py

feat(agents): add tool_agent_class parameter to AgentFactory and create_executor

2026-07-08 09:23:34 +00:00

runtime_dispatch_coverage_steps.py

test(coverage): add coverage scenarios for pre-existing code paths

2026-06-23 10:49:16 +01:00

runtime_extended_coverage_steps.py

feat(agents): add tool_agent_class parameter to AgentFactory and create_executor

2026-07-08 09:23:34 +00:00

simple_loaders_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

smart_yaml_loader_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

state_comprehensive_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

state_missing_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

stderr_suppression_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

stream_router_coverage_gaps_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

stream_router_final_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

stream_templates_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

template_config_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

template_loaders_coverage_steps.py.disabled

style: apply end-of-file and whitespace fixes across feature files

2026-05-27 21:41:03 +00:00

template_renderer_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

template_store_coverage_steps.py

feat(registry): extend TemplateType and integrate PackageReference into template system

2026-06-10 12:50:38 +01:00

templates_base_remaining_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

templates_base_steps.py

test(coverage): add coverage scenarios for pre-existing code paths

2026-06-23 10:49:16 +01:00

templates_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

tool_agent_class_injection_steps.py

feat(agents): add tool_agent_class parameter to AgentFactory and create_executor

2026-07-08 09:23:34 +00:00

tool_agent_context_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00

tool_agent_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

tool_coverage_gaps_steps.py

feat(agents): implement multi-turn tool-call loop and sandbox improvements

2026-06-23 12:39:20 +01:00

unit_json_sanitization_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

unit_temperature_override_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

unit_tool_command_processing_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

validate_dict_agents_steps.py

feat(validate_dict): expose public validate_dict(config_dict, platform_limits) API

2026-06-05 09:23:44 +00:00

validate_dict_export_steps.py

feat(validate_dict): expose public validate_dict(config_dict, platform_limits) API

2026-06-05 09:23:44 +00:00

validate_dict_graph_route_steps.py

feat(validate_dict): expose public validate_dict(config_dict, platform_limits) API

2026-06-05 09:23:44 +00:00

validate_dict_helpers.py

feat(validate_dict): expose public validate_dict(config_dict, platform_limits) API

2026-06-05 09:23:44 +00:00

validate_dict_limits_steps.py

feat(validate_dict): expose public validate_dict(config_dict, platform_limits) API

2026-06-05 09:23:44 +00:00

validate_dict_routes_steps.py

feat(validate_dict): expose public validate_dict(config_dict, platform_limits) API

2026-06-05 09:23:44 +00:00

validate_dict_steps.py

feat(validate_dict): expose public validate_dict(config_dict, platform_limits) API

2026-06-05 09:23:44 +00:00

validate_dict_stream_steps.py

feat(validate_dict): expose public validate_dict(config_dict, platform_limits) API

2026-06-05 09:23:44 +00:00

validate_dict_subgraph_steps.py

feat(validate_dict): expose public validate_dict(config_dict, platform_limits) API

2026-06-05 09:23:44 +00:00

validation_actor_coverage_steps.py

test(coverage): add coverage scenarios for pre-existing code paths

2026-06-23 10:49:16 +01:00

verbose_logging_levels_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

yaml_jinja_loader_specific_steps.py

test(coverage): add coverage scenarios for pre-existing code paths

2026-06-23 10:49:16 +01:00

yaml_preprocessor_unique_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

yaml_template_comprehensive_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

yaml_template_coverage_gaps_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

yaml_template_missing_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

yaml_template_non_dict_errors_steps.py

test(unit): increase BDD coverage to 97% for registry, LLM, YAML template error branches

2026-06-15 23:13:46 +01:00

yaml_template_specific_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

yaml_template_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

yaml_template_targeted_coverage_steps.py

style: fix all lint errors across source and test files

2026-05-26 22:21:50 +00:00

yte_remaining_coverage_steps.py

test: add coverage gap tests improving coverage from 81.4% to 96.90%

2026-06-02 20:02:01 +01:00