mirrors/hermes-agent
1
0
Fork 0
mirror of https://github.com/NousResearch/hermes-agent.git synced 2026-04-27 01:11:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

refactor: unify gateway session title sync flow

Browse source
This commit is contained in:
aaron 2026-04-23 09:09:33 +00:00
parent 59d45346ba
commit 3438de3623
8 changed files with 421 additions and 61 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/agent/test_title_generator.py
View file

@ -7,6 +7,7 @@ import pytest
from agent.title_generator import (
generate_title,
generate_title_if_missing,
auto_title_session,
maybe_auto_title,
)
@ -89,19 +90,26 @@ class TestAutoTitleSession:
def test_skips_if_no_session_db(self):
auto_title_session(None, "sess-1", "hi", "hello") # should not crash
def test_skips_if_title_exists(self):
def test_generate_title_if_missing_skips_if_title_exists(self):
db = MagicMock()
db.get_session_title.return_value = "Existing Title"
with patch("agent.title_generator.generate_title") as gen:
auto_title_session(db, "sess-1", "hi", "hello")
assert generate_title_if_missing(db, "sess-1", "hi", "hello") is None
gen.assert_not_called()
def test_generate_title_if_missing_returns_generated_title(self):
db = MagicMock()
db.get_session_title.return_value = None
with patch("agent.title_generator.generate_title", return_value="New Title"):
assert generate_title_if_missing(db, "sess-1", "hi", "hello") == "New Title"
def test_generates_and_sets_title(self):
db = MagicMock()
db.get_session_title.return_value = None
with patch("agent.title_generator.generate_title", return_value="New Title"):
with patch("agent.title_generator.generate_title_if_missing", return_value="New Title"):
auto_title_session(db, "sess-1", "hi", "hello")
db.set_session_title.assert_called_once_with("sess-1", "New Title")
@ -109,7 +117,7 @@ class TestAutoTitleSession:
db = MagicMock()
db.get_session_title.return_value = None
with patch("agent.title_generator.generate_title", return_value=None):
with patch("agent.title_generator.generate_title_if_missing", return_value=None):
auto_title_session(db, "sess-1", "hi", "hello")
db.set_session_title.assert_not_called()