86 lines
2.1 KiB
Python
86 lines
2.1 KiB
Python
|
"""
|
||
|
|
Example: Pulse & Brain with ONLY YOUR chosen checks.
|
||
|
|
|
||
|
|
By default, pulse_brain.py includes example checks.
|
||
|
|
This shows how to start with a CLEAN SLATE and only add what YOU want.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import time
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from agent import Agent
|
||
|
|
from pulse_brain import BrainTask, CheckType, PulseCheck, PulseBrain
|
||
|
|
|
||
|
|
|
||
|
|
def check_my_file() -> dict:
|
||
|
|
"""Check if the important data file exists."""
|
||
|
|
file = Path("important_data.json")
|
||
|
|
return {
|
||
|
|
"status": "ok" if file.exists() else "error",
|
||
|
|
"message": f"File exists: {file.exists()}",
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> None:
|
||
|
|
agent = Agent(provider="claude", enable_heartbeat=False)
|
||
|
|
|
||
|
|
# Create Pulse & Brain with NO automatic checks
|
||
|
|
pb = PulseBrain(agent, pulse_interval=60)
|
||
|
|
|
||
|
|
# Remove all default checks (start clean)
|
||
|
|
pb.pulse_checks = []
|
||
|
|
pb.brain_tasks = []
|
||
|
|
|
||
|
|
print(
|
||
|
|
"Starting with ZERO checks. "
|
||
|
|
"You have complete control.\n"
|
||
|
|
)
|
||
|
|
|
||
|
|
# Add ONLY the checks you want
|
||
|
|
pb.pulse_checks.append(
|
||
|
|
PulseCheck(
|
||
|
|
"my-file", check_my_file,
|
||
|
|
interval_seconds=300,
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
pb.brain_tasks.append(
|
||
|
|
BrainTask(
|
||
|
|
name="file-recovery",
|
||
|
|
check_type=CheckType.CONDITIONAL,
|
||
|
|
prompt_template=(
|
||
|
|
"The file important_data.json is missing. "
|
||
|
|
"What should I do to recover it?"
|
||
|
|
),
|
||
|
|
condition_func=lambda data: (
|
||
|
|
data.get("status") == "error"
|
||
|
|
),
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
print("Added 1 pulse check: my-file")
|
||
|
|
print("Added 1 brain task: file-recovery")
|
||
|
|
print("\nThe agent will ONLY:")
|
||
|
|
print(
|
||
|
|
" 1. Check if important_data.json exists "
|
||
|
|
"(every 5 min, zero cost)"
|
||
|
|
)
|
||
|
|
print(
|
||
|
|
" 2. Ask for recovery help IF it's missing "
|
||
|
|
"(costs tokens)"
|
||
|
|
)
|
||
|
|
print("\nNothing else. You have complete control.\n")
|
||
|
|
|
||
|
|
pb.start()
|
||
|
|
|
||
|
|
try:
|
||
|
|
print("Running... Press Ctrl+C to stop\n")
|
||
|
|
while True:
|
||
|
|
time.sleep(1)
|
||
|
|
except KeyboardInterrupt:
|
||
|
|
pb.stop()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|