Potential fix for code scanning alert no. 26: Incomplete URL substring sanitization

Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>
This commit is contained in:
Rip&Tear
2025-11-08 21:31:40 +08:00
committed by GitHub
parent 0f1c173d02
commit f557334c01

View File

@@ -4,7 +4,7 @@ import json
import logging
import os
from typing import TYPE_CHECKING, Any
from urllib.parse import urlparse
from pydantic import BaseModel
from crewai.utilities.agent_utils import is_context_length_exceeded
@@ -143,8 +143,10 @@ class AzureCompletion(BaseLLM):
prefix in model.lower() for prefix in ["gpt-", "o1-", "text-"]
)
parsed_endpoint = urlparse(self.endpoint)
host = parsed_endpoint.hostname or ""
self.is_azure_openai_endpoint = (
"openai.azure.com" in self.endpoint
(host == "openai.azure.com" or host.endswith(".openai.azure.com"))
and "/openai/deployments/" in self.endpoint
)