Skip to content

Commit ff2a57d

Browse files
committed
Fix OCI test: add missing mock_builder fixture and asyncio marker
1 parent 542ff4a commit ff2a57d

1 file changed

Lines changed: 9 additions & 0 deletions

File tree

packages/nvidia_nat_core/tests/nat/llm/test_oci_llm.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,19 @@
1818
from unittest.mock import MagicMock
1919
from unittest.mock import patch
2020

21+
import pytest
22+
2123
from nat.builder.llm import LLMProviderInfo
2224
from nat.llm.oci_llm import OCIModelConfig
2325
from nat.llm.oci_llm import oci_llm
2426

2527

28+
@pytest.fixture(name="mock_builder")
29+
def fixture_mock_builder():
30+
"""Create a mock builder."""
31+
return MagicMock()
32+
33+
2634
def test_oci_model_config_defaults():
2735
config = OCIModelConfig(model_name="nvidia/Llama-3.1-Nemotron-Nano-8B-v1")
2836

@@ -41,6 +49,7 @@ def test_oci_model_config_accepts_endpoint_aliases():
4149
assert config.endpoint == "https://inference.generativeai.us-chicago-1.oci.oraclecloud.com"
4250

4351

52+
@pytest.mark.asyncio
4453
async def test_oci_llm_provider_yields_provider_info(mock_builder):
4554
config = OCIModelConfig(
4655
model_name="nvidia/Llama-3.1-Nemotron-Nano-8B-v1",

0 commit comments

Comments
 (0)