From 7cb40fca0e080ce2545c3466409a68e262090287 Mon Sep 17 00:00:00 2001
From: goulustis <cnimaiv@gmail.com>
Date: Thu, 12 Feb 2026 15:17:46 +0800
Subject: [PATCH] add api key config

---
 lang_agent/front_api/build_server.py | 52 ++++++++++++++++++----------
 1 file changed, 33 insertions(+), 19 deletions(-)

diff --git a/lang_agent/front_api/build_server.py b/lang_agent/front_api/build_server.py
index 6061850..4a8731a 100644
--- a/lang_agent/front_api/build_server.py
+++ b/lang_agent/front_api/build_server.py
@@ -1,21 +1,27 @@
 from typing import List
 import subprocess
 
-def _build_template(graph:str,
-                    pipeline_id:str,
-                    prompt_set:str,
-                    tool_keys:List[str],
-                    port:str,
-                    entry_pnt:str="fastapi_server/server_dashscope.py",
-                    llm_name:str="qwen-plus"):
+def build_route(pipeline_id:str,
+                prompt_set:str,
+                tool_keys:List[str],
+                port:str,
+                api_key: str,
+                entry_pnt:str="fastapi_server/server_dashscope.py",
+                llm_name:str="qwen-plus"):
     cmd = [
         "python", entry_pnt,
-        "--llm-name", llm_name,
         "--port", str(port),
-        graph,
+        "route",            # ------------
+        "--llm-name", llm_name,
+        "--api-key", api_key,
         "--pipeline-id", pipeline_id,
         "--prompt-set-id", prompt_set,
-    ]
+        "tool_node",        # ------------
+        "--llm-name", llm_name,
+        "--api-key", api_key,
+        "--pipeline-id", pipeline_id,
+        "--prompt-set-id", prompt_set,
+        ]
     if tool_keys:
         cmd.extend(
             ["--tool-manager-config.client-tool-manager.tool-keys", *tool_keys]
@@ -24,22 +30,30 @@ def _build_template(graph:str,
 
     return sv_prc, f"http://0.0.0.0:{port}"
 
-def build_route(pipeline_id:str,
-                prompt_set:str,
-                tool_keys:List[str],
-                port:str,
-                entry_pnt:str="fastapi_server/server_dashscope.py",
-                llm_name:str="qwen-plus"):
-    return _build_template("route", pipeline_id, prompt_set, tool_keys, port, entry_pnt, llm_name)
-
 
 def build_react(pipeline_id:str,
                 prompt_set:str,
                 tool_keys:List[str],
                 port:str,
+                api_key: str,
                 entry_pnt:str="fastapi_server/server_dashscope.py",
                 llm_name:str="qwen-plus"):
-    return _build_template("react", pipeline_id, prompt_set, tool_keys, port, entry_pnt, llm_name)
+    cmd = [
+        "python", entry_pnt,
+        "--port", str(port),
+        "react",            # ------------
+        "--llm-name", llm_name,
+        "--api-key", api_key,
+        "--pipeline-id", pipeline_id,
+        "--prompt-set-id", prompt_set,
+        ]
+    if tool_keys:
+        cmd.extend(
+            ["--tool-manager-config.client-tool-manager.tool-keys", *tool_keys]
+        )
+    sv_prc = subprocess.Popen(cmd)
+
+    return sv_prc, f"http://0.0.0.0:{port}"
 
 # {pipeline_id: build_function}
 GRAPH_BUILD_FNCS = {