Skip to content

Enhance Interactive CLI and Mission Execution Lifecycle #210

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

Draft
hackerxj2010 wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

Enhance Interactive CLI and Mission Execution Lifecycle #210

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
268 changes: 198 additions & 70 deletions src/cognitive/cli.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,81 +36,209 @@ def build_parser() -> argparse.ArgumentParser:
return parser


async def run_shell(args: argparse.Namespace):
try:
import readline # Enable history and line editing
except ImportError:
pass

service = MissionExecutorService(workspace_root=args.workspace_root, mode=args.mode)
print(f"JeanBot interactive shell ({args.mode} mode)")
print(f"Workspace: {args.workspace_root} ({args.workspace_id})")
print("Type 'exit' or 'quit' to end session. Type 'help' for commands.")
class InteractiveShell:
def __init__(self, args: argparse.Namespace):
self.args = args
self.service = MissionExecutorService(
workspace_root=args.workspace_root,
mode=args.mode,
on_step_update=self.on_step_update,
)
self.mission_id: str | None = None
self.history: list[str] = []

last_result = None
mission_id = f"shell-{uuid.uuid4().hex[:8]}"
history: list[str] = []
def on_step_update(self, step_id: str, event: str):
color = "\033[94m" if event == "started" else "\033[92m"
reset = "\033[0m"
print(f" {color}[{event.upper()}]{reset} Step: {step_id}")

while True:
async def run(self):
try:
line = input("\njeanbot> ").strip()
if not line:
continue
if line.lower() in ("exit", "quit"):
break
import readline
except ImportError:
pass

print(f"JeanBot interactive shell ({self.args.mode} mode)")
print(f"Workspace: {self.args.workspace_root} ({self.args.workspace_id})")
print("Type 'exit' or 'quit' to end session. Type 'help' for commands.")

history.append(line)

if line.lower() == "help":
print("Commands:")
print(" help Show this help")
print(" history Show command history")
print(" exit | quit Exit shell")
print(" <objective> Plan and execute a mission")
print(" refine <feedback> Refine the last mission result with feedback")
continue

if line.lower() == "history":
for i, cmd in enumerate(history, 1):
print(f" {i:3} {cmd}")
continue

if line.lower().startswith("refine "):
if not last_result:
print("Nothing to refine. Run a mission first.")
while True:
try:
line = input("\njeanbot> ").strip()
if not line:
continue
feedback = line[7:].strip()
objective = (
f"Refine previous mission results based on: {feedback}\n"
f"Previous summary: {last_result.verification_summary}"
)
title = f"Refinement: {feedback[:30]}..."
else:
objective = line
title = f"Mission: {line[:30]}..."

payload = {
"mission_id": mission_id,
"workspace_id": args.workspace_id,
"title": title,
"objective": objective,
"mode": args.mode,
}

print(f"Executing: {title}")
last_result = await service.execute_payload(payload)

print(f"\nStatus: {last_result.status}")
print(f"Summary: {last_result.verification_summary}")
if last_result.artifacts:
print(f"Artifacts: {len(last_result.artifacts)}")
for artifact in last_result.artifacts:
print(f" - {artifact.title}: {artifact.path}")

except KeyboardInterrupt:
print("\nInterrupt received, type 'exit' to quit.")
except Exception as e:
print(f"\nError: {e}")
if line.lower() in ("exit", "quit"):
break

self.history.append(line)
await self.handle_command(line)

except EOFError:
print("\nExiting")
break
except KeyboardInterrupt:
print("\nInterrupt received, type 'exit' to quit.")
except Exception as e:
print(f"\033[91m\nError: {e}\033[0m")

async def handle_command(self, line: str):
cmd = line.lower()
if cmd == "help":
self.show_help()
elif cmd == "history":
self.show_history()
elif cmd == "status":
await self.show_status()
elif cmd == "plan":
await self.show_plan()
elif cmd == "artifacts":
await self.show_artifacts()
elif cmd.startswith("view "):
await self.view_artifact(line[5:].strip())
elif cmd.startswith("refine "):
await self.refine_mission(line[7:].strip())
else:
await self.start_mission(line)

def show_help(self):
print("Commands:")
print(" help Show this help")
print(" history Show command history")
print(" status Show status of current or last mission")
print(" plan Show plan for current or last mission")
print(" artifacts List artifacts for current or last mission")
print(" view <path> View content of an artifact")
print(" exit | quit Exit shell")
print(" <objective> Plan and execute a mission")
print(" refine <feedback> Refine the last mission result with feedback")

def show_history(self):
for i, cmd in enumerate(self.history, 1):
print(f" {i:3} {cmd}")

async def show_status(self, mission_id: str | None = None):
mid = mission_id or self.mission_id or self.service.get_last_mission_id()
if not mid:
print("No mission found.")
return
summary = self.service.get_mission_run_summary(mid)
if not summary:
print(f"No summary found for mission {mid}")
return

res = summary.get("result") or {}
status = res.get("status", "unknown")
color = "\033[92m" if status == "completed" else "\033[93m"
reset = "\033[0m"

