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