From d9ce5f446489196fc760ac233266ca3dceee8481 Mon Sep 17 00:00:00 2001 From: Chester Curme Date: Mon, 3 Feb 2025 14:43:05 -0500 Subject: [PATCH] lint --- libs/langchain/langchain/agents/openai_assistant/base.py | 8 +++++--- libs/langchain/langchain/chains/flare/base.py | 4 +++- libs/langchain/langchain/chains/moderation.py | 2 +- .../langchain/evaluation/embedding_distance/base.py | 8 ++++++-- libs/langchain/langchain/evaluation/loading.py | 4 +++- 5 files changed, 18 insertions(+), 8 deletions(-) diff --git a/libs/langchain/langchain/agents/openai_assistant/base.py b/libs/langchain/langchain/agents/openai_assistant/base.py index 148cfc85ab5..4137b949be5 100644 --- a/libs/langchain/langchain/agents/openai_assistant/base.py +++ b/libs/langchain/langchain/agents/openai_assistant/base.py @@ -28,7 +28,7 @@ from typing_extensions import Self if TYPE_CHECKING: import openai - from openai.types.beta.threads import ThreadMessage + from openai.types.beta.threads import ThreadMessage # type: ignore[attr-defined] from openai.types.beta.threads.required_action_function_tool_call import ( RequiredActionFunctionToolCall, ) @@ -590,7 +590,8 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]): isinstance(content, openai.types.beta.threads.TextContentBlock) if version_gte_1_14 else isinstance( - content, openai.types.beta.threads.MessageContentText + content, + openai.types.beta.threads.MessageContentText, # type: ignore[attr-defined] ) ) for content in answer @@ -743,7 +744,8 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]): isinstance(content, openai.types.beta.threads.TextContentBlock) if version_gte_1_14 else isinstance( - content, openai.types.beta.threads.MessageContentText + content, + openai.types.beta.threads.MessageContentText, # type: ignore[attr-defined] ) ) for content in answer diff --git a/libs/langchain/langchain/chains/flare/base.py b/libs/langchain/langchain/chains/flare/base.py index caf64fe77aa..430ad0baf36 100644 --- a/libs/langchain/langchain/chains/flare/base.py +++ b/libs/langchain/langchain/chains/flare/base.py @@ -236,7 +236,9 @@ class FlareChain(Chain): "Please install langchain-openai." "pip install langchain-openai" ) - llm = ChatOpenAI(max_tokens=max_generation_len, logprobs=True, temperature=0) + llm = ChatOpenAI( + max_completion_tokens=max_generation_len, logprobs=True, temperature=0 + ) response_chain = PROMPT | llm question_gen_chain = QUESTION_GENERATOR_PROMPT | llm | StrOutputParser() return cls( diff --git a/libs/langchain/langchain/chains/moderation.py b/libs/langchain/langchain/chains/moderation.py index f72cae6cc8d..f7175d45a91 100644 --- a/libs/langchain/langchain/chains/moderation.py +++ b/libs/langchain/langchain/chains/moderation.py @@ -65,7 +65,7 @@ class OpenAIModerationChain(Chain): except ValueError: values["openai_pre_1_0"] = True if values["openai_pre_1_0"]: - values["client"] = openai.Moderation + values["client"] = openai.Moderation # type: ignore[attr-defined] else: values["client"] = openai.OpenAI(api_key=openai_api_key) values["async_client"] = openai.AsyncOpenAI(api_key=openai_api_key) diff --git a/libs/langchain/langchain/evaluation/embedding_distance/base.py b/libs/langchain/langchain/evaluation/embedding_distance/base.py index 3afd3030bfe..4a1340d0695 100644 --- a/libs/langchain/langchain/evaluation/embedding_distance/base.py +++ b/libs/langchain/langchain/evaluation/embedding_distance/base.py @@ -30,7 +30,9 @@ def _embedding_factory() -> Embeddings: from langchain_openai import OpenAIEmbeddings except ImportError: try: - from langchain_community.embeddings.openai import OpenAIEmbeddings + from langchain_community.embeddings.openai import ( # type: ignore[no-redef] + OpenAIEmbeddings, + ) except ImportError: raise ImportError( "Could not import OpenAIEmbeddings. Please install the " @@ -89,7 +91,9 @@ class _EmbeddingDistanceChainMixin(Chain): pass try: - from langchain_community.embeddings.openai import OpenAIEmbeddings + from langchain_community.embeddings.openai import ( # type: ignore[no-redef] + OpenAIEmbeddings, + ) types_.append(OpenAIEmbeddings) except ImportError: diff --git a/libs/langchain/langchain/evaluation/loading.py b/libs/langchain/langchain/evaluation/loading.py index a9a576911bf..9e6c8b7fb5b 100644 --- a/libs/langchain/langchain/evaluation/loading.py +++ b/libs/langchain/langchain/evaluation/loading.py @@ -135,7 +135,9 @@ def load_evaluator( from langchain_openai import ChatOpenAI except ImportError: try: - from langchain_community.chat_models.openai import ChatOpenAI + from langchain_community.chat_models.openai import ( # type: ignore[no-redef] + ChatOpenAI, + ) except ImportError: raise ImportError( "Could not import langchain_openai or fallback onto "