Fix tests and previous commit

This commit is contained in:
2025-12-09 07:11:02 +01:00
parent ca63865b07
commit da3d6f123d
7 changed files with 66 additions and 26 deletions
+6 -6
View File
@@ -226,7 +226,7 @@ class TestFormatEpisodicContext:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_episodic_context()
context = builder._format_episodic_context(memory)
assert context == ""
@@ -235,7 +235,7 @@ class TestFormatEpisodicContext:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_episodic_context()
context = builder._format_episodic_context(memory_with_search_results)
assert "LAST SEARCH" in context
assert "Inception 1080p" in context
@@ -249,7 +249,7 @@ class TestFormatEpisodicContext:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_episodic_context()
context = builder._format_episodic_context(memory)
assert "LAST SEARCH" in context
assert "ACTIVE DOWNLOADS" in context
@@ -264,7 +264,7 @@ class TestFormatStmContext:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_stm_context()
context = builder._format_stm_context(memory)
# Should at least show language
assert "CONVERSATION LANGUAGE" in context or context == ""
@@ -276,7 +276,7 @@ class TestFormatStmContext:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_stm_context()
context = builder._format_stm_context(memory)
assert "CURRENT WORKFLOW" in context
assert "download" in context
@@ -290,7 +290,7 @@ class TestFormatStmContext:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_stm_context()
context = builder._format_stm_context(memory)
assert "CURRENT WORKFLOW" in context
assert "CURRENT TOPIC" in context