diff --git a/lang_agent/pipeline.py b/lang_agent/pipeline.py index 9c62035..ca96d4a 100644 --- a/lang_agent/pipeline.py +++ b/lang_agent/pipeline.py @@ -225,4 +225,16 @@ class Pipeline: async def aclear_memory(self): """Async version: Clear all memory from the graph.""" if hasattr(self.graph, "aclear_memory"): - await self.graph.aclear_memory() \ No newline at end of file + await self.graph.aclear_memory() + + +if __name__ == "__main__": + from lang_agent.graphs import ReactGraphConfig + from dotenv import load_dotenv + load_dotenv() + # config = PipelineConfig(graph_config=ReactGraphConfig()) + config = PipelineConfig() + pipeline: Pipeline = config.setup() + for out in pipeline.chat("use the calculator tool to calculate 92*55 and say the answer", as_stream=True): + # print(out) + continue \ No newline at end of file