forked from phoenix/litellm-mirror
refactor: move all testing to top-level of repo
Closes https://github.com/BerriAI/litellm/issues/486
This commit is contained in:
parent
5403c5828c
commit
3560f0ef2c
213 changed files with 74 additions and 217 deletions
79
tests/local_testing/test_supabase_integration.py
Normal file
79
tests/local_testing/test_supabase_integration.py
Normal 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 = []
|
Loading…
Add table
Add a link
Reference in a new issue