From 1ec0d86ae74688315563a749b415c859f1906064 Mon Sep 17 00:00:00 2001 From: goulustis Date: Fri, 10 Oct 2025 17:31:34 +0800 Subject: [PATCH] add a bunch of placeholder --- lang_agent/pipeline.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/lang_agent/pipeline.py b/lang_agent/pipeline.py index cb8b62a..fc01ebd 100644 --- a/lang_agent/pipeline.py +++ b/lang_agent/pipeline.py @@ -7,6 +7,9 @@ from websockets.asyncio.server import ServerConnection from loguru import logger from langchain.chat_models import init_chat_model +from langgraph.prebuilt import create_react_agent +from langgraph.checkpoint.memory import MemorySaver + from lang_agent.config import InstantiateConfig @@ -52,7 +55,10 @@ class Pipeline: api_key=self.config.api_key, base_url=self.config.base_url) - self.agent = self.llm # NOTE: placeholder for now, add graph later + # self.agent = self.llm # NOTE: placeholder for now, add graph later + memory = MemorySaver() + tools = [] + self.agent = create_react_agent(self.llm, tools, checkpointer=memory) def respond(self, msg:str): return self.agent.invoke(msg)