From b51a1eba4d8d033b435cd2126cb6027e63ec362e Mon Sep 17 00:00:00 2001 From: ccurme Date: Wed, 22 May 2024 14:22:50 -0700 Subject: [PATCH] langchain, community: move OpenAIAssistantV2Runnable to community (#22044) --- libs/community/langchain_community/agents/__init__.py | 0 .../langchain_community/agents/openai_assistant/__init__.py | 3 +++ .../langchain_community/agents/openai_assistant/base.py} | 3 +-- libs/langchain/langchain/agents/openai_assistant/__init__.py | 3 +-- 4 files changed, 5 insertions(+), 4 deletions(-) create mode 100644 libs/community/langchain_community/agents/__init__.py create mode 100644 libs/community/langchain_community/agents/openai_assistant/__init__.py rename libs/{langchain/langchain/agents/openai_assistant/base_v2.py => community/langchain_community/agents/openai_assistant/base.py} (99%) diff --git a/libs/community/langchain_community/agents/__init__.py b/libs/community/langchain_community/agents/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/libs/community/langchain_community/agents/openai_assistant/__init__.py b/libs/community/langchain_community/agents/openai_assistant/__init__.py new file mode 100644 index 0000000000..f7fdcbdd86 --- /dev/null +++ b/libs/community/langchain_community/agents/openai_assistant/__init__.py @@ -0,0 +1,3 @@ +from langchain_community.agents.openai_assistant.base import OpenAIAssistantV2Runnable + +__all__ = ["OpenAIAssistantV2Runnable"] diff --git a/libs/langchain/langchain/agents/openai_assistant/base_v2.py b/libs/community/langchain_community/agents/openai_assistant/base.py similarity index 99% rename from libs/langchain/langchain/agents/openai_assistant/base_v2.py rename to libs/community/langchain_community/agents/openai_assistant/base.py index 53fe1ce491..fc977d7c00 100644 --- a/libs/langchain/langchain/agents/openai_assistant/base_v2.py +++ b/libs/community/langchain_community/agents/openai_assistant/base.py @@ -11,6 +11,7 @@ from typing import ( Union, ) +from langchain.agents.openai_assistant.base import OpenAIAssistantRunnable, OutputType from langchain_core._api import beta from langchain_core.callbacks import CallbackManager from langchain_core.load import dumpd @@ -19,8 +20,6 @@ from langchain_core.runnables import RunnableConfig, ensure_config from langchain_core.tools import BaseTool from langchain_core.utils.function_calling import convert_to_openai_tool -from langchain.agents.openai_assistant.base import OpenAIAssistantRunnable, OutputType - if TYPE_CHECKING: import openai from openai._types import NotGiven diff --git a/libs/langchain/langchain/agents/openai_assistant/__init__.py b/libs/langchain/langchain/agents/openai_assistant/__init__.py index eb48ee098c..265c0ed88f 100644 --- a/libs/langchain/langchain/agents/openai_assistant/__init__.py +++ b/libs/langchain/langchain/agents/openai_assistant/__init__.py @@ -1,4 +1,3 @@ from langchain.agents.openai_assistant.base import OpenAIAssistantRunnable -from langchain.agents.openai_assistant.base_v2 import OpenAIAssistantV2Runnable -__all__ = ["OpenAIAssistantRunnable", "OpenAIAssistantV2Runnable"] +__all__ = ["OpenAIAssistantRunnable"]