forked from phoenix/litellm-mirror
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