34 lines
899 B
Python
34 lines
899 B
Python
import os
|
|
from src.graph import create_graph
|
|
from src.state import PlanningState
|
|
|
|
def main() -> None:
|
|
# Ensure the OpenAI API key is set
|
|
if "OPENAI_API_KEY" not in os.environ:
|
|
raise RuntimeError("Please set the OPENAI_API_KEY environment variable.")
|
|
|
|
task = "Compare Python and JavaScript"
|
|
initial_state: PlanningState = {
|
|
"task": task,
|
|
"plan": None,
|
|
"current_step": 0,
|
|
"results": []
|
|
}
|
|
|
|
graph = create_graph()
|
|
final_state = graph.invoke(initial_state)
|
|
|
|
print("\n=== Plan ===")
|
|
for i, step in enumerate(final_state["plan"], 1):
|
|
print(f"{i}. {step}")
|
|
|
|
print("\n=== Results ===")
|
|
for i, res in enumerate(final_state["results"], 1):
|
|
print(f"[Step {i}] {res}")
|
|
|
|
print("\n=== Final Summary ===")
|
|
summary = "\n".join(final_state["results"])
|
|
print(summary)
|
|
|
|
if __name__ == "__main__":
|
|
main() |