forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
908 B
Python
29 lines
908 B
Python
# flake8: noqa
|
|
from langchain.memory.prompt import (
|
|
ENTITY_EXTRACTION_PROMPT,
|
|
ENTITY_MEMORY_CONVERSATION_TEMPLATE,
|
|
ENTITY_SUMMARIZATION_PROMPT,
|
|
KNOWLEDGE_TRIPLE_EXTRACTION_PROMPT,
|
|
SUMMARY_PROMPT,
|
|
)
|
|
from langchain.prompts.prompt import PromptTemplate
|
|
|
|
DEFAULT_TEMPLATE = """The following is a friendly conversation between a human and an AI. The AI is talkative and provides lots of specific details from its context. If the AI does not know the answer to a question, it truthfully says it does not know.
|
|
|
|
Current conversation:
|
|
{history}
|
|
Human: {input}
|
|
AI:"""
|
|
PROMPT = PromptTemplate(input_variables=["history", "input"], template=DEFAULT_TEMPLATE)
|
|
|
|
# Only for backwards compatibility
|
|
|
|
__all__ = [
|
|
"SUMMARY_PROMPT",
|
|
"ENTITY_MEMORY_CONVERSATION_TEMPLATE",
|
|
"ENTITY_SUMMARIZATION_PROMPT",
|
|
"ENTITY_EXTRACTION_PROMPT",
|
|
"KNOWLEDGE_TRIPLE_EXTRACTION_PROMPT",
|
|
"PROMPT",
|
|
]
|