From 3dc0f3c371819afa378878bc8ed2d141243cc1b4 Mon Sep 17 00:00:00 2001 From: Leonid Ganeline Date: Tue, 26 Mar 2024 17:03:13 -0700 Subject: [PATCH] experimental[patch]: `PromptTemplate` import fix (#19617) Changed import of `PromptTemplate` from `langchain` to `langchain_core` in `langchain_experimental` --- .../autonomous_agents/baby_agi/task_creation.py | 2 +- .../autonomous_agents/baby_agi/task_execution.py | 2 +- .../autonomous_agents/baby_agi/task_prioritization.py | 2 +- .../autonomous_agents/hugginggpt/repsonse_generator.py | 2 +- .../generative_agents/generative_agent.py | 2 +- .../langchain_experimental/generative_agents/memory.py | 2 +- .../langchain_experimental/synthetic_data/__init__.py | 2 +- .../langchain_experimental/tabular_synthetic_data/openai.py | 2 +- libs/experimental/langchain_experimental/tot/prompts.py | 2 +- libs/experimental/tests/unit_tests/test_sql.py | 2 +- 10 files changed, 10 insertions(+), 10 deletions(-) diff --git a/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_creation.py b/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_creation.py index a22041fd56..e99ab08f06 100644 --- a/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_creation.py +++ b/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_creation.py @@ -1,6 +1,6 @@ from langchain.chains import LLMChain -from langchain.prompts import PromptTemplate from langchain_core.language_models import BaseLanguageModel +from langchain_core.prompts import PromptTemplate class TaskCreationChain(LLMChain): diff --git a/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_execution.py b/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_execution.py index 044a2b7eb6..1b57ef55a7 100644 --- a/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_execution.py +++ b/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_execution.py @@ -1,6 +1,6 @@ from langchain.chains import LLMChain -from langchain.prompts import PromptTemplate from langchain_core.language_models import BaseLanguageModel +from langchain_core.prompts import PromptTemplate class TaskExecutionChain(LLMChain): diff --git a/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_prioritization.py b/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_prioritization.py index 670576ae14..9b8cfba008 100644 --- a/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_prioritization.py +++ b/libs/experimental/langchain_experimental/autonomous_agents/baby_agi/task_prioritization.py @@ -1,6 +1,6 @@ from langchain.chains import LLMChain -from langchain.prompts import PromptTemplate from langchain_core.language_models import BaseLanguageModel +from langchain_core.prompts import PromptTemplate class TaskPrioritizationChain(LLMChain): diff --git a/libs/experimental/langchain_experimental/autonomous_agents/hugginggpt/repsonse_generator.py b/libs/experimental/langchain_experimental/autonomous_agents/hugginggpt/repsonse_generator.py index e47419e808..110a5313b5 100644 --- a/libs/experimental/langchain_experimental/autonomous_agents/hugginggpt/repsonse_generator.py +++ b/libs/experimental/langchain_experimental/autonomous_agents/hugginggpt/repsonse_generator.py @@ -3,7 +3,7 @@ from typing import Any, List, Optional from langchain.base_language import BaseLanguageModel from langchain.callbacks.manager import Callbacks from langchain.chains import LLMChain -from langchain.prompts import PromptTemplate +from langchain_core.prompts import PromptTemplate class ResponseGenerationChain(LLMChain): diff --git a/libs/experimental/langchain_experimental/generative_agents/generative_agent.py b/libs/experimental/langchain_experimental/generative_agents/generative_agent.py index 5069a0b7c4..db677ec24c 100644 --- a/libs/experimental/langchain_experimental/generative_agents/generative_agent.py +++ b/libs/experimental/langchain_experimental/generative_agents/generative_agent.py @@ -3,8 +3,8 @@ from datetime import datetime from typing import Any, Dict, List, Optional, Tuple from langchain.chains import LLMChain -from langchain.prompts import PromptTemplate from langchain_core.language_models import BaseLanguageModel +from langchain_core.prompts import PromptTemplate from langchain_experimental.generative_agents.memory import GenerativeAgentMemory from langchain_experimental.pydantic_v1 import BaseModel, Field diff --git a/libs/experimental/langchain_experimental/generative_agents/memory.py b/libs/experimental/langchain_experimental/generative_agents/memory.py index f95576675e..814fac0232 100644 --- a/libs/experimental/langchain_experimental/generative_agents/memory.py +++ b/libs/experimental/langchain_experimental/generative_agents/memory.py @@ -4,11 +4,11 @@ from datetime import datetime from typing import Any, Dict, List, Optional from langchain.chains import LLMChain -from langchain.prompts import PromptTemplate from langchain.retrievers import TimeWeightedVectorStoreRetriever from langchain.schema import BaseMemory, Document from langchain.utils import mock_now from langchain_core.language_models import BaseLanguageModel +from langchain_core.prompts import PromptTemplate logger = logging.getLogger(__name__) diff --git a/libs/experimental/langchain_experimental/synthetic_data/__init__.py b/libs/experimental/langchain_experimental/synthetic_data/__init__.py index ba1621a9d2..61051d047c 100644 --- a/libs/experimental/langchain_experimental/synthetic_data/__init__.py +++ b/libs/experimental/langchain_experimental/synthetic_data/__init__.py @@ -3,8 +3,8 @@ from typing import Any, Dict, List, Optional from langchain.chains.base import Chain from langchain.chains.llm import LLMChain -from langchain.prompts import PromptTemplate from langchain_core.language_models import BaseLanguageModel +from langchain_core.prompts import PromptTemplate from langchain_experimental.synthetic_data.prompts import SENTENCE_PROMPT diff --git a/libs/experimental/langchain_experimental/tabular_synthetic_data/openai.py b/libs/experimental/langchain_experimental/tabular_synthetic_data/openai.py index 6a13727e35..b8ed3253ba 100644 --- a/libs/experimental/langchain_experimental/tabular_synthetic_data/openai.py +++ b/libs/experimental/langchain_experimental/tabular_synthetic_data/openai.py @@ -1,10 +1,10 @@ from typing import Any, Dict, Optional, Type, Union from langchain.chains.openai_functions import create_structured_output_chain -from langchain.prompts import PromptTemplate from langchain.pydantic_v1 import BaseModel from langchain.schema import BaseLLMOutputParser, BasePromptTemplate from langchain_community.chat_models import ChatOpenAI +from langchain_core.prompts import PromptTemplate from langchain_experimental.tabular_synthetic_data.base import SyntheticDataGenerator diff --git a/libs/experimental/langchain_experimental/tot/prompts.py b/libs/experimental/langchain_experimental/tot/prompts.py index f54dbdc0d4..78d11a10aa 100644 --- a/libs/experimental/langchain_experimental/tot/prompts.py +++ b/libs/experimental/langchain_experimental/tot/prompts.py @@ -2,8 +2,8 @@ import json from textwrap import dedent from typing import List -from langchain.prompts import PromptTemplate from langchain_core.output_parsers import BaseOutputParser +from langchain_core.prompts import PromptTemplate from langchain_experimental.tot.thought import ThoughtValidity diff --git a/libs/experimental/tests/unit_tests/test_sql.py b/libs/experimental/tests/unit_tests/test_sql.py index 06589804bb..8514e5ce2a 100644 --- a/libs/experimental/tests/unit_tests/test_sql.py +++ b/libs/experimental/tests/unit_tests/test_sql.py @@ -1,7 +1,7 @@ from langchain.memory import ConversationBufferMemory from langchain.output_parsers.list import CommaSeparatedListOutputParser -from langchain.prompts import PromptTemplate from langchain.sql_database import SQLDatabase +from langchain_core.prompts import PromptTemplate from langchain_experimental.sql.base import SQLDatabaseChain, SQLDatabaseSequentialChain from tests.unit_tests.fake_llm import FakeLLM