Skip to content

Commit 208caa2

Browse files
fix: Use correct fixtures for tests
1 parent a26dee5 commit 208caa2

File tree

3 files changed

+200
-167
lines changed

3 files changed

+200
-167
lines changed

agents-api/agents_api/routers/utils/tools.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import json
22
from collections.abc import Awaitable, Callable
3-
from functools import partial, reduce
3+
from functools import partial
44
from typing import Any, ClassVar
55
from uuid import UUID
66

agents-api/tests/fixtures.py

+9-2
Original file line numberDiff line numberDiff line change
@@ -92,11 +92,18 @@ def patch_embed_acompletion():
9292
def make_acompletion_multiple_outputs(outputs: list[dict] | Callable):
9393
@fixture(scope="test")
9494
def patch_embed_acompletion_with_tool_calls(
95-
agent=test_agent, doc=test_doc, user=test_user, task=test_task, user_doc=test_user_doc, session=test_session,
95+
agent=test_agent,
96+
doc=test_doc,
97+
user=test_user,
98+
task=test_task,
99+
user_doc=test_user_doc,
100+
session=test_session,
96101
):
97102
nonlocal outputs
98103
if callable(outputs):
99-
outputs = outputs(agent=agent, doc=doc, user=user, task=task, user_doc=user_doc, session=session)
104+
outputs = outputs(
105+
agent=agent, doc=doc, user=user, task=task, user_doc=user_doc, session=session
106+
)
100107
with patch_embed_acompletion_multiple_outputs_ctx(outputs) as (embed, acompletion):
101108
yield embed, acompletion
102109

0 commit comments

Comments
 (0)