增加模型工厂的租户ID

2025-01-06 14:40:25 +08:00 · 2025-01-06 14:40:25 +08:00 · 8e469cfdc3
commit 8e469cfdc3
parent 04efc7f95d
1 changed files with 1 additions and 0 deletions
--- a/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
+++ b/yudao-module-llm/yudao-module-llm-biz/src/main/java/cn/iocoder/yudao/module/llm/service/http/TrainHttpService.java
@ -111,6 +111,7 @@ public class TrainHttpService {
            AigcAccountRespVO aigcAccountRespVO = account(headers);
            if (aigcAccountRespVO != null && CollUtil.isNotEmpty(aigcAccountRespVO.getTenants())) {
                aigcLoginRespVO.setTenantId(aigcAccountRespVO.getTenants().get(0).getId());
+                headers.put("X-Tenant-Id", aigcLoginRespVO.getTenantId());
                log.info(" login:{}", aigcAccountRespVO);
                return aigcLoginRespVO;
            }