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 a22041fd56c..e99ab08f069 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 044a2b7eb6b..1b57ef55a70 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 670576ae146..9b8cfba0081 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 e47419e808c..110a5313b59 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 5069a0b7c4d..db677ec24cc 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 f95576675e0..814fac0232f 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 ba1621a9d26..61051d047c1 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 6a13727e357..b8ed3253ba9 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 f54dbdc0d4a..78d11a10aac 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 06589804bb6..8514e5ce2ad 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