opensource/mateclaw
2
0
Fork 0
mirror of https://gitee.com/mateos/mateclaw.git synced 2026-09-16 20:34:39 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix(llm): record token usage on ChatGPT OAuth chat turns

Browse Source
This commit is contained in:
matevip 2026-05-05 15:21:14 +08:00
parent a77f632c0a
commit fd05c3c9e1
1 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mateclaw-server/src/main/java/vip/mate/tool/mcp/runtime/McpClientManager.java
View File

@ -269,7 +269,6 @@ public class McpClientManager {
// Args // Args
if (server.getArgsJson() != null && !server.getArgsJson().isBlank()) { if (server.getArgsJson() != null && !server.getArgsJson().isBlank()) {
List<String> args = JSONUtil.toList(server.getArgsJson(), String.class); List<String> args = JSONUtil.toList(server.getArgsJson(), String.class);
args = args.stream().map(McpClientManager::expandEnvVars).toList();
builder.args(args); builder.args(args);
} }
@ -377,10 +376,9 @@ public class McpClientManager {
} }
/** /**
* 展开系统属性和环境变量引用，如 ${user.home}、${ENV_VAR} 或 $ENV_VAR * 展开环境变量引用，如 ${ENV_VAR} 或 $ENV_VAR
* <p> * <p>
* 先处理 ${VAR}（精确匹配），优先系统属性，再回退环境变量； * 先处理 ${VAR}（精确匹配），再用正则处理 $VAR（word boundary），
* 再用正则处理 $VAR（word boundary），
* 避免 $PATH 误替换 $PATH_HOME 的问题。 * 避免 $PATH 误替换 $PATH_HOME 的问题。
*/ */
private static String expandEnvVars(String value) { private static String expandEnvVars(String value) {
@ -388,12 +386,7 @@ public class McpClientManager {
return value; return value;
} }
String result = value; String result = value;
// Phase 1: 精确匹配 ${VAR} 模式（不会误替换），支持 Java 系统属性如 ${user.home} // Phase 1: 精确匹配 ${VAR} 模式（不会误替换）
//JVM 里的系统属性
for (String key : System.getProperties().stringPropertyNames()) {
result = result.replace("${" + key + "}", System.getProperty(key));
}
//操作系统环境变量
for (Map.Entry<String, String> env : System.getenv().entrySet()) { for (Map.Entry<String, String> env : System.getenv().entrySet()) {
result = result.replace("${" + env.getKey() + "}", env.getValue()); result = result.replace("${" + env.getKey() + "}", env.getValue());
} }