alexion
4def357d09
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: d981de1229ef899957bbe968bc8dcda02a21f477
29 lines
950 B
TypeScript
29 lines
950 B
TypeScript
/**
|
|
* Utility functions for Azure OpenAI tests
|
|
*/
|
|
|
|
function parseDeploymentNameMap(value: string | undefined): Map<string, string> {
|
|
const map = new Map<string, string>();
|
|
if (!value) return map;
|
|
for (const entry of value.split(",")) {
|
|
const trimmed = entry.trim();
|
|
if (!trimmed) continue;
|
|
const [modelId, deploymentName] = trimmed.split("=", 2);
|
|
if (!modelId || !deploymentName) continue;
|
|
map.set(modelId.trim(), deploymentName.trim());
|
|
}
|
|
return map;
|
|
}
|
|
|
|
export function hasAzureOpenAICredentials(): boolean {
|
|
const hasKey = !!process.env.AZURE_OPENAI_API_KEY;
|
|
const hasBaseUrl = !!(process.env.AZURE_OPENAI_BASE_URL || process.env.AZURE_OPENAI_RESOURCE_NAME);
|
|
return hasKey && hasBaseUrl;
|
|
}
|
|
|
|
export function resolveAzureDeploymentName(modelId: string): string | undefined {
|
|
const mapValue = process.env.AZURE_OPENAI_DEPLOYMENT_NAME_MAP;
|
|
if (!mapValue) return undefined;
|
|
return parseDeploymentNameMap(mapValue).get(modelId);
|
|
}
|