litellm-mirror/tests/llm_translation/test_aiohttp_openai.py
2025-01-03 15:12:56 -08:00

23 lines
581 B
Python

import json
import os
import sys
from datetime import datetime
import pytest
sys.path.insert(
0, os.path.abspath("../../")
) # Adds the parent directory to the system path
import litellm
@pytest.mark.asyncio
async def test_aiohttp_openai():
litellm.set_verbose = True
response = await litellm.acompletion(
model="aiohttp_openai/fake-model",
messages=[{"role": "user", "content": "Hello, world!"}],
api_base="https://exampleopenaiendpoint-production.up.railway.app/v1/chat/completions",
api_key="fake-key",
)
print(response)