From 2096adf05d19fb2e781a97cd06076aae594de699 Mon Sep 17 00:00:00 2001 From: shuofengzhang Date: Sat, 28 Mar 2026 04:43:01 +0000 Subject: [PATCH] Fix test imports for package context and modern agents SDK --- tests/__init__.py | 0 tests/helpers/__init__.py | 0 tests/helpers/mock_store.py | 2 +- tests/test_agents.py | 2 +- tests/test_chatkit_server.py | 6 +++--- tests/test_store.py | 4 ++-- 6 files changed, 7 insertions(+), 7 deletions(-) create mode 100644 tests/__init__.py create mode 100644 tests/helpers/__init__.py diff --git a/tests/__init__.py b/tests/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/helpers/__init__.py b/tests/helpers/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/tests/helpers/mock_store.py b/tests/helpers/mock_store.py index 8e76411..71b04c6 100644 --- a/tests/helpers/mock_store.py +++ b/tests/helpers/mock_store.py @@ -11,7 +11,7 @@ ThreadItem, ThreadMetadata, ) -from tests._types import RequestContext +from .._types import RequestContext from .mock_widget import SampleWidget diff --git a/tests/test_agents.py b/tests/test_agents.py index 1dfe81d..eacce02 100644 --- a/tests/test_agents.py +++ b/tests/test_agents.py @@ -23,7 +23,7 @@ StreamEvent, ToolCallItem, ) -from agents._run_impl import QueueCompleteSentinel +from agents.result import QueueCompleteSentinel from openai.types.responses import ( EasyInputMessageParam, ResponseFileSearchToolCall, diff --git a/tests/test_chatkit_server.py b/tests/test_chatkit_server.py index ec30b51..c0a7fc4 100644 --- a/tests/test_chatkit_server.py +++ b/tests/test_chatkit_server.py @@ -7,7 +7,7 @@ from typing import Any, AsyncIterator, Callable, cast import pytest -from helpers.mock_store import SQLiteStore +from .helpers.mock_store import SQLiteStore from pydantic import AnyUrl, TypeAdapter from chatkit.actions import Action @@ -93,8 +93,8 @@ WorkflowTaskAdded, ) from chatkit.widgets import Card, Text -from tests._types import RequestContext -from tests.test_store import make_thread_items +from ._types import RequestContext +from .test_store import make_thread_items server_id = 0 diff --git a/tests/test_store.py b/tests/test_store.py index a27f68e..67f766e 100644 --- a/tests/test_store.py +++ b/tests/test_store.py @@ -3,7 +3,7 @@ from datetime import datetime, timedelta import pytest -from helpers.mock_store import SQLiteStore +from .helpers.mock_store import SQLiteStore from pydantic import AnyUrl from chatkit.store import NotFoundError, Store @@ -20,7 +20,7 @@ WidgetItem, ) from chatkit.widgets import Card, Col, Text -from tests._types import RequestContext +from ._types import RequestContext def make_thread(thread_id="test_thread", created_at=None):