chore(langchain): add mypy pydantic plugin (#32610)

Author: cbornet

libs/langchain/langchain/agents/openai_functions_agent/base.py

@@ -277,7 +277,7 @@ def from_llm_and_tools(
             extra_prompt_messages=extra_prompt_messages,
             system_message=system_message_,
         )
-        return cls(  # type: ignore[call-arg]
+        return cls(
             llm=llm,
             prompt=prompt,
             tools=tools,

libs/langchain/langchain/agents/openai_functions_multi_agent/base.py

@@ -328,7 +328,7 @@ def from_llm_and_tools(
             extra_prompt_messages=extra_prompt_messages,
             system_message=system_message_,
         )
-        return cls(  # type: ignore[call-arg]
+        return cls(
             llm=llm,
             prompt=prompt,
             tools=tools,

libs/langchain/langchain/chains/llm_checker/base.py

@@ -56,7 +56,7 @@ def _load_question_to_checked_assertions_chain(
         revised_answer_chain,
     ]
     return SequentialChain(
-        chains=chains,  # type: ignore[arg-type]
+        chains=chains,
         input_variables=["question"],
         output_variables=["revised_statement"],
         verbose=True,

libs/langchain/langchain/chains/loading.py

@@ -169,7 +169,7 @@ def _load_map_reduce_documents_chain(
 
     return MapReduceDocumentsChain(
         llm_chain=llm_chain,
-        reduce_documents_chain=reduce_documents_chain,  # type: ignore[arg-type]
+        reduce_documents_chain=reduce_documents_chain,
         **config,
     )
 
@@ -293,10 +293,10 @@ def _load_llm_checker_chain(config: dict, **kwargs: Any) -> LLMCheckerChain:
         revised_answer_prompt = load_prompt(config.pop("revised_answer_prompt_path"))
     return LLMCheckerChain(
         llm=llm,
-        create_draft_answer_prompt=create_draft_answer_prompt,  # type: ignore[arg-type]
-        list_assertions_prompt=list_assertions_prompt,  # type: ignore[arg-type]
-        check_assertions_prompt=check_assertions_prompt,  # type: ignore[arg-type]
-        revised_answer_prompt=revised_answer_prompt,  # type: ignore[arg-type]
+        create_draft_answer_prompt=create_draft_answer_prompt,
+        list_assertions_prompt=list_assertions_prompt,
+        check_assertions_prompt=check_assertions_prompt,
+        revised_answer_prompt=revised_answer_prompt,
         **config,
     )
 
@@ -325,7 +325,7 @@ def _load_llm_math_chain(config: dict, **kwargs: Any) -> LLMMathChain:
     elif "prompt_path" in config:
         prompt = load_prompt(config.pop("prompt_path"))
     if llm_chain:
-        return LLMMathChain(llm_chain=llm_chain, prompt=prompt, **config)  # type: ignore[arg-type]
+        return LLMMathChain(llm_chain=llm_chain, prompt=prompt, **config)
     return LLMMathChain(llm=llm, prompt=prompt, **config)
 
 
@@ -341,7 +341,7 @@ def _load_map_rerank_documents_chain(
     else:
         msg = "One of `llm_chain` or `llm_chain_path` must be present."
         raise ValueError(msg)
-    return MapRerankDocumentsChain(llm_chain=llm_chain, **config)  # type: ignore[arg-type]
+    return MapRerankDocumentsChain(llm_chain=llm_chain, **config)
 
 
 def _load_pal_chain(config: dict, **kwargs: Any) -> Any:
@@ -377,8 +377,8 @@ def _load_refine_documents_chain(config: dict, **kwargs: Any) -> RefineDocuments
     elif "document_prompt_path" in config:
         document_prompt = load_prompt(config.pop("document_prompt_path"))
     return RefineDocumentsChain(
-        initial_llm_chain=initial_llm_chain,  # type: ignore[arg-type]
-        refine_llm_chain=refine_llm_chain,  # type: ignore[arg-type]
+        initial_llm_chain=initial_llm_chain,
+        refine_llm_chain=refine_llm_chain,
         document_prompt=document_prompt,
         **config,
     )
@@ -402,7 +402,7 @@ def _load_qa_with_sources_chain(config: dict, **kwargs: Any) -> QAWithSourcesCha
             "`combine_documents_chain_path` must be present."
         )
         raise ValueError(msg)
-    return QAWithSourcesChain(combine_documents_chain=combine_documents_chain, **config)  # type: ignore[arg-type]
+    return QAWithSourcesChain(combine_documents_chain=combine_documents_chain, **config)
 
 
 def _load_sql_database_chain(config: dict, **kwargs: Any) -> Any:
@@ -445,7 +445,7 @@ def _load_vector_db_qa_with_sources_chain(
         )
         raise ValueError(msg)
     return VectorDBQAWithSourcesChain(
-        combine_documents_chain=combine_documents_chain,  # type: ignore[arg-type]
+        combine_documents_chain=combine_documents_chain,
         vectorstore=vectorstore,
         **config,
     )
@@ -475,7 +475,7 @@ def _load_retrieval_qa(config: dict, **kwargs: Any) -> RetrievalQA:
         )
         raise ValueError(msg)
     return RetrievalQA(
-        combine_documents_chain=combine_documents_chain,  # type: ignore[arg-type]
+        combine_documents_chain=combine_documents_chain,
         retriever=retriever,
         **config,
     )
@@ -508,7 +508,7 @@ def _load_retrieval_qa_with_sources_chain(
         )
         raise ValueError(msg)
     return RetrievalQAWithSourcesChain(
-        combine_documents_chain=combine_documents_chain,  # type: ignore[arg-type]
+        combine_documents_chain=combine_documents_chain,
         retriever=retriever,
         **config,
     )
@@ -538,7 +538,7 @@ def _load_vector_db_qa(config: dict, **kwargs: Any) -> VectorDBQA:
         )
         raise ValueError(msg)
     return VectorDBQA(
-        combine_documents_chain=combine_documents_chain,  # type: ignore[arg-type]
+        combine_documents_chain=combine_documents_chain,
         vectorstore=vectorstore,
         **config,
     )
@@ -606,8 +606,8 @@ def _load_api_chain(config: dict, **kwargs: Any) -> APIChain:
         msg = "`requests_wrapper` must be present."
         raise ValueError(msg)
     return APIChain(
-        api_request_chain=api_request_chain,  # type: ignore[arg-type]
-        api_answer_chain=api_answer_chain,  # type: ignore[arg-type]
+        api_request_chain=api_request_chain,
+        api_answer_chain=api_answer_chain,
         requests_wrapper=requests_wrapper,
         **config,
     )

libs/langchain/langchain/chains/qa_with_sources/loading.py

@@ -66,12 +66,12 @@ def _load_stuff_chain(
     verbose: Optional[bool] = None,
     **kwargs: Any,
 ) -> StuffDocumentsChain:
-    llm_chain = LLMChain(llm=llm, prompt=prompt, verbose=verbose)  # type: ignore[arg-type]
+    llm_chain = LLMChain(llm=llm, prompt=prompt, verbose=verbose)
     return StuffDocumentsChain(
         llm_chain=llm_chain,
         document_variable_name=document_variable_name,
         document_prompt=document_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         **kwargs,
     )
 
@@ -91,14 +91,14 @@ def _load_map_reduce_chain(
     token_max: int = 3000,
     **kwargs: Any,
 ) -> MapReduceDocumentsChain:
-    map_chain = LLMChain(llm=llm, prompt=question_prompt, verbose=verbose)  # type: ignore[arg-type]
+    map_chain = LLMChain(llm=llm, prompt=question_prompt, verbose=verbose)
     _reduce_llm = reduce_llm or llm
-    reduce_chain = LLMChain(llm=_reduce_llm, prompt=combine_prompt, verbose=verbose)  # type: ignore[arg-type]
+    reduce_chain = LLMChain(llm=_reduce_llm, prompt=combine_prompt, verbose=verbose)
     combine_documents_chain = StuffDocumentsChain(
         llm_chain=reduce_chain,
         document_variable_name=combine_document_variable_name,
         document_prompt=document_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
     )
     if collapse_prompt is None:
         collapse_chain = None
@@ -114,7 +114,7 @@ def _load_map_reduce_chain(
             llm_chain=LLMChain(
                 llm=_collapse_llm,
                 prompt=collapse_prompt,
-                verbose=verbose,  # type: ignore[arg-type]
+                verbose=verbose,
             ),
             document_variable_name=combine_document_variable_name,
             document_prompt=document_prompt,
@@ -123,13 +123,13 @@ def _load_map_reduce_chain(
         combine_documents_chain=combine_documents_chain,
         collapse_documents_chain=collapse_chain,
         token_max=token_max,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
     )
     return MapReduceDocumentsChain(
         llm_chain=map_chain,
         reduce_documents_chain=reduce_documents_chain,
         document_variable_name=map_reduce_document_variable_name,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         **kwargs,
     )
 
@@ -146,16 +146,16 @@ def _load_refine_chain(
     verbose: Optional[bool] = None,
     **kwargs: Any,
 ) -> RefineDocumentsChain:
-    initial_chain = LLMChain(llm=llm, prompt=question_prompt, verbose=verbose)  # type: ignore[arg-type]
+    initial_chain = LLMChain(llm=llm, prompt=question_prompt, verbose=verbose)
     _refine_llm = refine_llm or llm
-    refine_chain = LLMChain(llm=_refine_llm, prompt=refine_prompt, verbose=verbose)  # type: ignore[arg-type]
+    refine_chain = LLMChain(llm=_refine_llm, prompt=refine_prompt, verbose=verbose)
     return RefineDocumentsChain(
         initial_llm_chain=initial_chain,
         refine_llm_chain=refine_chain,
         document_variable_name=document_variable_name,
         initial_response_name=initial_response_name,
         document_prompt=document_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         **kwargs,
     )
 

libs/langchain/langchain/chains/question_answering/chain.py

@@ -80,15 +80,15 @@ def _load_stuff_chain(
     llm_chain = LLMChain(
         llm=llm,
         prompt=_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callback_manager=callback_manager,
         callbacks=callbacks,
     )
     # TODO: document prompt
     return StuffDocumentsChain(
         llm_chain=llm_chain,
         document_variable_name=document_variable_name,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callback_manager=callback_manager,
         callbacks=callbacks,
         **kwargs,
@@ -120,23 +120,23 @@ def _load_map_reduce_chain(
     map_chain = LLMChain(
         llm=llm,
         prompt=_question_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callback_manager=callback_manager,
         callbacks=callbacks,
     )
     _reduce_llm = reduce_llm or llm
     reduce_chain = LLMChain(
         llm=_reduce_llm,
         prompt=_combine_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callback_manager=callback_manager,
         callbacks=callbacks,
     )
     # TODO: document prompt
     combine_documents_chain = StuffDocumentsChain(
         llm_chain=reduce_chain,
         document_variable_name=combine_document_variable_name,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callback_manager=callback_manager,
         callbacks=callbacks,
     )
@@ -154,12 +154,12 @@ def _load_map_reduce_chain(
             llm_chain=LLMChain(
                 llm=_collapse_llm,
                 prompt=collapse_prompt,
-                verbose=verbose,  # type: ignore[arg-type]
+                verbose=verbose,
                 callback_manager=callback_manager,
                 callbacks=callbacks,
             ),
             document_variable_name=combine_document_variable_name,
-            verbose=verbose,  # type: ignore[arg-type]
+            verbose=verbose,
             callback_manager=callback_manager,
         )
     reduce_documents_chain = ReduceDocumentsChain(
@@ -172,7 +172,7 @@ def _load_map_reduce_chain(
         llm_chain=map_chain,
         document_variable_name=map_reduce_document_variable_name,
         reduce_documents_chain=reduce_documents_chain,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callback_manager=callback_manager,
         callbacks=callbacks,
         **kwargs,
@@ -201,15 +201,15 @@ def _load_refine_chain(
     initial_chain = LLMChain(
         llm=llm,
         prompt=_question_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callback_manager=callback_manager,
         callbacks=callbacks,
     )
     _refine_llm = refine_llm or llm
     refine_chain = LLMChain(
         llm=_refine_llm,
         prompt=_refine_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callback_manager=callback_manager,
         callbacks=callbacks,
     )
@@ -218,7 +218,7 @@ def _load_refine_chain(
         refine_llm_chain=refine_chain,
         document_variable_name=document_variable_name,
         initial_response_name=initial_response_name,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callback_manager=callback_manager,
         callbacks=callbacks,
         **kwargs,

libs/langchain/langchain/chains/summarize/chain.py

@@ -35,12 +35,12 @@ def _load_stuff_chain(
     verbose: Optional[bool] = None,
     **kwargs: Any,
 ) -> StuffDocumentsChain:
-    llm_chain = LLMChain(llm=llm, prompt=prompt, verbose=verbose)  # type: ignore[arg-type]
+    llm_chain = LLMChain(llm=llm, prompt=prompt, verbose=verbose)
     # TODO: document prompt
     return StuffDocumentsChain(
         llm_chain=llm_chain,
         document_variable_name=document_variable_name,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         **kwargs,
     )
 
@@ -64,21 +64,21 @@ def _load_map_reduce_chain(
     map_chain = LLMChain(
         llm=llm,
         prompt=map_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callbacks=callbacks,
     )
     _reduce_llm = reduce_llm or llm
     reduce_chain = LLMChain(
         llm=_reduce_llm,
         prompt=combine_prompt,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callbacks=callbacks,
     )
     # TODO: document prompt
     combine_documents_chain = StuffDocumentsChain(
         llm_chain=reduce_chain,
         document_variable_name=combine_document_variable_name,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callbacks=callbacks,
     )
     if collapse_prompt is None:
@@ -95,7 +95,7 @@ def _load_map_reduce_chain(
             llm_chain=LLMChain(
                 llm=_collapse_llm,
                 prompt=collapse_prompt,
-                verbose=verbose,  # type: ignore[arg-type]
+                verbose=verbose,
                 callbacks=callbacks,
             ),
             document_variable_name=combine_document_variable_name,
@@ -104,15 +104,15 @@ def _load_map_reduce_chain(
         combine_documents_chain=combine_documents_chain,
         collapse_documents_chain=collapse_chain,
         token_max=token_max,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callbacks=callbacks,
         collapse_max_retries=collapse_max_retries,
     )
     return MapReduceDocumentsChain(
         llm_chain=map_chain,
         reduce_documents_chain=reduce_documents_chain,
         document_variable_name=map_reduce_document_variable_name,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         callbacks=callbacks,
         **kwargs,
     )
@@ -129,15 +129,15 @@ def _load_refine_chain(
     verbose: Optional[bool] = None,
     **kwargs: Any,
 ) -> RefineDocumentsChain:
-    initial_chain = LLMChain(llm=llm, prompt=question_prompt, verbose=verbose)  # type: ignore[arg-type]
+    initial_chain = LLMChain(llm=llm, prompt=question_prompt, verbose=verbose)
     _refine_llm = refine_llm or llm
-    refine_chain = LLMChain(llm=_refine_llm, prompt=refine_prompt, verbose=verbose)  # type: ignore[arg-type]
+    refine_chain = LLMChain(llm=_refine_llm, prompt=refine_prompt, verbose=verbose)
     return RefineDocumentsChain(
         initial_llm_chain=initial_chain,
         refine_llm_chain=refine_chain,
         document_variable_name=document_variable_name,
         initial_response_name=initial_response_name,
-        verbose=verbose,  # type: ignore[arg-type]
+        verbose=verbose,
         **kwargs,
     )