github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-14 14:05:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

add http protocol support

Browse Source
This commit is contained in:
Kaspars Strods
2025-04-16 22:26:08 +01:00
parent 9e82f1df4e
commit 69b454ce03
1 changed files with 20 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
libs/community/langchain_community/retrievers/azure_ai_search.py
View File

@@ -146,22 +146,30 @@ class AzureAISearchRetriever(BaseRetriever):
def _build_search_url(self, query: str) -> str: def _build_search_url(self, query: str) -> str:
url_suffix = get_from_env("", "AZURE_AI_SEARCH_URL_SUFFIX", DEFAULT_URL_SUFFIX) url_suffix = get_from_env("", "AZURE_AI_SEARCH_URL_SUFFIX", DEFAULT_URL_SUFFIX)
if url_suffix in self.service_name and "https://" in self.service_name:
base_url = f"{self.service_name}/" # Extract protocol and remaining part if protocol exists
elif url_suffix in self.service_name and "https://" not in self.service_name: if "://" in self.service_name:
base_url = f"https://{self.service_name}/" protocol, remaining = self.service_name.split("://", 1)
elif url_suffix not in self.service_name and "https://" in self.service_name: has_protocol = True
base_url = f"{self.service_name}.{url_suffix}/"
elif (
url_suffix not in self.service_name and "https://" not in self.service_name
):
base_url = f"https://{self.service_name}.{url_suffix}/"
else: else:
# pass to Azure to throw a specific error protocol = "https" # Default to HTTPS
base_url = self.service_name remaining = self.service_name
has_protocol = False
# Handle different based on whether service_name already contains url_suffix
if url_suffix in remaining:
base_url = (
f"{protocol}://{remaining}/"
if not has_protocol
else f"{self.service_name}/"
)
else:
base_url = f"{protocol}://{remaining}.{url_suffix}/"
endpoint_path = f"indexes/{self.index_name}/docs?api-version={self.api_version}" endpoint_path = f"indexes/{self.index_name}/docs?api-version={self.api_version}"
top_param = f"&$top={self.top_k}" if self.top_k else "" top_param = f"&$top={self.top_k}" if self.top_k else ""
filter_param = f"&$filter={self.filter}" if self.filter else "" filter_param = f"&$filter={self.filter}" if self.filter else ""
return base_url + endpoint_path + f"&search={query}" + top_param + filter_param return base_url + endpoint_path + f"&search={query}" + top_param + filter_param
@property @property