Fix some ruff issues in code
This commit is contained in:
@@ -164,7 +164,7 @@ class TestStepEdgeCases:
|
||||
mock_llm.complete = Mock(side_effect=mock_complete)
|
||||
agent = Agent(llm=mock_llm, max_tool_iterations=3)
|
||||
|
||||
response = agent.step("Loop test")
|
||||
agent.step("Loop test")
|
||||
|
||||
assert call_count[0] == 4
|
||||
|
||||
@@ -191,7 +191,7 @@ class TestStepEdgeCases:
|
||||
)
|
||||
agent = Agent(llm=mock_llm)
|
||||
|
||||
response = agent.step("Hello")
|
||||
agent.step("Hello")
|
||||
|
||||
call_args = mock_llm.complete.call_args[0][0]
|
||||
system_prompt = call_args[0]["content"]
|
||||
@@ -208,7 +208,7 @@ class TestStepEdgeCases:
|
||||
)
|
||||
agent = Agent(llm=mock_llm)
|
||||
|
||||
response = agent.step("Hello")
|
||||
agent.step("Hello")
|
||||
|
||||
call_args = mock_llm.complete.call_args[0][0]
|
||||
system_prompt = call_args[0]["content"]
|
||||
@@ -268,7 +268,7 @@ class TestAgentConcurrencyEdgeCases:
|
||||
mock_llm.complete = Mock(side_effect=mock_complete)
|
||||
agent = Agent(llm=mock_llm)
|
||||
|
||||
response = agent.step("Set movie folder")
|
||||
agent.step("Set movie folder")
|
||||
|
||||
mem = get_memory()
|
||||
assert mem.ltm.get_config("movie_folder") == str(real_folder["movies"])
|
||||
|
||||
Reference in New Issue
Block a user