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 langflow.graph.utils import extract_input_variables_from_prompt
class Code(BaseModel):

View file

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

View file

@ -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]] = {

View file

@ -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

View file

@ -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: