mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-25 18:54:30 +00:00
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:
parent
48b2f69c93
commit
bb06c51ede
2 changed files with 26 additions and 1 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue