github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-12 06:13:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

community: undo azure_ad_access_token breaking change (#25818)

Browse Source
This commit is contained in:
Erick Friis 2024-08-29 19:06:14 -07:00 committed by GitHub
parent 6047138379
commit f7e62754a1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 27 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
libs/community/langchain_community/vectorstores/azuresearch.py
View File

@ -265,7 +265,6 @@ class AzureSearch(VectorStore):
self,
azure_search_endpoint: str,
azure_search_key: str,
azure_ad_access_token: Optional[str],
index_name: str,
embedding_function: Union[Callable, Embeddings],
search_type: str = "hybrid",
@ -281,6 +280,7 @@ class AzureSearch(VectorStore):
*,
vector_search_dimensions: Optional[int] = None,
additional_search_client_options: Optional[Dict[str, Any]] = None,
azure_ad_access_token: Optional[str] = None,
**kwargs: Any,
):
try:
@ -1298,14 +1298,18 @@ class AzureSearch(VectorStore):
}
),
**{
"captions": {
"text": result.get("@search.captions", [{}])[0].text,
"captions": (
{
"text": result.get("@search.captions", [{}])[
0
].text,
"highlights": result.get("@search.captions", [{}])[
0
].highlights,
}
if result.get("@search.captions")
else {},
else {}
),
"answers": semantic_answers_dict.get(
result.get(FIELDS_ID, ""),
"",
@ -1382,14 +1386,18 @@ class AzureSearch(VectorStore):
}
),
**{
"captions": {
"text": result.get("@search.captions", [{}])[0].text,
"captions": (
{
"text": result.get("@search.captions", [{}])[
0
].text,
"highlights": result.get("@search.captions", [{}])[
0
].highlights,
}
if result.get("@search.captions")
else {},
else {}
),
"answers": semantic_answers_dict.get(
result.get(FIELDS_ID, ""),
"",
@ -1421,10 +1429,10 @@ class AzureSearch(VectorStore):
azure_search = cls(
azure_search_endpoint,
azure_search_key,
azure_ad_access_token,
index_name,
embedding,
fields=fields,
azure_ad_access_token=azure_ad_access_token,
**kwargs,
)
azure_search.add_texts(texts, metadatas, **kwargs)
@ -1447,10 +1455,10 @@ class AzureSearch(VectorStore):
azure_search = cls(
azure_search_endpoint,
azure_search_key,
azure_ad_access_token,
index_name,
embedding,
fields=fields,
azure_ad_access_token=azure_ad_access_token,
**kwargs,
)
await azure_search.aadd_texts(texts, metadatas, **kwargs)