Use mypy pydantic plugin

Author: cbornet

langchain/langchain_vectorize/retrievers.py

@@ -160,7 +160,7 @@ def _get_relevant_documents(
         rerank: bool | None = None,
         metadata_filters: list[dict[str, Any]] | None = None,
     ) -> list[Document]:
-        request = RetrieveDocumentsRequest(  # type: ignore[call-arg]
+        request = RetrieveDocumentsRequest(
             question=query,
             num_results=num_results or self.num_results,
             rerank=rerank or self.rerank,

langchain/pyproject.toml

@@ -62,6 +62,7 @@ pyupgrade.keep-runtime-typing = true
 ]
 
 [tool.mypy]
+plugins = ["pydantic.mypy"]
 strict = true
 strict_bytes = true
 enable_error_code = "deprecated"

langchain/tests/conftest.py

@@ -116,7 +116,7 @@ def pipeline_id(api_client: ApiClient, org_id: str) -> Iterator[str]:
     upload_response = uploads_api.start_file_upload_to_connector(
         org_id,
         source_connector_id,
-        StartFileUploadToConnectorRequest(  # type: ignore[call-arg]
+        StartFileUploadToConnectorRequest(
             name="research.pdf",
             content_type="application/pdf",
             metadata=json.dumps({"created-from-api": True}),
@@ -157,7 +157,7 @@ def pipeline_id(api_client: ApiClient, org_id: str) -> Iterator[str]:
 
     pipeline_response = pipelines.create_pipeline(
         org_id,
-        PipelineConfigurationSchema(  # type: ignore[call-arg]
+        PipelineConfigurationSchema(
             source_connectors=[
                 PipelineSourceConnectorSchema(
                     id=source_connector_id,
@@ -182,7 +182,7 @@ def pipeline_id(api_client: ApiClient, org_id: str) -> Iterator[str]:
     pipeline_id = pipeline_response.data.id
 
     # Wait for the pipeline to be created
-    request = RetrieveDocumentsRequest(  # type: ignore[call-arg]
+    request = RetrieveDocumentsRequest(
         question="query",
         num_results=2,
     )