Change env credentials to be returned with same std properties

rrg92 · Nov 28, 2024 · 2cbaa2f · 2cbaa2f
1 parent cd7517a
commit 2cbaa2f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/powershai/AiCredentials.ps1 b/powershai/AiCredentials.ps1
@@ -454,7 +454,7 @@ function Get-AiDefaultCredential {
 			if($EnvValue){
 				verbose "Found credential in environment variable $VarName";
 				$Sources += @{ 
-						credential 	= $EnvValue
+						credential 	= @{ credential = $EnvValue }
 						source 		= "environment" 
 						desc 		= "environment $VarName"
 					};

diff --git a/powershai/providers/openai.ps1 b/powershai/providers/openai.ps1
@@ -137,11 +137,7 @@ function GetCurrentOpenaiToken {
 	}
 
 	if($Credential){
-		if($Credential.source -eq "environment"){
-			return $Credential.credential;
-		} else {
-			return $Credential.credential.GetCredential()
-		}
+		return $Credential.credential.credential;
 	} else {
 		write-warning "No default credential was found. Failback to deprecated method. Check if you have multiple creds and set a default credentials!"
 	}