From 0d2ef2e31c96acd5d28da8f4d4d8780f186ff271 Mon Sep 17 00:00:00 2001 From: Aayush Kataria Date: Fri, 13 Dec 2024 16:47:23 -0800 Subject: [PATCH] linting --- .../langchain_community/vectorstores/azure_cosmos_db_no_sql.py | 2 +- .../vectorstores/test_azure_cosmos_db_no_sql.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/community/langchain_community/vectorstores/azure_cosmos_db_no_sql.py b/libs/community/langchain_community/vectorstores/azure_cosmos_db_no_sql.py index 2d5d0d302317f..097df0a333e55 100644 --- a/libs/community/langchain_community/vectorstores/azure_cosmos_db_no_sql.py +++ b/libs/community/langchain_community/vectorstores/azure_cosmos_db_no_sql.py @@ -552,7 +552,7 @@ def _construct_query( query_type: CosmosDBQueryType, embeddings: Optional[List[float]] = None, search_text: Optional[str] = None, - pre_filter: Optional[Dict] = None, + pre_filter: Optional[Dict[str, Any]] = None, offset_limit: Optional[str] = None, projection_mapping: Optional[Dict[str, Any]] = None, ) -> Tuple[str, List[Dict[str, Any]]]: diff --git a/libs/community/tests/integration_tests/vectorstores/test_azure_cosmos_db_no_sql.py b/libs/community/tests/integration_tests/vectorstores/test_azure_cosmos_db_no_sql.py index f5c3707a0f346..54b1336e9a97a 100644 --- a/libs/community/tests/integration_tests/vectorstores/test_azure_cosmos_db_no_sql.py +++ b/libs/community/tests/integration_tests/vectorstores/test_azure_cosmos_db_no_sql.py @@ -80,7 +80,7 @@ def get_vector_embedding_policy( } -def get_full_text_policy(): +def get_full_text_policy() -> dict: return { "defaultLanguage": "en-US", "fullTextPaths": [{"path": "/text", "language": "en-US"}],