mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-18 07:33:58 +00:00
Ishaan Jaff
76f1064229
* fix imports OpenAI SDK * ResponseText fixes * fixes ResponseText * fix imports * catch AttributeError * fix import * use openai==1.100.1 * fix build from PIP * fix lint test * Print OpenAI version * fix Install dependencies
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
import os
|
|
import sys
|
|
import pytest
|
|
import asyncio
|
|
from typing import Optional
|
|
from unittest.mock import patch, AsyncMock
|
|
|
|
sys.path.insert(0, os.path.abspath("../.."))
|
|
import litellm
|
|
from litellm.integrations.custom_logger import CustomLogger
|
|
import json
|
|
from litellm.types.utils import StandardLoggingPayload
|
|
from litellm.types.llms.openai import (
|
|
ResponseCompletedEvent,
|
|
ResponsesAPIResponse,
|
|
ResponseAPIUsage,
|
|
IncompleteDetails,
|
|
)
|
|
from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler
|
|
from base_responses_api import BaseResponsesAPITest
|
|
|
|
class TestAzureResponsesAPITest(BaseResponsesAPITest):
|
|
def get_base_completion_call_args(self):
|
|
return {
|
|
"model": "azure/computer-use-preview",
|
|
"truncation": "auto",
|
|
"api_base": os.getenv("AZURE_RESPONSES_OPENAI_ENDPOINT"),
|
|
"api_key": os.getenv("AZURE_RESPONSES_OPENAI_API_KEY"),
|
|
"api_version": os.getenv("AZURE_RESPONSES_OPENAI_API_VERSION"),
|
|
}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_azure_responses_api_preview_api_version():
|
|
"""
|
|
Ensure new azure preview api version is working
|
|
"""
|
|
litellm._turn_on_debug()
|
|
response = await litellm.aresponses(
|
|
model="azure/computer-use-preview",
|
|
truncation="auto",
|
|
api_version="preview",
|
|
api_base=os.getenv("AZURE_RESPONSES_OPENAI_ENDPOINT"),
|
|
api_key=os.getenv("AZURE_RESPONSES_OPENAI_API_KEY"),
|
|
input="Hello, can you tell me a short joke?",
|
|
) |