Fix imports.

sec-gemini-python/tests/__init__.py

@@ -0,0 +1 @@
+

sec-gemini-python/tests/conftest.py

@@ -19,7 +19,8 @@
 import httpx
 import pytest
 from pytest_httpx import HTTPXMock
-from utils import require_env_variable
+
+from .utils import require_env_variable
 
 from sec_gemini import SecGemini
 from sec_gemini.models.enums import State, UserType

sec-gemini-python/tests/test_local_tools_integration.py

@@ -9,6 +9,8 @@
 from mcp.server.fastmcp.tools import Tool
 from rich.markdown import Markdown
 
+from .utils import async_require_env_variable
+
 from sec_gemini import SecGemini, State
 from sec_gemini.models.enums import MessageType, MimeType
 from sec_gemini.models.local_tool import LocalTool
@@ -270,6 +272,7 @@ def tool_with_optional_int(value: int | None = None):
     print()
 
 
+@async_require_env_variable("SEC_GEMINI_API_KEY")
 @pytest.mark.asyncio
 async def test_list_dir_and_sha256_local_tools() -> None:
   """Tests the local tool integration by streaming a prompt that requires file listing and hashing."""

sec-gemini-python/tests/test_secgemini.py

@@ -22,10 +22,11 @@
 from typing import TypeVar
 
 import pytest
-from conftest import MOCK_SEC_GEMINI_API_HOST
 from pytest_httpx import HTTPXMock
 from rich import print as rprint
-from utils import (
+
+from .conftest import MOCK_SEC_GEMINI_API_HOST
+from .utils import (
   async_require_env_variable,
   parse_secgemini_response,
   require_env_variable,