Skip to content

Commit 40d8862

Browse files
authored
Update tests/test_agent.py
1 parent 1dd2fcc commit 40d8862

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/test_agent.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6137,10 +6137,7 @@ def llm(messages: list[ModelMessage], _info: AgentInfo) -> ModelResponse:
61376137
def test_message_history_cannot_start_with_model_response():
61386138
"""Test that message history starting with ModelResponse raises UserError."""
61396139

6140-
def simple_response(_messages: list[ModelMessage], _info: AgentInfo) -> ModelResponse:
6141-
return ModelResponse(parts=[TextPart(content='Final response')]) # pragma: no cover
6142-
6143-
agent = Agent(FunctionModel(simple_response))
6140+
agent = Agent('test')
61446141

61456142
invalid_history = [
61466143
ModelResponse(parts=[TextPart(content='ai response')]),

0 commit comments

Comments
 (0)