Удалить agent.py
This commit is contained in:
@@ -1,68 +0,0 @@
|
|||||||
"""
|
|
||||||
A simple self-correcting agent example using LangGraph.
|
|
||||||
|
|
||||||
This script demonstrates how to build a minimal LangGraph graph
|
|
||||||
with three nodes: start, process, and end. The graph concatenates
|
|
||||||
a greeting message and prints it at the end. The example ensures
|
|
||||||
that imports from `langgraph.graph` work correctly.
|
|
||||||
"""
|
|
||||||
|
|
||||||
from langgraph.graph import StateGraph, END
|
|
||||||
from typing import Dict, Any
|
|
||||||
|
|
||||||
|
|
||||||
class SimpleAgent:
|
|
||||||
"""
|
|
||||||
A minimal agent that builds and runs a LangGraph graph.
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self) -> None:
|
|
||||||
# Create a new StateGraph instance
|
|
||||||
self.graph = StateGraph()
|
|
||||||
|
|
||||||
# Add nodes to the graph
|
|
||||||
self.graph.add_node("start", self.start_node)
|
|
||||||
self.graph.add_node("process", self.process_node)
|
|
||||||
self.graph.add_node("end", self.end_node)
|
|
||||||
|
|
||||||
# Define the entry point and edges
|
|
||||||
self.graph.set_entry_point("start")
|
|
||||||
self.graph.add_edge("start", "process")
|
|
||||||
self.graph.add_edge("process", "end")
|
|
||||||
self.graph.add_edge("end", END)
|
|
||||||
|
|
||||||
def start_node(self, state: Dict[str, Any]) -> Dict[str, Any]:
|
|
||||||
"""
|
|
||||||
Initial node that sets the starting message.
|
|
||||||
"""
|
|
||||||
state["message"] = "Hello"
|
|
||||||
return state
|
|
||||||
|
|
||||||
def process_node(self, state: Dict[str, Any]) -> Dict[str, Any]:
|
|
||||||
"""
|
|
||||||
Process node that appends to the message.
|
|
||||||
"""
|
|
||||||
state["message"] += " World"
|
|
||||||
return state
|
|
||||||
|
|
||||||
def end_node(self, state: Dict[str, Any]) -> Dict[str, Any]:
|
|
||||||
"""
|
|
||||||
End node that prints the final message.
|
|
||||||
"""
|
|
||||||
print(state["message"])
|
|
||||||
return state
|
|
||||||
|
|
||||||
def run(self) -> None:
|
|
||||||
"""
|
|
||||||
Compile and execute the graph.
|
|
||||||
"""
|
|
||||||
# Compile the graph into a runnable function
|
|
||||||
runnable = self.graph.compile()
|
|
||||||
|
|
||||||
# Execute the graph with an empty initial state
|
|
||||||
runnable({})
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
|
||||||
agent = SimpleAgent()
|
|
||||||
agent.run()
|
|
||||||
Reference in New Issue
Block a user