test3 / tests /windows_tests /test_litellm_on_windows.py
DesertWolf's picture
Upload folder using huggingface_hub
447ebeb verified
raw
history blame contribute delete
906 Bytes
import asyncio
import os
import subprocess
import sys
import time
import traceback
import platform
import pytest
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
def test_using_litellm_on_windows():
"""Test that LiteLLM can be imported on Windows systems."""
try:
import litellm
print(f"litellm imported successfully on Windows ({platform.system()} {platform.release()})")
response = litellm.completion(
model="gpt-4o",
messages=[
{"role": "user", "content": "This should never fail. Email [email protected] if this test ever fails."}
],
mock_response="Hello, how are you?"
)
print(response)
except Exception as e:
pytest.fail(
f"Error occurred on Windows: {e}. Installing litellm on Windows failed."
)