This commit is contained in:
Gabriel Almeida 2023-04-04 22:21:19 -03:00
commit f3a56181ce
5 changed files with 2 additions and 7 deletions

View file

@ -1,6 +1,5 @@
from pydantic import BaseModel, validator from pydantic import BaseModel, validator
from langflow.graph.utils import extract_input_variables_from_prompt
class Code(BaseModel): class Code(BaseModel):

View file

@ -1,4 +1,3 @@
import abc
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from typing import Any, Dict, List, Optional, Type, Union from typing import Any, Dict, List, Optional, Type, Union

View file

@ -81,7 +81,7 @@ class MidJourneyPromptChain(BaseCustomChain):
Current conversation: Current conversation:
{history} {history}
Human: {input} Human: {input}
AI:""" AI:""" # noqa: E501
class TimeTravelGuideChain(BaseCustomChain): class TimeTravelGuideChain(BaseCustomChain):
@ -91,7 +91,7 @@ class TimeTravelGuideChain(BaseCustomChain):
Current conversation: Current conversation:
{history} {history}
Human: {input} Human: {input}
AI:""" AI:""" # noqa: E501
CUSTOM_CHAINS: Dict[str, Type[ConversationChain]] = { CUSTOM_CHAINS: Dict[str, Type[ConversationChain]] = {

View file

@ -1,7 +1,6 @@
from typing import Dict, List, Optional, Type from typing import Dict, List, Optional, Type
from langchain import prompts from langchain import prompts
from langchain.prompts import loading
from langflow.custom.customs import get_custom_nodes from langflow.custom.customs import get_custom_nodes
from langflow.interface.base import LangChainTypeCreator from langflow.interface.base import LangChainTypeCreator

View file

@ -4,8 +4,6 @@ from langchain.prompts import PromptTemplate
from pydantic import root_validator from pydantic import root_validator
from langflow.graph.utils import extract_input_variables_from_prompt 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: # Steps to create a BaseCustomPrompt:
# 1. Create a prompt template that endes with: # 1. Create a prompt template that endes with: