mirror of
https://github.com/tiennm99/litellm.git
synced 2026-06-18 23:40:42 +00:00
Ishaan Jaff
d3e04eac7f
* initial commit for azure responses api support * update get complete url * fixes for responses API * working azure responses API * working responses API * test suite for responses API * azure responses API test suite * fix test with complete url * fix test refactor * test fix metadata checks * fix code quality check
32 lines
1021 B
Python
32 lines
1021 B
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,
|
|
ResponseTextConfig,
|
|
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"),
|
|
}
|