use LLMNodeConfig

lang_agent/pipeline.py

@@ -13,7 +13,7 @@ from langchain_core.messages import SystemMessage, HumanMessage, BaseMessage
 from langchain.agents import create_agent
 from langgraph.checkpoint.memory import MemorySaver
         
-from lang_agent.config import LLMKeyConfig
+from lang_agent.config import LLMNodeConfig
 from lang_agent.graphs import AnnotatedGraph, ReactGraphConfig, RoutingConfig
 from lang_agent.base import GraphBase
 from lang_agent.components import conv_store
@@ -52,7 +52,7 @@ DEFAULT_PROMPT="""你是半盏新青年茶馆的服务员，擅长倾听、共
 
 @tyro.conf.configure(tyro.conf.SuppressFixed)
 @dataclass
-class PipelineConfig(LLMKeyConfig):
+class PipelineConfig(LLMNodeConfig):
     _target: Type = field(default_factory=lambda: Pipeline)
 
     config_f: str = None