agent: look at current info, its aUEC, feat: pull up notifcations.
This commit is contained in:
+14
-1
@@ -1,6 +1,6 @@
|
||||
import pytest
|
||||
|
||||
from traderai.agent import OllamaAgent
|
||||
from traderai.agent import OllamaAgent, SYSTEM_PROMPT
|
||||
from traderai.memory import MemoryStore
|
||||
|
||||
|
||||
@@ -77,3 +77,16 @@ def test_stream_metrics_include_reading_and_writing_rates():
|
||||
assert metrics["reading_tokens_per_second"] == 10
|
||||
assert metrics["writing_tokens"] == 30
|
||||
assert metrics["writing_tokens_per_second"] == 10
|
||||
|
||||
|
||||
def test_system_prompt_prefers_current_marketplace_data():
|
||||
assert "open/current" in SYSTEM_PROMPT
|
||||
assert "Do not use historical sale data" in SYSTEM_PROMPT
|
||||
assert "aUEC/UEC credits" in SYSTEM_PROMPT
|
||||
assert "never real-world dollars" in SYSTEM_PROMPT
|
||||
|
||||
|
||||
def test_ollama_options_include_num_ctx():
|
||||
agent = OllamaAgent("http://127.0.0.1:1", "missing-model", EmptyTools(), num_ctx=64000)
|
||||
|
||||
assert agent._ollama_options() == {"num_ctx": 64000}
|
||||
|
||||
Reference in New Issue
Block a user