Skip to content

Commit bd56631

Browse files
Update tests/fixtures.py
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
1 parent 7a83c97 commit bd56631

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/fixtures.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,14 +66,15 @@ def completed_task(task_builder):
6666

6767

6868
@pytest.fixture
69-
def task_with_history(task_builder, message_builder):
69+
def task_with_history(task_builder):
7070
messages = [
71-
message_builder.as_user().with_text('Hello').build(),
72-
message_builder.as_agent().with_text('Hi there!').build(),
71+
MessageBuilder().as_user().with_text('Hello').build(),
72+
MessageBuilder().as_agent().with_text('Hi there!').build(),
7373
]
7474
return task_builder.with_history(*messages).build()
7575

7676

77+
7778
@pytest.fixture
7879
def task_with_artifacts(task_builder, artifact_builder):
7980
artifacts = [

0 commit comments

Comments
 (0)