RomanNum3ral
/
cai
mirror of https://github.com/aliasrobotics/cai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix test_agent_runner

This commit is contained in:
Mery-Sanz 2025-04-09 15:51:05 +02:00
parent 4ae76ecd29
commit 05fab61056
2 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_agent_runner.py
View File

@ -6,7 +6,7 @@ from typing import Any
import pytest
from typing_extensions import TypedDict
from agents import (
from cai.sdk.agents import (
Agent,
GuardrailFunctionOutput,
Handoff,
@ -23,8 +23,8 @@ from agents import (
UserError,
handoff,
)
from agents.agent import ToolsToFinalOutputResult
from agents.tool import FunctionToolResult, function_tool
from cai.sdk.agents.agent import ToolsToFinalOutputResult
from cai.sdk.agents.tool import FunctionToolResult, function_tool
from .fake_model import FakeModel
from .test_responses import (

8
tests/test_agent_runner_streamed.py
View File

@ -6,7 +6,7 @@ from typing import Any
import pytest
from typing_extensions import TypedDict
from agents import (
from cai.sdk.agents import (
Agent,
GuardrailFunctionOutput,
Handoff,
@ -20,9 +20,9 @@ from agents import (
UserError,
handoff,
)
from agents.items import RunItem
from agents.run import RunConfig
from agents.stream_events import AgentUpdatedStreamEvent
from cai.sdk.agents.items import RunItem
from cai.sdk.agents.run import RunConfig
from cai.sdk.agents.stream_events import AgentUpdatedStreamEvent
from .fake_model import FakeModel
from .test_responses import (