Added LiteLLM to the stack
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
|
||||
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 ishaan@berri.ai 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."
|
||||
)
|
||||
|
Reference in New Issue
Block a user