Skip to content

Commit

Permalink
linting
Browse files Browse the repository at this point in the history
  • Loading branch information
aayush3011 committed Dec 14, 2024
1 parent 1899a55 commit b92ce90
Showing 1 changed file with 17 additions and 17 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -231,7 +231,7 @@ def _from_kwargs(
indexing_policy: Dict[str, Any],
cosmos_container_properties: Dict[str, Any],
cosmos_database_properties: Dict[str, Any],
full_text_policy: Dict[str, Any] = None,
full_text_policy: Optional[Dict[str, Any]] = None,
database_name: str = "vectorSearchDB",
container_name: str = "vectorSearchContainer",
text_key: str = "text",
Expand Down Expand Up @@ -319,11 +319,11 @@ def _similarity_search_with_score(
query_type: CosmosDBQueryType,
embeddings: List[float],
k: int = 4,
pre_filter: Optional[Dict] = None,
pre_filter: Optional[Dict[str, Any]] = None,
with_embedding: bool = False,
offset_limit: str = None,
offset_limit: Optional[str] = None,
*,
projection_mapping: Dict[str, Any] = None,
projection_mapping: Optional[Dict[str, Any]] = None,
**kwargs: Any,
) -> List[Tuple[Document, float]]:
query, parameters = self._construct_query(
Expand All @@ -346,12 +346,12 @@ def _similarity_search_with_score(
def _full_text_search(
self,
query_type: CosmosDBQueryType,
search_text: str = None,
search_text: Optional[str] = None,
k: int = 4,
pre_filter: Optional[Dict] = None,
offset_limit: str = None,
pre_filter: Optional[Dict[str, Any]] = None,
offset_limit: Optional[str] = None,
*,
projection_mapping: Dict[str, Any] = None,
projection_mapping: Optional[Dict[str, Any]] = None,
**kwargs: Any,
) -> List[Tuple[Document, float]]:
query, parameters = self._construct_query(
Expand All @@ -377,11 +377,11 @@ def _hybrid_search_with_score(
embeddings: List[float],
search_text: str,
k: int = 4,
pre_filter: Dict[str, Any] = None,
pre_filter: Optional[Dict[str, Any]] = None,
with_embedding: bool = False,
offset_limit: str = None,
offset_limit: str = "",
*,
projection_mapping: Dict[str, Any] = None,
projection_mapping: Optional[Dict[str, Any]] = None,
**kwargs: Any,
) -> List[Tuple[Document, float]]:
query, parameters = self._construct_query(
Expand All @@ -405,10 +405,10 @@ def similarity_search_with_score(
self,
query: str,
k: int = 4,
pre_filter: Optional[Dict] = None,
pre_filter: Optional[Dict[str, Any]] = None,
with_embedding: bool = False,
query_type: CosmosDBQueryType = CosmosDBQueryType.VECTOR,
offset_limit: str = None,
offset_limit: Optional[str] = None,
**kwargs: Any,
) -> List[Tuple[Document, float]]:
embeddings = self._embedding.embed_query(query)
Expand Down Expand Up @@ -458,10 +458,10 @@ def similarity_search(
self,
query: str,
k: int = 4,
pre_filter: Optional[Dict] = None,
pre_filter: Optional[Dict[str, Any]] = None,
with_embedding: bool = False,
query_type: CosmosDBQueryType = CosmosDBQueryType.VECTOR,
offset_limit: str = None,
offset_limit: Optional[str] = None,
**kwargs: Any,
) -> List[Document]:
if query_type not in CosmosDBQueryType.__members__.values():
Expand Down Expand Up @@ -551,9 +551,9 @@ def _construct_query(
k: int,
query_type: CosmosDBQueryType,
embeddings: Optional[List[float]] = None,
search_text: str = None,
search_text: Optional[str] = None,
pre_filter: Optional[Dict] = None,
offset_limit: str = None,
offset_limit: Optional[str] = None,
projection_mapping: Optional[Dict[str, Any]] = None,
) -> Tuple[str, List[Dict[str, Any]]]:
if (
Expand Down

0 comments on commit b92ce90

Please sign in to comment.