tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-13 14:23:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

(feat) update non default values

Browse Source
This commit is contained in:
ishaan-jaff
2024-02-20 18:55:20 -08:00
parent a4365926a1
commit 476f401b74
2 changed files with 30 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/proxy/proxy_server.py
+19 -2
View File
@@ -3307,7 +3307,15 @@ async def update_key_fn(request: Request, data: UpdateKeyRequest):
if prisma_client is None:
raise Exception("Not connected to DB!")
non_default_values = {k: v for k, v in data_json.items() if v is not None}
# get non default values for key
non_default_values = {}
for k, v in data_json.items():
if v is not None and v not in (
[],
{},
0,
): # models default to [], spend defaults to 0, we should not reset these values
non_default_values[k] = v
response = await prisma_client.update_data(
token=key, data={**non_default_values, "token": key}
)
@@ -4116,7 +4124,16 @@ async def user_update(data: UpdateUserRequest):
if prisma_client is None:
raise Exception("Not connected to DB!")
non_default_values = {k: v for k, v in data_json.items() if v is not None}
# get non default values for key
non_default_values = {}
for k, v in data_json.items():
if v is not None and v not in (
[],
{},
0,
): # models default to [], spend defaults to 0, we should not reset these values
non_default_values[k] = v
response = await prisma_client.update_data(
user_id=data_json["user_id"],
data=non_default_values,
litellm/tests/test_key_generate_prisma.py
+11 -1
View File
@@ -887,7 +887,7 @@ def test_generate_and_update_key(prisma_client):
request._url = URL(url="/update/key")
# update the key
await update_key_fn(
response1 = await update_key_fn(
request=Request,
data=UpdateKeyRequest(
key=generated_key,
@@ -895,6 +895,15 @@ def test_generate_and_update_key(prisma_client):
),
)
print("response1=", response1)
# update the team id
response2 = await update_key_fn(
request=Request,
data=UpdateKeyRequest(key=generated_key, team_id="ishaan"),
)
print("response2=", response2)
# get info on key after update
result = await info_key_fn(key=generated_key)
print("result from info_key_fn", result)
@@ -906,6 +915,7 @@ def test_generate_and_update_key(prisma_client):
"project": "litellm-project3",
}
assert result["info"]["models"] == ["ada", "babbage", "curie", "davinci"]
assert result["info"]["team_id"] == "ishaan"
# cleanup - delete key
delete_key_request = KeyRequest(keys=[generated_key])