forked from strands-agents/sdk-python
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathconftest.py
More file actions
25 lines (20 loc) · 920 Bytes
/
conftest.py
File metadata and controls
25 lines (20 loc) · 920 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
"""Pytest configuration for model tests."""
import sys
import unittest.mock
# Mock OpenAI version check before the openai_responses module is imported.
# This is necessary because the version check happens at module import time.
# We patch importlib.metadata.version directly since that's where get_package_version comes from.
if "strands.models.openai_responses" not in sys.modules:
_original_version = None
try:
from importlib.metadata import version as _original_version_func
_original_version = _original_version_func
except ImportError:
pass
def _mock_version(package_name: str) -> str:
if package_name == "openai":
return "2.0.0"
if _original_version:
return _original_version(package_name)
raise Exception(f"Package {package_name} not found")
unittest.mock.patch("importlib.metadata.version", _mock_version).start()