linting

langchain-ai · Dec 14, 2024 · 0d2ef2e · 0d2ef2e
1 parent fe24791
commit 0d2ef2e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 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
@@ -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
@@ -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"}],