Skip to content

improved guide agent context #93

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
dimavrem22 merged 2 commits into from
Jan 19, 2026
Merged

improved guide agent context #93

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
2e46aa9
better now
dimavrem22 Jan 19, 2026
baecac4
routine fix
dimavrem22 Jan 19, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
131 changes: 92 additions & 39 deletions scripts/run_guide_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -196,6 +196,24 @@ def print_error(error: str) -> None:
console.print()


def safe_parse_routine(routine_str: str | None) -> tuple[dict[str, Any] | None, str | None]:
"""
Safely parse a routine string to dict.

Returns:
Tuple of (parsed_dict, error_message).
If parsing succeeds: (dict, None)
If parsing fails: (None, error_message)
If routine_str is None: (None, None)
"""
if routine_str is None:
return None, None
try:
return json.loads(routine_str), None
except json.JSONDecodeError as e:
return None, f"Invalid JSON: {e}"


def print_routine_info(routine_json: dict[str, Any], validation_result: dict | None = None) -> None:
"""Print routine info in a nice table."""
table = Table(box=box.ROUNDED, show_header=False)
Expand Down Expand Up @@ -281,13 +299,17 @@ def __init__(

def _get_prompt(self) -> str:
"""Get the input prompt with routine name if loaded."""
routine_json = self._agent.routine_state.current_routine_json
if routine_json:
name = routine_json.get("name", "routine")
# Truncate long names
if len(name) > 20:
name = name[:17] + "..."
return f"[bold green]You[/bold green] [dim]({name})[/dim][bold green]>[/bold green] "
routine_str = self._agent.routine_state.current_routine_str
if routine_str:
routine_dict, _ = safe_parse_routine(routine_str)
if routine_dict:
name = routine_dict.get("name", "routine")
# Truncate long names
if len(name) > 20:
name = name[:17] + "..."
return f"[bold green]You[/bold green] [dim]({name})[/dim][bold green]>[/bold green] "
else:
return f"[bold green]You[/bold green] [dim](invalid json)[/dim][bold green]>[/bold green] "
return "[bold green]You>[/bold green] "

def _handle_stream_chunk(self, chunk: str) -> None:
Expand Down Expand Up @@ -360,31 +382,34 @@ def _handle_load_command(self, file_path: str) -> None:
console.print()
return

# Load as raw string (preserves content even if JSON is invalid)
with open(path, encoding="utf-8") as f:
routine_json = json.load(f)
routine_str = f.read()

self._loaded_routine_path = path
self._agent.routine_state.update_current_routine(routine_json)
self._agent.routine_state.update_current_routine(routine_str)

# Auto-validate on load
validation_result = validate_routine(routine_json)
# Try to parse and validate
routine_dict, parse_error = safe_parse_routine(routine_str)
if routine_dict is not None:
validation_result = validate_routine(routine_dict)
print_routine_info(routine_dict, validation_result)
else:
console.print()
console.print(f"[yellow]⚠ Loaded file with invalid JSON: {parse_error}[/yellow]")
console.print("[dim]Ask the agent for help fixing the JSON.[/dim]")

print_routine_info(routine_json, validation_result)
console.print(f"[dim]Watching: {path}[/dim]")
console.print()

except json.JSONDecodeError as e:
console.print()
console.print(f"[red]✗ Invalid JSON: {e}[/red]")
console.print()
except Exception as e:
console.print()
console.print(f"[red]✗ Error loading routine: {e}[/red]")
console.print()

def _handle_unload_command(self) -> None:
"""Handle /unload command to clear the current routine."""
if self._agent.routine_state.current_routine_json is None:
if self._agent.routine_state.current_routine_str is None:
console.print()
console.print("[yellow]No routine loaded.[/yellow]")
console.print()
Expand All @@ -398,23 +423,32 @@ def _handle_unload_command(self) -> None:

def _handle_show_command(self) -> None:
"""Handle /show command to display current routine details."""
routine_json = self._agent.routine_state.current_routine_json
if routine_json is None:
routine_str = self._agent.routine_state.current_routine_str
if routine_str is None:
console.print()
console.print("[yellow]No routine loaded. Use /load <file.json> first.[/yellow]")
console.print()
return

routine_dict, parse_error = safe_parse_routine(routine_str)
if routine_dict is None:
console.print()
console.print(f"[red]✗ {parse_error}[/red]")
console.print("[dim]Raw content:[/dim]")
console.print(routine_str[:500] + ("..." if len(routine_str) > 500 else ""))
console.print()
return

# Build detailed view
table = Table(box=box.ROUNDED, show_header=False, expand=True)
table.add_column("Field", style="dim", width=15)
table.add_column("Value", style="white")

table.add_row("Name", routine_json.get("name", "N/A"))
table.add_row("Description", routine_json.get("description", "N/A"))
table.add_row("Name", routine_dict.get("name", "N/A"))
table.add_row("Description", routine_dict.get("description", "N/A"))

# Parameters
params = routine_json.get("parameters", [])
params = routine_dict.get("parameters", [])
if params:
param_lines = []
for p in params:
Expand All @@ -425,7 +459,7 @@ def _handle_show_command(self) -> None:
table.add_row("Parameters", "[dim]None[/dim]")

# Operations summary
ops = routine_json.get("operations", [])
ops = routine_dict.get("operations", [])
if ops:
op_lines = []
for i, op in enumerate(ops, 1):
Expand All @@ -444,14 +478,21 @@ def _handle_show_command(self) -> None:

def _handle_validate_command(self) -> None:
"""Handle /validate command to validate the current routine."""
routine_json = self._agent.routine_state.current_routine_json
if routine_json is None:
routine_str = self._agent.routine_state.current_routine_str
if routine_str is None:
console.print()
console.print("[yellow]No routine loaded. Use /load <file.json> first.[/yellow]")
console.print()
return

result = validate_routine(routine_json)
routine_dict, parse_error = safe_parse_routine(routine_str)
if routine_dict is None:
console.print()
console.print(f"[bold red]✗ {parse_error}[/bold red]")
console.print()
return

result = validate_routine(routine_dict)
console.print()
if result.get("valid"):
console.print(f"[bold green]✓ Valid:[/bold green] {result.get('message', 'Routine is valid')}")
Expand All @@ -470,11 +511,15 @@ def _handle_status_command(self) -> None:
table.add_column("Value", style="white")

# Routine status
routine_json = self._agent.routine_state.current_routine_json
if routine_json:
table.add_row("Routine", f"[green]{routine_json.get('name', 'Unnamed')}[/green]")
table.add_row("Operations", str(len(routine_json.get("operations", []))))
table.add_row("Parameters", str(len(routine_json.get("parameters", []))))
routine_str = self._agent.routine_state.current_routine_str
if routine_str:
routine_dict, parse_error = safe_parse_routine(routine_str)
if routine_dict:
table.add_row("Routine", f"[green]{routine_dict.get('name', 'Unnamed')}[/green]")
table.add_row("Operations", str(len(routine_dict.get("operations", []))))
table.add_row("Parameters", str(len(routine_dict.get("parameters", []))))
else:
table.add_row("Routine", f"[yellow]Invalid JSON[/yellow]")
else:
table.add_row("Routine", "[dim]None loaded[/dim]")

Expand Down Expand Up @@ -509,8 +554,8 @@ def _reload_routine_from_file(self) -> None:
return
try:
with open(self._loaded_routine_path, encoding="utf-8") as f:
routine_json = json.load(f)
self._agent.routine_state.update_current_routine(routine_json)
routine_str = f.read()
self._agent.routine_state.update_current_routine(routine_str)
except Exception:
pass # Silently ignore reload errors

Expand Down Expand Up @@ -571,13 +616,21 @@ def _prompt_for_parameters(self, routine_json: dict[str, Any]) -> dict[str, Any]

def _handle_execute_command(self, params_path: str | None) -> None:
"""Handle /execute command to execute the loaded routine."""
routine_json = self._agent.routine_state.current_routine_json
if routine_json is None:
routine_str = self._agent.routine_state.current_routine_str
if routine_str is None:
console.print()
console.print("[red]✗ No routine loaded. Use /load <file.json> first.[/red]")
console.print()
return

# Parse routine
routine_dict, parse_error = safe_parse_routine(routine_str)
if routine_dict is None:
console.print()
console.print(f"[red]✗ Cannot execute: {parse_error}[/red]")
console.print()
return

# Load parameters
params: dict[str, Any] = {}
if params_path:
Expand All @@ -597,25 +650,25 @@ def _handle_execute_command(self, params_path: str | None) -> None:
return
else:
# Check if routine has required parameters
routine_params = routine_json.get("parameters", [])
routine_params = routine_dict.get("parameters", [])
has_required = any(p.get("required", False) for p in routine_params)

if has_required:
params = self._prompt_for_parameters(routine_json)
params = self._prompt_for_parameters(routine_dict)
if params is None:
return # User cancelled

try:
console.print()
with console.status("[bold yellow]Executing routine...[/bold yellow]"):
# Create and execute routine
routine = Routine(**routine_json)
routine = Routine(**routine_dict)
result = routine.execute(params)
result_dict = result.model_dump()

# Update agent state with execution result
self._agent.routine_state.update_last_execution(
routine_json=routine_json,
routine_json=routine_dict,
routine_params=params,
routine_result=result_dict,
)
Expand Down
Loading