From dc5d26fd8447fc0d97919466b796aa503b774f58 Mon Sep 17 00:00:00 2001 From: rasmi Date: Thu, 11 Jun 2026 18:09:52 -0400 Subject: [PATCH 1/5] Fix imports. --- sec-gemini-python/tests/test_secgemini.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/sec-gemini-python/tests/test_secgemini.py b/sec-gemini-python/tests/test_secgemini.py index 69ef288..854f272 100644 --- a/sec-gemini-python/tests/test_secgemini.py +++ b/sec-gemini-python/tests/test_secgemini.py @@ -22,10 +22,12 @@ 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, From 2f3bcd3a4060737737f8e1c879f36529b9b88d02 Mon Sep 17 00:00:00 2001 From: rasmi Date: Thu, 11 Jun 2026 18:11:48 -0400 Subject: [PATCH 2/5] Mark test with async_require_env_variable --- sec-gemini-python/tests/test_local_tools_integration.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/sec-gemini-python/tests/test_local_tools_integration.py b/sec-gemini-python/tests/test_local_tools_integration.py index 6d52c16..39ab6ff 100644 --- a/sec-gemini-python/tests/test_local_tools_integration.py +++ b/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.""" From cba42f7c2c4b655ca9f5deada8b3fd8b368b41d7 Mon Sep 17 00:00:00 2001 From: rasmi Date: Thu, 11 Jun 2026 18:12:56 -0400 Subject: [PATCH 3/5] Update imports. --- sec-gemini-python/tests/conftest.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sec-gemini-python/tests/conftest.py b/sec-gemini-python/tests/conftest.py index bb1200a..f4f262d 100644 --- a/sec-gemini-python/tests/conftest.py +++ b/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 From 5949c75c4291ba01d004120dd4f5f881b19318b4 Mon Sep 17 00:00:00 2001 From: rasmi Date: Thu, 11 Jun 2026 18:13:33 -0400 Subject: [PATCH 4/5] Formatting. --- sec-gemini-python/tests/test_secgemini.py | 1 - 1 file changed, 1 deletion(-) diff --git a/sec-gemini-python/tests/test_secgemini.py b/sec-gemini-python/tests/test_secgemini.py index 854f272..533aebc 100644 --- a/sec-gemini-python/tests/test_secgemini.py +++ b/sec-gemini-python/tests/test_secgemini.py @@ -25,7 +25,6 @@ from pytest_httpx import HTTPXMock from rich import print as rprint - from .conftest import MOCK_SEC_GEMINI_API_HOST from .utils import ( async_require_env_variable, From 7b8f1a63282c16e57a5c44797ae991d952c840fe Mon Sep 17 00:00:00 2001 From: rasmi Date: Thu, 11 Jun 2026 18:16:25 -0400 Subject: [PATCH 5/5] Create __init__.py --- sec-gemini-python/tests/__init__.py | 1 + 1 file changed, 1 insertion(+) create mode 100644 sec-gemini-python/tests/__init__.py diff --git a/sec-gemini-python/tests/__init__.py b/sec-gemini-python/tests/__init__.py new file mode 100644 index 0000000..8b13789 --- /dev/null +++ b/sec-gemini-python/tests/__init__.py @@ -0,0 +1 @@ +