refactor: move all testing to top-level of repo

Closes https://github.com/BerriAI/litellm/issues/486
This commit is contained in:
Krrish Dholakia 2024-09-28 13:23:39 -07:00
parent 5403c5828c
commit 3560f0ef2c
213 changed files with 74 additions and 217 deletions

View file

@ -0,0 +1,79 @@
#### What this tests ####
# This tests if logging to the supabase integration actually works
import sys, os
import traceback
import pytest
sys.path.insert(
0, os.path.abspath("../..")
) # Adds the parent directory to the system path
import litellm
from litellm import embedding, completion
litellm.input_callback = ["supabase"]
litellm.success_callback = ["supabase"]
litellm.failure_callback = ["supabase"]
litellm.set_verbose = False
def test_supabase_logging():
try:
response = completion(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": "Hello tell me hi"}],
user="ishaanRegular",
max_tokens=10,
)
print(response)
except Exception as e:
print(e)
# test_supabase_logging()
def test_acompletion_sync():
import asyncio
import time
async def completion_call():
try:
response = await litellm.acompletion(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": "write a poem"}],
max_tokens=10,
stream=True,
user="ishaanStreamingUser",
timeout=5,
)
complete_response = ""
start_time = time.time()
async for chunk in response:
chunk_time = time.time()
# print(chunk)
complete_response += chunk["choices"][0]["delta"].get("content", "")
# print(complete_response)
# print(f"time since initial request: {chunk_time - start_time:.5f}")
if chunk["choices"][0].get("finish_reason", None) != None:
print("🤗🤗🤗 DONE")
return
except litellm.Timeout as e:
pass
except:
print(f"error occurred: {traceback.format_exc()}")
pass
asyncio.run(completion_call())
# test_acompletion_sync()
# reset callbacks
litellm.input_callback = []
litellm.success_callback = []
litellm.failure_callback = []