Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

load config

Browse Source
This commit is contained in:
goulustis
2026-03-03 14:54:22 +08:00
parent 1fcd5b4c61
commit 686c1d6a1f
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lang_agent/pipeline.py
View File

@@ -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 LLMNodeConfig
from lang_agent.config import LLMNodeConfig, load_tyro_conf
from lang_agent.graphs import AnnotatedGraph, ReactGraphConfig, RoutingConfig
from lang_agent.base import GraphBase
from lang_agent.components import conv_store
@@ -67,6 +67,12 @@ class PipelineConfig(LLMNodeConfig):
# graph_config: AnnotatedGraph = field(default_factory=ReactGraphConfig)
graph_config: AnnotatedGraph = field(default_factory=RoutingConfig)
def __post_init__(self):
if self.config_f is not None:
logger.info(f"loading config from {self.config_f}")
self.config = load_tyro_conf(self.config_f)
super().__post_init__()