mirror of https://github.com/aliasrobotics/cai.git
add 2 more test
This commit is contained in:
parent
05fab61056
commit
f88365b11d
|
|
@ -5,7 +5,7 @@ import asyncio
|
|||
import pytest
|
||||
from inline_snapshot import snapshot
|
||||
|
||||
from agents import Agent, RunConfig, Runner, trace
|
||||
from cai.sdk.agents import Agent, RunConfig, Runner, trace
|
||||
|
||||
from .fake_model import FakeModel
|
||||
from .test_responses import get_text_message
|
||||
|
|
|
|||
|
|
@ -21,7 +21,7 @@ from openai.types.responses.response_computer_tool_call import (
|
|||
ResponseComputerToolCall,
|
||||
)
|
||||
|
||||
from agents import (
|
||||
from cai.sdk.agents import (
|
||||
Agent,
|
||||
AgentHooks,
|
||||
AsyncComputer,
|
||||
|
|
@ -31,8 +31,8 @@ from agents import (
|
|||
RunContextWrapper,
|
||||
RunHooks,
|
||||
)
|
||||
from agents._run_impl import ComputerAction, ToolRunComputerAction
|
||||
from agents.items import ToolCallOutputItem
|
||||
from cai.sdk.agents._run_impl import ComputerAction, ToolRunComputerAction
|
||||
from cai.sdk.agents.items import ToolCallOutputItem
|
||||
|
||||
|
||||
class LoggingComputer(Computer):
|
||||
|
|
|
|||
Loading…
Reference in New Issue