mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-18 07:33:58 +00:00
Ishaan Jaff
0f63cbea59
* add BaseInteractionsTest * add interactions_api_handler * init bridge * init LiteLLMResponsesInteractionsConfig * LiteLLMResponsesInteractionsHandler * mv test * fixes api spec * docs * fix transform+iterators * docs fix * fix iterator
114 lines
3.8 KiB
Python
114 lines
3.8 KiB
Python
"""
|
|
Abstract base class for Interactions API tests.
|
|
|
|
This class provides common test cases that can be inherited by provider-specific
|
|
test classes. Subclasses must implement get_model() and get_api_key().
|
|
"""
|
|
|
|
import os
|
|
from abc import ABC, abstractmethod
|
|
|
|
import pytest
|
|
import litellm
|
|
import litellm.interactions as interactions
|
|
|
|
|
|
class BaseInteractionsTest(ABC):
|
|
"""Abstract base class for interactions API tests.
|
|
|
|
Subclasses must implement get_model() and get_api_key().
|
|
All test methods are inherited and run against the specific provider.
|
|
"""
|
|
|
|
@abstractmethod
|
|
def get_model(self) -> str:
|
|
"""Return the model string for this provider."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_api_key(self) -> str:
|
|
"""Return the API key for this provider."""
|
|
pass
|
|
|
|
def test_create_simple_string_input(self):
|
|
"""Test creating an interaction with a simple string input."""
|
|
litellm._turn_on_debug()
|
|
api_key = self.get_api_key()
|
|
if not api_key:
|
|
pytest.skip(f"API key not set for {self.__class__.__name__}")
|
|
|
|
response = interactions.create(
|
|
model=self.get_model(),
|
|
input="Hello, what is 2 + 2?",
|
|
api_key=api_key,
|
|
)
|
|
assert response is not None
|
|
assert response.id is not None or response.status is not None
|
|
|
|
# Check outputs per OpenAPI spec
|
|
if response.outputs:
|
|
assert len(response.outputs) > 0
|
|
|
|
# Check usage per OpenAPI spec
|
|
# The spec defines: total_input_tokens, total_output_tokens
|
|
if response.usage:
|
|
# Usage is a dict in InteractionsAPIResponse
|
|
if isinstance(response.usage, dict):
|
|
assert response.usage.get("total_input_tokens") is not None or response.usage.get("total_output_tokens") is not None
|
|
else:
|
|
# If it's an object, check attributes
|
|
assert hasattr(response.usage, "total_input_tokens") or hasattr(response.usage, "total_output_tokens")
|
|
|
|
def test_create_with_system_instruction(self):
|
|
"""Test creating an interaction with system_instruction."""
|
|
api_key = self.get_api_key()
|
|
if not api_key:
|
|
pytest.skip(f"API key not set for {self.__class__.__name__}")
|
|
|
|
response = interactions.create(
|
|
model=self.get_model(),
|
|
input="What are you?",
|
|
system_instruction="You are a helpful pirate assistant. Always respond like a pirate.",
|
|
api_key=api_key,
|
|
)
|
|
assert response is not None
|
|
# Verify the response reflects the system instruction
|
|
if response.outputs:
|
|
assert len(response.outputs) > 0
|
|
|
|
def test_create_streaming(self):
|
|
"""Test creating a streaming interaction."""
|
|
api_key = self.get_api_key()
|
|
if not api_key:
|
|
pytest.skip(f"API key not set for {self.__class__.__name__}")
|
|
|
|
response_stream = interactions.create(
|
|
model=self.get_model(),
|
|
input="Count from 1 to 3.",
|
|
stream=True,
|
|
api_key=api_key,
|
|
)
|
|
|
|
# Collect all chunks
|
|
chunks = []
|
|
for chunk in response_stream:
|
|
chunks.append(chunk)
|
|
|
|
assert len(chunks) > 0
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_acreate_simple(self):
|
|
"""Test async interaction creation."""
|
|
api_key = self.get_api_key()
|
|
if not api_key:
|
|
pytest.skip(f"API key not set for {self.__class__.__name__}")
|
|
|
|
response = await interactions.acreate(
|
|
model=self.get_model(),
|
|
input="What is the speed of light?",
|
|
api_key=api_key,
|
|
)
|
|
assert response is not None
|
|
assert response.id is not None or response.status is not None
|
|
|