mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-11-02 19:22:22 +01:00
validate executable artifact completion code
This commit is contained in:
+1
-1
@@ -1 +1 @@
|
|||||||
0.20.15
|
0.20.16
|
||||||
|
|||||||
@@ -1530,6 +1530,37 @@ def successful_required_artifact_mutation(tool_name, arguments, required_artifac
|
|||||||
return execution_targets_required_artifact(tool_name, arguments, targets)
|
return execution_targets_required_artifact(tool_name, arguments, targets)
|
||||||
|
|
||||||
|
|
||||||
|
def artifact_completion_python_code_error(arguments, required_artifacts):
|
||||||
|
"""Reject path-only or off-target Python during reserved completion.
|
||||||
|
|
||||||
|
JSON-Schema regex guidance on a free-form code string encourages structured
|
||||||
|
decoders to emit the shortest matching value (often the bare output path),
|
||||||
|
which is not executable Python. Validate syntax and the declared target
|
||||||
|
explicitly instead; post-execution artifact inspection remains the
|
||||||
|
authoritative proof that a non-empty output was actually created.
|
||||||
|
"""
|
||||||
|
targets = [str(path or '').strip().rstrip('/') for path in required_artifacts]
|
||||||
|
targets = [path for path in targets if path]
|
||||||
|
if len(targets) != 1:
|
||||||
|
return ''
|
||||||
|
code = str((arguments or {}).get('code') or '')
|
||||||
|
try:
|
||||||
|
compile(code, '<artifact-completion>', 'exec')
|
||||||
|
except (SyntaxError, TypeError, ValueError):
|
||||||
|
return (
|
||||||
|
'Artifact completion requires valid executable Python, not only a path '
|
||||||
|
'string. Provide complete Python code that creates the required output.'
|
||||||
|
)
|
||||||
|
target = targets[0]
|
||||||
|
if target not in code:
|
||||||
|
kind = 'output directory' if not Path(target).suffix else 'output file'
|
||||||
|
return (
|
||||||
|
f'Artifact completion Python must reference the required {kind} {target} '
|
||||||
|
'and create non-empty output there; do not only inspect or delete sources.'
|
||||||
|
)
|
||||||
|
return ''
|
||||||
|
|
||||||
|
|
||||||
def artifact_completion_tool_schemas(offered_schemas, required_artifacts):
|
def artifact_completion_tool_schemas(offered_schemas, required_artifacts):
|
||||||
"""Bind the sole artifact writer to the runner-declared output file.
|
"""Bind the sole artifact writer to the runner-declared output file.
|
||||||
|
|
||||||
@@ -1561,11 +1592,11 @@ def artifact_completion_tool_schemas(offered_schemas, required_artifacts):
|
|||||||
elif canonical(function.get('name')) == 'python':
|
elif canonical(function.get('name')) == 'python':
|
||||||
code_schema = properties.get('code')
|
code_schema = properties.get('code')
|
||||||
if isinstance(code_schema, dict):
|
if isinstance(code_schema, dict):
|
||||||
code_schema['pattern'] = re.escape(target + '/')
|
|
||||||
code_schema['description'] = (
|
code_schema['description'] = (
|
||||||
'Python code that creates one or more non-empty files inside '
|
'Complete executable Python that creates one or more non-empty '
|
||||||
f'the required directory {target}. The code must reference a '
|
f'files inside the required directory {target}; not only a path '
|
||||||
'descendant path; do not only inspect or delete source files.'
|
'string. Reference a descendant path and do not only inspect or '
|
||||||
|
'delete source files.'
|
||||||
)
|
)
|
||||||
return bound
|
return bound
|
||||||
if Path(target).suffix.lower() in _NON_TEXT_ARTIFACT_SUFFIXES:
|
if Path(target).suffix.lower() in _NON_TEXT_ARTIFACT_SUFFIXES:
|
||||||
@@ -1582,10 +1613,9 @@ def artifact_completion_tool_schemas(offered_schemas, required_artifacts):
|
|||||||
properties = (function.get('parameters') or {}).get('properties') or {}
|
properties = (function.get('parameters') or {}).get('properties') or {}
|
||||||
code_schema = properties.get('code')
|
code_schema = properties.get('code')
|
||||||
if isinstance(code_schema, dict):
|
if isinstance(code_schema, dict):
|
||||||
code_schema['pattern'] = re.escape(target)
|
|
||||||
code_schema['description'] = (
|
code_schema['description'] = (
|
||||||
'Python code that creates or updates this exact required '
|
'Complete executable Python that creates or updates this exact '
|
||||||
f'binary artifact path: {target}'
|
f'required binary artifact path, not only a path string: {target}'
|
||||||
)
|
)
|
||||||
bound.append(schema)
|
bound.append(schema)
|
||||||
return bound
|
return bound
|
||||||
@@ -5701,6 +5731,17 @@ async def stream_preview(*, endpoint_url, model, messages, headers, turn_contrac
|
|||||||
if schema is None or not turn_contract.permits(name):
|
if schema is None or not turn_contract.permits(name):
|
||||||
raise ValueError('Tool is not offered or permitted.')
|
raise ValueError('Tool is not offered or permitted.')
|
||||||
jsonschema.validate(args, schema['function']['parameters'])
|
jsonschema.validate(args, schema['function']['parameters'])
|
||||||
|
if (
|
||||||
|
artifact_write_phase
|
||||||
|
and canonical(name) == 'python'
|
||||||
|
and not successful_artifact_write
|
||||||
|
):
|
||||||
|
artifact_code_error = artifact_completion_python_code_error(
|
||||||
|
args, required_artifacts,
|
||||||
|
)
|
||||||
|
if artifact_code_error:
|
||||||
|
round_recovery_messages.append(artifact_code_error)
|
||||||
|
raise ValueError(artifact_code_error)
|
||||||
decision = evaluate_preview_call(
|
decision = evaluate_preview_call(
|
||||||
name, args, latest_user,
|
name, args, latest_user,
|
||||||
experiment_fixture_ids=experiment_fixture_ids,
|
experiment_fixture_ids=experiment_fixture_ids,
|
||||||
@@ -5892,6 +5933,22 @@ async def stream_preview(*, endpoint_url, model, messages, headers, turn_contrac
|
|||||||
result.get('error')
|
result.get('error')
|
||||||
or result.get('exit_code') not in (None, 0)
|
or result.get('exit_code') not in (None, 0)
|
||||||
)
|
)
|
||||||
|
if (
|
||||||
|
artifact_write_phase
|
||||||
|
and not failed
|
||||||
|
and canonical(actual_tool) == 'python'
|
||||||
|
and required_artifacts
|
||||||
|
and not successful_artifact_write
|
||||||
|
):
|
||||||
|
recovery = (
|
||||||
|
'The Python call ran but did not create a non-empty file at the '
|
||||||
|
'required output path. Correct the code and write the actual '
|
||||||
|
'artifact before finishing.'
|
||||||
|
)
|
||||||
|
round_recovery_messages.append(recovery)
|
||||||
|
result = {**result, 'error': recovery, 'exit_code': 1}
|
||||||
|
output = preview_tool_result_text(result, actual_tool, args)
|
||||||
|
failed = True
|
||||||
if (
|
if (
|
||||||
not failed
|
not failed
|
||||||
and canonical(actual_tool) == 'web_fetch'
|
and canonical(actual_tool) == 'web_fetch'
|
||||||
|
|||||||
@@ -6,7 +6,7 @@ import jsonschema
|
|||||||
import pytest
|
import pytest
|
||||||
import re
|
import re
|
||||||
|
|
||||||
from src.clean_agent_preview import conversation, readonly_call, preview_call_allowed, evaluate_preview_call, authorized_write_families, compact_schemas, normalize_preview_function_args, normalize_preview_call_args, private_browser_dom_batch, private_browser_state_transition, private_browser_success_repeat_limit, stream_preview, denied_response, execution_has_write_effect, requests_mutation, claims_completion, recent_successful_write_families, scope_preview_contract, multimodal_image_count, attachment_reference_count, active_document_context_message, active_email_context_message, targets_active_editor, active_editor_whole_draft_request, active_editor_suggestion_request, scope_active_editor_contract, native_execution_limits, interactive_execution_limit, runtime_required_artifacts, execution_targets_required_artifact, artifact_completion_tool_schemas, required_artifact_completion_tool_choice, document_suggestions_event, document_suggestion_quality_error, required_read_tool_choice, required_active_editor_tool_choice, sealed_read_arguments, email_identifier_error, requested_item_limit, contract_item_limit, notes_terminal_response, documents_terminal_response, shell_listing_terminal_response, shell_output_terminal_response, direct_shell_output_request, ui_panel_terminal_response, ui_toggle_state_result, calendar_terminal_response, memory_terminal_response, tasks_terminal_response, task_list_requires_synthesis, skills_terminal_response, cookbook_servers_terminal_response, prior_short_answer_for_no_tool_summary, prior_collection_repeat_answer, prior_failed_operation_answer, prior_cookbook_server_answer, prior_workspace_path_answer, prior_web_source_answer, bounded_web_evidence_answer, inherit_referential_read_arguments, normalized_search_intent, requested_web_source_links, web_source_links, requested_web_link_limit, preserve_requested_web_recency, ground_referenced_note_content, note_search_result_empty, note_referent_error, research_referent_error, private_browser_open_url, private_browser_effective_url, web_fetch_observation_is_boilerplate, broad_current_web_request, record_tool_execution, align_structured_tool_history, provider_request_messages, provider_compatible_tool_choice_request, offered_tool_alias, dependent_write_prerequisite_error, bounded_research_tool_policy, retrieved_source_urls, serialize_required_email_attachment_chain
|
from src.clean_agent_preview import conversation, readonly_call, preview_call_allowed, evaluate_preview_call, authorized_write_families, compact_schemas, normalize_preview_function_args, normalize_preview_call_args, private_browser_dom_batch, private_browser_state_transition, private_browser_success_repeat_limit, stream_preview, denied_response, execution_has_write_effect, requests_mutation, claims_completion, recent_successful_write_families, scope_preview_contract, multimodal_image_count, attachment_reference_count, active_document_context_message, active_email_context_message, targets_active_editor, active_editor_whole_draft_request, active_editor_suggestion_request, scope_active_editor_contract, native_execution_limits, interactive_execution_limit, runtime_required_artifacts, execution_targets_required_artifact, artifact_completion_python_code_error, artifact_completion_tool_schemas, required_artifact_completion_tool_choice, document_suggestions_event, document_suggestion_quality_error, required_read_tool_choice, required_active_editor_tool_choice, sealed_read_arguments, email_identifier_error, requested_item_limit, contract_item_limit, notes_terminal_response, documents_terminal_response, shell_listing_terminal_response, shell_output_terminal_response, direct_shell_output_request, ui_panel_terminal_response, ui_toggle_state_result, calendar_terminal_response, memory_terminal_response, tasks_terminal_response, task_list_requires_synthesis, skills_terminal_response, cookbook_servers_terminal_response, prior_short_answer_for_no_tool_summary, prior_collection_repeat_answer, prior_failed_operation_answer, prior_cookbook_server_answer, prior_workspace_path_answer, prior_web_source_answer, bounded_web_evidence_answer, inherit_referential_read_arguments, normalized_search_intent, requested_web_source_links, web_source_links, requested_web_link_limit, preserve_requested_web_recency, ground_referenced_note_content, note_search_result_empty, note_referent_error, research_referent_error, private_browser_open_url, private_browser_effective_url, web_fetch_observation_is_boilerplate, broad_current_web_request, record_tool_execution, align_structured_tool_history, provider_request_messages, provider_compatible_tool_choice_request, offered_tool_alias, dependent_write_prerequisite_error, bounded_research_tool_policy, retrieved_source_urls, serialize_required_email_attachment_chain
|
||||||
from src.tool_capabilities import capabilities_for_tool
|
from src.tool_capabilities import capabilities_for_tool
|
||||||
|
|
||||||
|
|
||||||
@@ -657,17 +657,34 @@ def test_artifact_completion_schema_binds_directory_descendant_but_not_multiple_
|
|||||||
assert 'inside the required directory' in directory_path['description']
|
assert 'inside the required directory' in directory_path['description']
|
||||||
assert 'pattern' not in source[0]['function']['parameters']['properties']['path']
|
assert 'pattern' not in source[0]['function']['parameters']['properties']['path']
|
||||||
directory_code = directory_bound[1]['function']['parameters']['properties']['code']
|
directory_code = directory_bound[1]['function']['parameters']['properties']['code']
|
||||||
assert re.search(
|
assert 'pattern' not in directory_code
|
||||||
directory_code['pattern'],
|
|
||||||
"Path('/workspace/results/1.tex').write_text('table')",
|
|
||||||
)
|
|
||||||
assert not re.search(directory_code['pattern'], "Path('/workspace/source.tar').unlink()")
|
|
||||||
assert 'non-empty files inside' in directory_code['description']
|
assert 'non-empty files inside' in directory_code['description']
|
||||||
|
assert 'Complete executable Python' in directory_code['description']
|
||||||
|
assert 'not only a path string' in directory_code['description']
|
||||||
assert artifact_completion_tool_schemas(
|
assert artifact_completion_tool_schemas(
|
||||||
source, ('/workspace/a.txt', '/workspace/b.txt'),
|
source, ('/workspace/a.txt', '/workspace/b.txt'),
|
||||||
) == source
|
) == source
|
||||||
|
|
||||||
|
|
||||||
|
def test_artifact_completion_python_code_requires_valid_code_and_output_reference():
|
||||||
|
required = ('/workspace/results',)
|
||||||
|
|
||||||
|
assert 'valid executable Python' in artifact_completion_python_code_error(
|
||||||
|
{'code': '/workspace/results/'}, required,
|
||||||
|
)
|
||||||
|
assert 'required output directory' in artifact_completion_python_code_error(
|
||||||
|
{'code': "Path('/workspace/source.tar').unlink()"}, required,
|
||||||
|
)
|
||||||
|
assert artifact_completion_python_code_error(
|
||||||
|
{'code': (
|
||||||
|
"from pathlib import Path\n"
|
||||||
|
"p = Path('/workspace/results/1.tex')\n"
|
||||||
|
"p.write_text('table')"
|
||||||
|
)},
|
||||||
|
required,
|
||||||
|
) == ''
|
||||||
|
|
||||||
|
|
||||||
def test_required_binary_artifact_forces_python_instead_of_text_writer():
|
def test_required_binary_artifact_forces_python_instead_of_text_writer():
|
||||||
offered = [
|
offered = [
|
||||||
{'type': 'function', 'function': {'name': 'write_file'}},
|
{'type': 'function', 'function': {'name': 'write_file'}},
|
||||||
@@ -695,8 +712,9 @@ def test_required_binary_artifact_forces_python_instead_of_text_writer():
|
|||||||
)
|
)
|
||||||
assert [schema['function']['name'] for schema in binary_bound] == ['python']
|
assert [schema['function']['name'] for schema in binary_bound] == ['python']
|
||||||
code_schema = binary_bound[0]['function']['parameters']['properties']['code']
|
code_schema = binary_bound[0]['function']['parameters']['properties']['code']
|
||||||
assert re.search(code_schema['pattern'], "open('/workspace/output.png', 'wb')")
|
assert 'pattern' not in code_schema
|
||||||
assert not re.search(code_schema['pattern'], "print('more analysis')")
|
assert 'Complete executable Python' in code_schema['description']
|
||||||
|
assert 'not only a path string' in code_schema['description']
|
||||||
assert '/workspace/output.png' in code_schema['description']
|
assert '/workspace/output.png' in code_schema['description']
|
||||||
assert [schema['function']['name'] for schema in offered] == [
|
assert [schema['function']['name'] for schema in offered] == [
|
||||||
'write_file', 'python',
|
'write_file', 'python',
|
||||||
@@ -5624,6 +5642,11 @@ async def test_native_stream_reserves_remaining_budget_for_required_artifact(mon
|
|||||||
assert requests[12]["tool_choice"] == {
|
assert requests[12]["tool_choice"] == {
|
||||||
"type": "function", "function": {"name": "python"},
|
"type": "function", "function": {"name": "python"},
|
||||||
}
|
}
|
||||||
|
completion_code_schema = (
|
||||||
|
requests[12]["tools"][0]["function"]["parameters"]["properties"]["code"]
|
||||||
|
)
|
||||||
|
assert "pattern" not in completion_code_schema
|
||||||
|
assert "Complete executable Python" in completion_code_schema["description"]
|
||||||
assert "create one or more files inside" in requests[12]["messages"][-1]["content"]
|
assert "create one or more files inside" in requests[12]["messages"][-1]["content"]
|
||||||
assert "Do not pass the directory itself as a file path" in requests[12]["messages"][-1]["content"]
|
assert "Do not pass the directory itself as a file path" in requests[12]["messages"][-1]["content"]
|
||||||
assert executed[-1] == "python"
|
assert executed[-1] == "python"
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ ROOT = Path(__file__).resolve().parents[1]
|
|||||||
def test_original_harness_version_is_canonical_semver() -> None:
|
def test_original_harness_version_is_canonical_semver() -> None:
|
||||||
version = (ROOT / "HARNESS_VERSION").read_text(encoding="utf-8").strip()
|
version = (ROOT / "HARNESS_VERSION").read_text(encoding="utf-8").strip()
|
||||||
|
|
||||||
assert version == "0.20.15"
|
assert version == "0.20.16"
|
||||||
assert re.fullmatch(r"\d+\.\d+\.\d+", version)
|
assert re.fullmatch(r"\d+\.\d+\.\d+", version)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user