mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 16:18:30 +00:00
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:
@@ -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
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user