from haystack.agents.memory import ConversationSummaryMemory | |
from haystack.nodes import PromptNode | |
def return_memory_node(openai_key: str) -> ConversationSummaryMemory: | |
""" | |
Returns the memory node. | |
:param openai_key: the OpenAI key | |
:return: the memory node | |
""" | |
memory_prompt_node = PromptNode('gpt-3.5-turbo-16k', api_key=openai_key, max_length=1024) | |
return ConversationSummaryMemory(memory_prompt_node) | |