From f3a56181ced908a9a03755dc604d8150dad109b2 Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Tue, 4 Apr 2023 22:21:19 -0300
Subject: [PATCH] fixes

---
 src/backend/langflow/api/base.py                 | 1 -
 src/backend/langflow/interface/base.py           | 1 -
 src/backend/langflow/interface/chains/custom.py  | 4 ++--
 src/backend/langflow/interface/prompts/base.py   | 1 -
 src/backend/langflow/interface/prompts/custom.py | 2 --
 5 files changed, 2 insertions(+), 7 deletions(-)

diff --git a/src/backend/langflow/api/base.py b/src/backend/langflow/api/base.py
index 11b31b004..66afb654b 100644
--- a/src/backend/langflow/api/base.py
+++ b/src/backend/langflow/api/base.py
@@ -1,6 +1,5 @@
 from pydantic import BaseModel, validator
 
-from langflow.graph.utils import extract_input_variables_from_prompt
 
 
 class Code(BaseModel):
diff --git a/src/backend/langflow/interface/base.py b/src/backend/langflow/interface/base.py
index 24293a478..2694fd7a5 100644
--- a/src/backend/langflow/interface/base.py
+++ b/src/backend/langflow/interface/base.py
@@ -1,4 +1,3 @@
-import abc
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional, Type, Union
 
diff --git a/src/backend/langflow/interface/chains/custom.py b/src/backend/langflow/interface/chains/custom.py
index 8a191f594..98470d54b 100644
--- a/src/backend/langflow/interface/chains/custom.py
+++ b/src/backend/langflow/interface/chains/custom.py
@@ -81,7 +81,7 @@ class MidJourneyPromptChain(BaseCustomChain):
     Current conversation:
     {history}
     Human: {input}
-    AI:"""
+    AI:"""  # noqa: E501
 
 
 class TimeTravelGuideChain(BaseCustomChain):
@@ -91,7 +91,7 @@ class TimeTravelGuideChain(BaseCustomChain):
     Current conversation:
     {history}
     Human: {input}
-    AI:"""
+    AI:"""  # noqa: E501
 
 
 CUSTOM_CHAINS: Dict[str, Type[ConversationChain]] = {
diff --git a/src/backend/langflow/interface/prompts/base.py b/src/backend/langflow/interface/prompts/base.py
index 42942a0dc..6718d03a0 100644
--- a/src/backend/langflow/interface/prompts/base.py
+++ b/src/backend/langflow/interface/prompts/base.py
@@ -1,7 +1,6 @@
 from typing import Dict, List, Optional, Type
 
 from langchain import prompts
-from langchain.prompts import loading
 
 from langflow.custom.customs import get_custom_nodes
 from langflow.interface.base import LangChainTypeCreator
diff --git a/src/backend/langflow/interface/prompts/custom.py b/src/backend/langflow/interface/prompts/custom.py
index e80d62312..b1dbef370 100644
--- a/src/backend/langflow/interface/prompts/custom.py
+++ b/src/backend/langflow/interface/prompts/custom.py
@@ -4,8 +4,6 @@ from langchain.prompts import PromptTemplate
 from pydantic import root_validator
 
 from langflow.graph.utils import extract_input_variables_from_prompt
-from langflow.template.base import Template, TemplateField
-from langflow.template.nodes import PromptTemplateNode
 
 # Steps to create a BaseCustomPrompt:
 # 1. Create a prompt template that endes with: