mirrors/hermes-agent
1
0
Fork 0
mirror of https://github.com/NousResearch/hermes-agent.git synced 2026-06-19 10:02:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 11

test(openviking): update plugin expectations

Browse source
This commit is contained in:
Hao Zhe 2026-06-17 15:05:51 +08:00
parent 3ac6551ba3
commit 99a20f8d9a
1 changed files with 20 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

30
tests/openviking_plugin/test_openviking.py
View file

@ -148,7 +148,7 @@ class TestOpenVikingSkillQuerySafety:
assert RecordingVikingClient.calls == [
(
"/api/v1/search/find",
{"query": "make a skill for release triage", "top_k": 5},
{"query": "make a skill for release triage", "limit": 5},
)
]
@ -178,7 +178,7 @@ class TestOpenVikingSkillQuerySafety:
assert RecordingVikingClient.calls == [
(
"/api/v1/search/find",
{"query": "fix the failing retrieval test", "top_k": 5},
{"query": "fix the failing retrieval test", "limit": 5},
)
]
@ -220,16 +220,25 @@ class TestOpenVikingSkillQuerySafety:
)
provider.sync_turn(skill_message, "Done.")
provider._sync_thread.join(timeout=5.0)
assert provider._drain_writers("session-1", timeout=5.0)
assert RecordingVikingClient.calls == [
(
"/api/v1/sessions/session-1/messages",
{"role": "user", "content": "make a skill for release triage"},
),
(
"/api/v1/sessions/session-1/messages",
{"role": "assistant", "content": "Done."},
"/api/v1/sessions/session-1/messages/batch",
{
"messages": [
{
"role": "user",
"parts": [
{"type": "text", "text": "make a skill for release triage"},
],
},
{
"role": "assistant",
"parts": [{"type": "text", "text": "Done."}],
},
]
},
),
]
@ -247,7 +256,8 @@ class TestOpenVikingSkillQuerySafety:
provider.sync_turn(skill_message, "Done.")
assert provider._sync_thread is None
assert provider._turn_count == 0
assert provider._inflight_writers == {}
assert RecordingVikingClient.calls == []