andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-02 07:42:40 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

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

Browse Source 
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
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
lib/crewai/src/crewai/llms/providers/azure/completion.py
View File

@@ -4,7 +4,7 @@ import json
import logging import logging
import os import os
from typing import TYPE_CHECKING, Any from typing import TYPE_CHECKING, Any
from urllib.parse import urlparse
from pydantic import BaseModel from pydantic import BaseModel
from crewai.utilities.agent_utils import is_context_length_exceeded 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-"] 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 = ( 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 and "/openai/deployments/" in self.endpoint
) )