kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-14 06:27:45 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[None][infra] Fixing credential loading in lockfile generation pipeline (#10020)

Browse Source 
Signed-off-by: Yuanjing Xue <197832395+yuanjingx87@users.noreply.github.com>
This commit is contained in:
yuanjingx87 2025-12-15 23:38:29 -08:00 committed by GitHub
parent 28b02b4f5a
commit 0a4c59136a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
jenkins/GenerateLock.groovy
View File

@ -38,14 +38,6 @@ def createKubernetesPodConfig()
return podConfig return podConfig
} }
def getGitCredentialId (String repoUrlKey) {
if (repoUrlKey == "tensorrt_llm_internal") {
return 'svc_tensorrt_gitlab_api_token_no_username_as_string'
} else {
return 'github-cred-trtllm-ci'
}
}
def generate() def generate()
{ {
sh "pwd && ls -alh" sh "pwd && ls -alh"
@ -63,7 +55,6 @@ def generate()
} }
LLM_REPO = params.customRepoUrl LLM_REPO = params.customRepoUrl
} }
def CREDENTIAL_ID = getGitCredentialId(params.repoUrlKey)
sh "apt update" sh "apt update"
sh "apt install -y python3-dev git curl git-lfs" sh "apt install -y python3-dev git curl git-lfs"
sh "git config --global --add safe.directory ${env.WORKSPACE}" sh "git config --global --add safe.directory ${env.WORKSPACE}"
@ -83,8 +74,20 @@ def generate()
sh "git status" sh "git status"
sh "git add \$(find . -type f \\( -name 'poetry.lock' -o -name 'pyproject.toml' -o -name 'metadata.json' \\))" sh "git add \$(find . -type f \\( -name 'poetry.lock' -o -name 'pyproject.toml' -o -name 'metadata.json' \\))"
sh "git commit -s -m \"[None][infra] Check in most recent lock file from nightly pipeline\"" sh "git commit -s -m \"[None][infra] Check in most recent lock file from nightly pipeline\""
withCredentials([string(credentialsId: CREDENTIAL_ID, variable: 'API_TOKEN')]) { withCredentials([
def authedUrl = LLM_REPO.replaceFirst('https://', "https://svc_tensorrt:${API_TOKEN}@") string(credentialsId: 'svc_tensorrt_gitlab_api_token_no_username_as_string', variable: 'GITLAB_API_TOKEN'),
usernamePassword(
credentialsId: 'github-cred-trtllm-ci',
usernameVariable: 'NOT_IN_USE',
passwordVariable: 'GITHUB_API_TOKEN'
)
]) {
def authedUrl
if (params.repoUrlKey == "tensorrt_llm_internal") {
authedUrl = LLM_REPO.replaceFirst('https://', "https://svc_tensorrt:${GITLAB_API_TOKEN}@")
} else {
authedUrl = LLM_REPO.replaceFirst('https://', "https://svc_tensorrt:${GITHUB_API_TOKEN}@")
}
sh "git remote set-url origin ${authedUrl}" sh "git remote set-url origin ${authedUrl}"
sh "git fetch origin ${params.branchName}" sh "git fetch origin ${params.branchName}"
sh "git status" sh "git status"