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
+7 -7
View File
@@ -309,7 +309,7 @@ class TestFormatEpisodicContextEdgeCases:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_episodic_context()
context = builder._format_episodic_context(memory)
assert "LAST SEARCH" in context
@@ -327,7 +327,7 @@ class TestFormatEpisodicContextEdgeCases:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_episodic_context()
context = builder._format_episodic_context(memory)
# Should not crash
assert "LAST SEARCH" in context
@@ -339,7 +339,7 @@ class TestFormatEpisodicContextEdgeCases:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_episodic_context()
context = builder._format_episodic_context(memory)
assert "ACTIVE DOWNLOADS" in context
assert "0%" in context # Default progress
@@ -358,7 +358,7 @@ class TestFormatStmContextEdgeCases:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_stm_context()
context = builder._format_stm_context(memory)
assert "CURRENT WORKFLOW" in context
@@ -370,7 +370,7 @@ class TestFormatStmContextEdgeCases:
builder = PromptBuilder(tools)
try:
context = builder._format_stm_context()
context = builder._format_stm_context(memory)
assert "CURRENT WORKFLOW" in context or True
except (AttributeError, TypeError):
# Expected if None target causes issues
@@ -383,7 +383,7 @@ class TestFormatStmContextEdgeCases:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_stm_context()
context = builder._format_stm_context(memory)
# Empty topic might not be shown
assert isinstance(context, str)
@@ -395,6 +395,6 @@ class TestFormatStmContextEdgeCases:
tools = make_tools()
builder = PromptBuilder(tools)
context = builder._format_stm_context()
context = builder._format_stm_context(memory)
assert "EXTRACTED ENTITIES" in context