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

add api key config

Browse Source
This commit is contained in:
goulustis
2026-02-12 15:17:46 +08:00
parent 9ab9f0c36e
commit 7cb40fca0e
1 changed files with 33 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
lang_agent/front_api/build_server.py
View File

@@ -1,21 +1,27 @@
from typing import List from typing import List
import subprocess import subprocess
def _build_template(graph:str, def build_route(pipeline_id:str,
pipeline_id:str, prompt_set:str,
prompt_set:str, tool_keys:List[str],
tool_keys:List[str], port:str,
port:str, api_key: str,
entry_pnt:str="fastapi_server/server_dashscope.py", entry_pnt:str="fastapi_server/server_dashscope.py",
llm_name:str="qwen-plus"): llm_name:str="qwen-plus"):
cmd = [ cmd = [
"python", entry_pnt, "python", entry_pnt,
"--llm-name", llm_name,
"--port", str(port), "--port", str(port),
graph, "route", # ------------
"--llm-name", llm_name,
"--api-key", api_key,
"--pipeline-id", pipeline_id, "--pipeline-id", pipeline_id,
"--prompt-set-id", prompt_set, "--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: if tool_keys:
cmd.extend( cmd.extend(
["--tool-manager-config.client-tool-manager.tool-keys", *tool_keys] ["--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}" 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, def build_react(pipeline_id:str,
prompt_set:str, prompt_set:str,
tool_keys:List[str], tool_keys:List[str],
port:str, port:str,
api_key: str,
entry_pnt:str="fastapi_server/server_dashscope.py", entry_pnt:str="fastapi_server/server_dashscope.py",
llm_name:str="qwen-plus"): 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} # {pipeline_id: build_function}
GRAPH_BUILD_FNCS = { GRAPH_BUILD_FNCS = {