serversdown
a5477ae15c
She can now *do* things mid-conversation, not just reply. Adds a tool-calling loop to the chat path and her first two tools; the same mechanism will carry the poker tools (start_session, log_result, get_stats, solver) next. - tools.py: registry of OpenAI-style tool specs + handlers + safe dispatch; journal_write (knowing journaling) and note (tagged notepad, e.g. poker reads) - llm.chat_call(): OpenAI-style call that returns tool_calls (cloud/mi50); local has no tool support and returns plain content - chat.respond(): tool loop — offer tools, run any calls, feed results back, repeat until a text reply (capped at MAX_TOOL_ROUNDS); persists final reply - tests: dispatch + full chat loop (tool call -> result -> reply) Verified live: she invoked `note`, tagged it 'poker', stored a villain read. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
"""Lyra's tools: dispatch + the chat tool loop (call -> run -> feed back -> reply)."""
|
|
from __future__ import annotations
|
|
|
|
import importlib
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture
|
|
def lyra(tmp_path, monkeypatch):
|
|
monkeypatch.setenv("LYRA_DB_PATH", str(tmp_path / "test.db"))
|
|
from lyra import llm
|
|
monkeypatch.setattr(llm, "embed", lambda texts: [[0.1, 0.2, 0.3] for _ in texts])
|
|
import lyra.memory as memory
|
|
importlib.reload(memory)
|
|
return memory
|
|
|
|
|
|
def test_journal_write_tool(lyra):
|
|
from lyra import tools
|
|
out = tools.dispatch("journal_write", '{"entry": "a private thought"}')
|
|
assert "journal" in out.lower()
|
|
entries = lyra.list_journal(kinds=("journal",))
|
|
assert any(e["content"] == "a private thought" and e["source"] == "chat" for e in entries)
|
|
|
|
|
|
def test_note_tool_with_tag(lyra):
|
|
from lyra import tools
|
|
tools.dispatch("note", {"content": "villain 3-bets light", "tag": "poker"})
|
|
notes = lyra.list_journal(kinds=("note",))
|
|
assert any("[poker] villain 3-bets light" == e["content"] for e in notes)
|
|
|
|
|
|
def test_unknown_tool_is_safe(lyra):
|
|
from lyra import tools
|
|
assert "unknown tool" in tools.dispatch("nope", {})
|
|
|
|
|
|
def test_chat_runs_tool_then_replies(lyra, monkeypatch):
|
|
from lyra import llm, chat
|
|
calls = {"n": 0}
|
|
|
|
def fake_chat_call(messages, backend="cloud", model=None, tools=None):
|
|
calls["n"] += 1
|
|
if calls["n"] == 1:
|
|
return ({"role": "assistant", "content": None, "tool_calls": []},
|
|
[{"id": "c1", "name": "journal_write", "arguments": '{"entry": "noted from chat"}'}])
|
|
return ({"role": "assistant", "content": "Done, Brian."}, None)
|
|
|
|
monkeypatch.setattr(llm, "chat_call", fake_chat_call)
|
|
reply = chat.respond("s1", "write that down for me", backend="cloud")
|
|
|
|
assert reply == "Done, Brian."
|
|
assert calls["n"] == 2 # one tool round, then the text reply
|
|
assert any("noted from chat" in e["content"] for e in lyra.list_journal())
|