chore: fixed imports and tests configuration
This commit is contained in:
@@ -1,11 +1,9 @@
|
||||
"""Edge case tests for the Agent."""
|
||||
|
||||
import pytest
|
||||
from unittest.mock import Mock
|
||||
|
||||
import pytest
|
||||
|
||||
from agent.agent import Agent
|
||||
from infrastructure.persistence import get_memory
|
||||
from alfred.agent.agent import Agent
|
||||
from alfred.infrastructure.persistence import get_memory
|
||||
|
||||
|
||||
class TestExecuteToolCallEdgeCases:
|
||||
@@ -16,7 +14,7 @@ class TestExecuteToolCallEdgeCases:
|
||||
agent = Agent(llm=mock_llm)
|
||||
|
||||
# Mock a tool that returns None
|
||||
from agent.registry import Tool
|
||||
from alfred.agent.registry import Tool
|
||||
|
||||
agent.tools["test_tool"] = Tool(
|
||||
name="test_tool", description="Test", func=lambda: None, parameters={}
|
||||
@@ -34,7 +32,7 @@ class TestExecuteToolCallEdgeCases:
|
||||
"""Should propagate KeyboardInterrupt."""
|
||||
agent = Agent(llm=mock_llm)
|
||||
|
||||
from agent.registry import Tool
|
||||
from alfred.agent.registry import Tool
|
||||
|
||||
def raise_interrupt():
|
||||
raise KeyboardInterrupt()
|
||||
|
||||
Reference in New Issue
Block a user