not propagating pipeline_id
This commit is contained in:
@@ -91,12 +91,6 @@ class RoutingGraph(GraphBase):
|
||||
tool_manager:ToolManager = self.config.tool_manager_config.setup()
|
||||
self.chat_model = create_agent(self.chat_llm, [], checkpointer=self.memory)
|
||||
|
||||
# Propagate pipeline_id and prompt_set_id to tool node config
|
||||
if self.config.pipeline_id and hasattr(self.config.tool_node_config, 'pipeline_id'):
|
||||
self.config.tool_node_config.pipeline_id = self.config.pipeline_id
|
||||
if self.config.prompt_set_id and hasattr(self.config.tool_node_config, 'prompt_set_id'):
|
||||
self.config.tool_node_config.prompt_set_id = self.config.prompt_set_id
|
||||
|
||||
self.tool_node:ToolNodeBase = self.config.tool_node_config.setup(tool_manager=tool_manager,
|
||||
memory=self.memory)
|
||||
|
||||
|
||||
@@ -139,12 +139,6 @@ class ChattyToolNode(ToolNodeBase):
|
||||
self.chatty_agent = create_agent(self.chatty_llm, [], checkpointer=self.mem)
|
||||
# self.tool_agent = create_agent(self.tool_llm, self.tool_manager.get_list_langchain_tools(), checkpointer=self.mem)
|
||||
|
||||
# Propagate pipeline_id and prompt_set_id to inner tool_node_conf
|
||||
if self.config.pipeline_id and hasattr(self.config.tool_node_conf, 'pipeline_id'):
|
||||
self.config.tool_node_conf.pipeline_id = self.config.pipeline_id
|
||||
if self.config.prompt_set_id and hasattr(self.config.tool_node_conf, 'prompt_set_id'):
|
||||
self.config.tool_node_conf.prompt_set_id = self.config.prompt_set_id
|
||||
|
||||
self.tool_agent = self.config.tool_node_conf.setup(tool_manager=self.tool_manager,
|
||||
memory=self.mem)
|
||||
|
||||
|
||||
@@ -103,12 +103,6 @@ class Pipeline:
|
||||
self.config.graph_config.base_url = self.config.base_url if self.config.base_url is not None else self.config.graph_config.base_url
|
||||
self.config.graph_config.api_key = self.config.api_key
|
||||
|
||||
# Propagate pipeline_id and prompt_set_id to graph config for DB prompt loading
|
||||
if self.config.pipeline_id is not None and hasattr(self.config.graph_config, 'pipeline_id'):
|
||||
self.config.graph_config.pipeline_id = self.config.pipeline_id
|
||||
if self.config.prompt_set_id is not None and hasattr(self.config.graph_config, 'prompt_set_id'):
|
||||
self.config.graph_config.prompt_set_id = self.config.prompt_set_id
|
||||
|
||||
self.graph:GraphBase = self.config.graph_config.setup()
|
||||
|
||||
def show_graph(self):
|
||||
|
||||
Reference in New Issue
Block a user