64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
import pytest
|
|
|
|
from traderai.agent import OllamaAgent
|
|
from traderai.memory import MemoryStore
|
|
|
|
|
|
class EmptyTools:
|
|
schemas = []
|
|
|
|
@property
|
|
def pending_actions(self):
|
|
return {}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_chat_events_warns_when_ollama_offline():
|
|
agent = OllamaAgent("http://127.0.0.1:1", "missing-model", EmptyTools())
|
|
events = []
|
|
async for event in agent.chat_events("hello"):
|
|
events.append(event)
|
|
|
|
assert events[0]["type"] == "warning"
|
|
assert "Ollama is offline" in events[0]["message"]
|
|
assert events[-1]["type"] == "done"
|
|
|
|
|
|
def test_runtime_context_uses_previous_interaction_not_current_message(tmp_path):
|
|
memory = MemoryStore(str(tmp_path / "memory.sqlite3"))
|
|
memory.add_conversation("assistant", "older answer")
|
|
previous = memory.last_interaction()
|
|
assert previous is not None
|
|
|
|
memory.add_conversation("user", "current question")
|
|
current = memory.last_interaction()
|
|
|
|
agent = OllamaAgent("http://127.0.0.1:1", "missing-model", EmptyTools(), memory=memory)
|
|
context = agent._runtime_context("current question", previous_interaction=previous)
|
|
|
|
assert f"Previous interaction before this message: {previous['created_at']}" in context
|
|
assert f"Previous interaction before this message: {current['created_at']}" not in context
|
|
assert "Current local date/time:" in context
|
|
|
|
|
|
def test_runtime_context_includes_uex_user_identity(tmp_path):
|
|
memory = MemoryStore(str(tmp_path / "memory.sqlite3"))
|
|
memory.set_profile(
|
|
"uex_user",
|
|
{
|
|
"username": "pilot_hudson",
|
|
"name": "Hudson",
|
|
"email": "hudson@example.test",
|
|
"timezone": "America/New_York",
|
|
"specializations": "trading,hauling",
|
|
},
|
|
)
|
|
|
|
agent = OllamaAgent("http://127.0.0.1:1", "missing-model", EmptyTools(), memory=memory)
|
|
context = agent._runtime_context("")
|
|
|
|
assert "You are speaking with UEX user pilot_hudson (Hudson)." in context
|
|
assert "timezone: America/New_York" in context
|
|
assert "specializations: trading,hauling" in context
|
|
assert "hudson@example.test" not in context
|