mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-06-26 22:00:13 +00:00
18 lines
627 B
Python
18 lines
627 B
Python
|
import os
|
||
|
|
|
||
|
|
from haystack.agents.conversational import ConversationalAgent
|
||
|
|
from haystack.nodes import PromptNode
|
||
|
|
|
||
|
|
pn = PromptNode("gpt-3.5-turbo", api_key=os.environ.get("OPENAI_API_KEY"), max_length=256)
|
||
|
|
agent = ConversationalAgent(pn)
|
||
|
|
|
||
|
|
while True:
|
||
|
|
user_input = input("Human (type 'exit' or 'quit' to quit, 'memory' for agent's memory): ")
|
||
|
|
if user_input.lower() == "exit" or user_input.lower() == "quit":
|
||
|
|
break
|
||
|
|
if user_input.lower() == "memory":
|
||
|
|
print("\nMemory:\n", agent.memory.load())
|
||
|
|
else:
|
||
|
|
assistant_response = agent.run(user_input)
|
||
|
|
print("\nAssistant:", assistant_response)
|