mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-08-14 11:37:45 +00:00
949b1b63b3
* initial integration * upgrade of prompthub * fix get_prompt_template * feedback * add prompthub-py to dependencies * tests * mypy * stray changes * review feedback * missing init * fix test * move logic in prompttemplate * linting * bugfixes * fix unit tests * fix cache * simplify prompttemplate init * remove unused function * removing wrong params * try remove all instances of prompt names * more tests * fix agent tests * more tests * fix tests * pylint * comma * black * fix test * docstring * review feedback * review feedback * fix mocks * mypy * fix mocks * fix reference to missing templates * feedback * remove direct references to default template var * tests * Update haystack/nodes/prompt/prompt_node.py Co-authored-by: Silvano Cerza <3314350+silvanocerza@users.noreply.github.com> --------- Co-authored-by: Silvano Cerza <3314350+silvanocerza@users.noreply.github.com>
55 lines
2.2 KiB
Python
55 lines
2.2 KiB
Python
import pytest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from haystack.agents.conversational import ConversationalAgent
|
|
from haystack.agents.memory import ConversationSummaryMemory, ConversationMemory, NoMemory
|
|
from haystack.nodes import PromptNode
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_init():
|
|
with patch("haystack.nodes.prompt.prompt_template.PromptTemplate._fetch_from_prompthub") as mock_prompthub:
|
|
mock_prompthub.side_effect = [("This is a test prompt. Use your knowledge to answer this question: {question}")]
|
|
prompt_node = PromptNode()
|
|
agent = ConversationalAgent(prompt_node)
|
|
|
|
# Test normal case
|
|
assert isinstance(agent.memory, ConversationMemory)
|
|
assert callable(agent.prompt_parameters_resolver)
|
|
assert agent.max_steps == 2
|
|
assert agent.final_answer_pattern == r"^([\s\S]+)$"
|
|
|
|
# ConversationalAgent doesn't have tools
|
|
assert not agent.tm.tools
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_init_with_summary_memory():
|
|
prompt_node = PromptNode(default_prompt_template="this is a test")
|
|
# Test with summary memory
|
|
agent = ConversationalAgent(prompt_node, memory=ConversationSummaryMemory(prompt_node))
|
|
assert isinstance(agent.memory, ConversationSummaryMemory)
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_init_with_no_memory():
|
|
with patch("haystack.nodes.prompt.prompt_template.PromptTemplate._fetch_from_prompthub") as mock_prompthub:
|
|
mock_prompthub.side_effect = [("This is a test prompt. Use your knowledge to answer this question: {question}")]
|
|
prompt_node = PromptNode()
|
|
# Test with no memory
|
|
agent = ConversationalAgent(prompt_node, memory=NoMemory())
|
|
assert isinstance(agent.memory, NoMemory)
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_run():
|
|
with patch("haystack.nodes.prompt.prompt_template.PromptTemplate._fetch_from_prompthub") as mock_prompthub:
|
|
mock_prompthub.side_effect = [("This is a test prompt. Use your knowledge to answer this question: {question}")]
|
|
prompt_node = PromptNode()
|
|
agent = ConversationalAgent(prompt_node)
|
|
|
|
# Mock the Agent run method
|
|
agent.run = MagicMock(return_value="Hello")
|
|
assert agent.run("query") == "Hello"
|
|
agent.run.assert_called_once_with("query")
|