Spaces:
Configuration error
Configuration error
File size: 851 Bytes
447ebeb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
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)
@pytest.mark.asyncio()
async def test_aiohttp_openai_gpt_4o():
litellm.set_verbose = True
response = await litellm.acompletion(
model="aiohttp_openai/gpt-4o",
messages=[{"role": "user", "content": "Hello, world!"}],
)
print(response)
|