print(f"Mission: {summary['mission']['title']} ({mid})")
print(f"Status: {color}{status.upper()}{reset}")
if res:
print(f"Summary: {res.get('verification_summary', 'N/A')}")
metrics = res.get("metrics", {})
print(f"Progress: {metrics.get('completed_steps', 0)}/{metrics.get('total_steps', 0)} steps")
print(f"Average Score: {metrics.get('average_score', 0.0)}")

async def show_plan(self, mission_id: str | None = None):
mid = mission_id or self.mission_id or self.service.get_last_mission_id()
if not mid:
print("No mission found.")
return
payload = self.service.get_mission_payload(mid)
if not payload:
print(f"No payload found for mission {mid}")
return

print(f"Plan for: {payload['title']}")
for step in payload.get("steps", []):
status = step.get("status", "pending")
color = "\033[92m" if status == "completed" else "\033[94m" if status == "running" else ""
reset = "\033[0m"
print(f" - [{color}{status.upper()}{reset}] {step['id']}: {step['title']}")
print(f" {step['description']}")

async def show_artifacts(self, mission_id: str | None = None):
mid = mission_id or self.mission_id or self.service.get_last_mission_id()
if not mid:
print("No mission found.")
return
summary = self.service.get_mission_run_summary(mid)
if not summary or not summary.get("result"):
print(f"No results found for mission {mid}")
return

artifacts = summary["result"].get("artifacts", [])
print(f"Artifacts for mission {mid}:")
for art in artifacts:
print(f" - {art['title']}: {art['path']}")

async def view_artifact(self, path: str):
p = Path(path)
if not p.exists():
# Try relative to workspace
p = Path(self.args.workspace_root) / path

if not p.exists():
print(f"Artifact not found: {path}")
return

print(f"--- Content of {path} ---")
print(p.read_text(encoding="utf-8"))
print("--- End of Content ---")

async def refine_mission(self, feedback: str):
mid = self.mission_id or self.service.get_last_mission_id()
if not mid:
print("Nothing to refine. Run a mission first.")
return

summary = self.service.get_mission_run_summary(mid)
res = summary.get("result") if summary else None

objective = (
f"Refine previous mission results based on: {feedback}\n"
f"Previous summary: {res.get('verification_summary') if res else 'N/A'}"
)
title = f"Refinement: {feedback[:30]}..."
await self.start_mission(objective, title)

async def start_mission(self, objective: str, title: str | None = None):
self.mission_id = f"shell-{uuid.uuid4().hex[:8]}"
payload = {
"mission_id": self.mission_id,
"workspace_id": self.args.workspace_id,
"title": title or f"Mission: {objective[:30]}...",
"objective": objective,
"mode": self.args.mode,
}

print(f"Planning mission: {payload['title']}")
bundle = await self.service.plan_mission(payload)

print("\nProposed Plan:")
for step in bundle.record.plan.steps:
print(f" - {step.id}: {step.title}")
print(f" {step.description}")

confirm = input("\nExecute this plan? (Y/n): ").strip().lower()
if confirm and confirm != "y":
print("Mission cancelled.")
return

print(f"\nExecuting: {payload['title']}")
result = await self.service.execute_bundle(bundle)

print(f"\nStatus: {result.status}")
print(f"Summary: {result.verification_summary}")
if result.artifacts:
print(f"Artifacts: {len(result.artifacts)}")
for artifact in result.artifacts:
print(f" - {artifact.title}: {artifact.path}")


async def run_shell(args: argparse.Namespace):
shell = InteractiveShell(args)
await shell.run()


async def run_command(args: argparse.Namespace) -> dict:
Expand Down
12 changes: 9 additions & 3 deletions src/cognitive/executor.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
import uuid
from dataclasses import dataclass, field
from datetime import datetime, timezone
from typing import Any, Awaitable, Protocol
from typing import Any, Awaitable, Callable, Protocol


def sleep_ms(ms: int) -> Awaitable[None]:
Expand Down Expand Up @@ -467,13 +467,15 @@ def __init__(
sub_agent_service: SubAgentService,
file_service: FileService,
policy_service: PolicyService,
on_step_update: Callable[[str, str], Any] | None = None,
):
self.runtime = runtime
self.memory_service = memory_service
self.audit_service = audit_service
self.sub_agent_service = sub_agent_service
self.file_service = file_service
self.policy_service = policy_service
self.on_step_update = on_step_update
self.intelligence = MissionExecutionIntelligence()
self.replanner = AdaptiveReplanner()

Expand Down Expand Up @@ -869,7 +871,9 @@ async def _execute_step(
) -> StepOutcome:
step_started_at = utc_now_iso()
step.status = "running"

if self.on_step_update:
self.on_step_update(step.id, "started")

await self.audit_service.record(
"mission.step.started",
step.id,
Expand Down Expand Up @@ -915,7 +919,9 @@ async def _execute_step(
)

step.status = "completed"

if self.on_step_update:
self.on_step_update(step.id, "completed")

report = StepExecutionRecord(
step_id=sub_agent_result.step_report.step_id,
started_at=step_started_at,
Expand Down
Loading
Loading