mirror of
https://github.com/Mintplex-Labs/langchain-python.git
synced 2026-08-27 19:29:56 -04:00
Harrison Chase
31 lines
916 B
Python
31 lines
916 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",
|
|
]
|