Skip to content

Commit 1db69b1

Browse files
authored
fix: adapting guardrail for use in agentkit (#484)
1 parent b5afab3 commit 1db69b1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

veadk/tools/builtin_tools/llm_shield.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,9 @@ def __init__(self, timeout: int = 50) -> None:
6464
super().__init__(name=self.name)
6565

6666
self.appid = getenv("TOOL_LLM_SHIELD_APP_ID")
67-
self.region = getenv("TOOL_LLM_SHIELD_REGION", "cn-beijing")
67+
self.region = (
68+
os.getenv("REGION") or os.getenv("TOOL_LLM_SHIELD_REGION") or "cn-beijing"
69+
)
6870
self.timeout = timeout
6971
self.url = getenv(
7072
"TOOL_LLM_SHIELD_URL",

0 commit comments

Comments
 (0)