Fix tests and previous commit
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user