Hive agent framework

import asyncio
from framework.graph import Goal, GraphSpec, NodeSpec, EdgeSpec, EdgeCondition
from framework.graph.executor import GraphExecutor
from framework.runtime.core import Runtime

# Define the goal
goal = Goal(
    id="greet-user",
    name="Greet User",
    description="Generate a friendly uppercase greeting",
    success_criteria=[{
        "id": "greeting_generated",
        "description": "Greeting produced",
        "metric": "custom",
        "target": "any"
    }]
)

# Define nodes
greeter = NodeSpec(
    id="greeter",
    name="Greeter",
    description="Generates a simple greeting",
    node_type="event_loop",
    input_keys=["name"],
    output_keys=["greeting"]
)

uppercaser = NodeSpec(
    id="uppercaser",
    name="Uppercaser",
    description="Converts greeting to uppercase",
    node_type="event_loop",
    input_keys=["greeting"],
    output_keys=["final_greeting"]
)

# Connect nodes with an edge
edge = EdgeSpec(
    id="greet-to-upper",
    source="greeter",
    target="uppercaser",
    condition=EdgeCondition.ON_SUCCESS
)

# Create the graph
graph = GraphSpec(
    id="greeting-agent",
    goal_id="greet-user",
    entry_node="greeter",
    terminal_nodes=["uppercaser"],
    nodes=[greeter, uppercaser],
    edges=[edge]
)

# Execute the agent
async def main():
    runtime = Runtime(storage_path=Path("./agent_logs"))
    executor = GraphExecutor(runtime=runtime)
    
    result = await executor.execute(
        graph=graph,
        goal=goal,
        input_data={"name": "Alice"}
    )
    
    print(f"Result: {result.output.get('final_greeting')}")

asyncio.run(main())