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

community[minor]: Adds a vector store for Azure Cosmos DB for NoSQL (#21676)

Browse Source 
This PR add supports for Azure Cosmos DB for NoSQL vector store.

Summary:

Description: added vector store integration for Azure Cosmos DB for
NoSQL Vector Store,
Dependencies: azure-cosmos dependency,
Tag maintainer: @hwchase17, @baskaryan @efriis @eyurtsev

---------

Co-authored-by: Bagatur <baskaryan@gmail.com>
Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com>
This commit is contained in:
Aayush Kataria
2024-06-11 10:34:01 -07:00
committed by GitHub
parent 36cad5d25c
commit 71811e0547
14 changed files with 917 additions and 97 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
libs/community/langchain_community/utilities/rememberizer.py
View File

@@ -1,5 +1,5 @@
"""Wrapper for Rememberizer APIs."""
from typing import Dict, List, Optional
from typing import Dict, List, Optional, cast
import requests
from langchain_core.documents import Document
@@ -26,7 +26,9 @@ class RememberizerAPIWrapper(BaseModel):
def search(self, query: str) -> dict:
"""Search for a query in the Rememberizer API."""
url = f"https://api.rememberizer.ai/api/v1/documents/search?q={query}&n={self.top_k_results}"
response = requests.get(url, headers={"x-api-key": self.rememberizer_api_key})
response = requests.get(
url, headers={"x-api-key": cast(str, self.rememberizer_api_key)}
)
data = response.json()
if response.status_code != 200: