Added test to check if acompletion is using the same parameters as CompletionRequest attributes. Added functools to client decorator to expose acompletion parameters from outside.

This commit is contained in:
Mateo Cámara 2024-01-09 12:06:49 +01:00
parent 48b2f69c93
commit bb06c51ede
2 changed files with 26 additions and 1 deletions

View file

@ -14,6 +14,7 @@ import subprocess, os
import litellm, openai
import itertools
import random, uuid, requests
from functools import wraps
import datetime, time
import tiktoken
import uuid
@ -1934,6 +1935,7 @@ def client(original_function):
# [Non-Blocking Error]
pass
@wraps(original_function)
def wrapper(*args, **kwargs):
start_time = datetime.datetime.now()
result = None
@ -2128,6 +2130,7 @@ def client(original_function):
e.message += f"\n Check the log in your dashboard - {liteDebuggerClient.dashboard_url}"
raise e
@wraps(original_function)
async def wrapper_async(*args, **kwargs):
start_time = datetime.datetime.now()
result = None