diff --git a/pyproject.toml b/pyproject.toml index fa258fbe0..d0e0d5cb4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "langflow" -version = "0.5.0a2" +version = "0.5.0a3" description = "A Python package with a built-in web application" authors = ["Logspace "] maintainers = [ diff --git a/src/backend/langflow/components/chains/PromptRunner.py b/src/backend/langflow/components/chains/PromptRunner.py index db9283b07..96a64f6fc 100644 --- a/src/backend/langflow/components/chains/PromptRunner.py +++ b/src/backend/langflow/components/chains/PromptRunner.py @@ -1,7 +1,7 @@ from langflow import CustomComponent from langchain.llms.base import BaseLLM -from langchain import PromptTemplate +from langchain.prompts import PromptTemplate from langchain.schema import Document diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py index b4e2b9bac..735b27917 100644 --- a/src/backend/langflow/interface/agents/custom.py +++ b/src/backend/langflow/interface/agents/custom.py @@ -1,6 +1,6 @@ from typing import Any, List, Optional -from langchain import LLMChain +from langchain.chains.llm import LLMChain from langchain.agents import ( AgentExecutor, Tool, diff --git a/src/backend/langflow/interface/agents/prebuilt.py b/src/backend/langflow/interface/agents/prebuilt.py index 5b81b7713..ec4799a81 100644 --- a/src/backend/langflow/interface/agents/prebuilt.py +++ b/src/backend/langflow/interface/agents/prebuilt.py @@ -1,4 +1,4 @@ -from langchain import LLMChain +from langchain.chains.llm import LLMChain from langchain.agents import AgentExecutor, ZeroShotAgent from langchain.agents.agent_toolkits.json.prompt import JSON_PREFIX, JSON_SUFFIX from langchain.agents.agent_toolkits.json.toolkit import JsonToolkit