# Copyright 2025 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# SPDX-License-Identifier: Apache-2.0
"""Tests for xAI plugin."""
from unittest.mock import patch
import pytest
from genkit.core.error import GenkitError
from genkit.core.registry import ActionKind
from genkit.plugins.xai import XAI, xai_name
from genkit.plugins.xai.model_info import SUPPORTED_XAI_MODELS, get_model_info
def test_xai_name():
assert xai_name('grok-3') == 'xai/grok-3'
def test_init_with_api_key():
plugin = XAI(api_key='test-key')
assert plugin._xai_client is not None
assert plugin.models == list(SUPPORTED_XAI_MODELS.keys())
def test_init_without_api_key_raises():
with patch.dict('os.environ', {}, clear=True):
try:
XAI()
raise AssertionError('Expected GenkitError')
except GenkitError:
pass
def test_init_with_env_var():
with patch.dict('os.environ', {'XAI_API_KEY': 'env-key'}):
plugin = XAI()
assert plugin._xai_client is not None
def test_custom_models():
plugin = XAI(api_key='test-key', models=['grok-3', 'grok-3-mini'])
assert plugin.models == ['grok-3', 'grok-3-mini']
@pytest.mark.asyncio
async def test_plugin_initialize():
plugin = XAI(api_key='test-key')
actions = await plugin.init()
assert len(actions) == len(SUPPORTED_XAI_MODELS)
@pytest.mark.asyncio
async def test_resolve_action_model():
plugin = XAI(api_key='test-key')
action = await plugin.resolve(ActionKind.MODEL, 'xai/grok-3')
assert action is not None
assert action.kind == ActionKind.MODEL
assert action.name == 'xai/grok-3'
def test_supported_models():
assert len(SUPPORTED_XAI_MODELS) >= 4
for _name, info in SUPPORTED_XAI_MODELS.items():
assert info.label.startswith('xAI - ')
assert len(info.versions) > 0
assert info.supports.tools
def test_get_model_info_known():
info = get_model_info('grok-3')
assert 'grok-3' in info.versions[0]
assert info.supports.multiturn
def test_get_model_info_unknown():
info = get_model_info('unknown-model')
assert 'unknown-model' in info.label
