From f88365b11da1816e34d2b6e9a4128a8f4216c9a5 Mon Sep 17 00:00:00 2001
From: Mery-Sanz <maria@aliasrobotics.com>
Date: Wed, 9 Apr 2025 15:54:58 +0200
Subject: [PATCH] add 2 more test

---
 tests/test_agent_tracing.py   | 2 +-
 tests/test_computer_action.py | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/test_agent_tracing.py b/tests/test_agent_tracing.py
index bb16cab2..4f6f5c69 100644
--- a/tests/test_agent_tracing.py
+++ b/tests/test_agent_tracing.py
@@ -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
diff --git a/tests/test_computer_action.py b/tests/test_computer_action.py
index 70dcabd5..9c4e9cff 100644
--- a/tests/test_computer_action.py
+++ b/tests/test_computer_action.py
@@ -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):