phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

refactor: linting fixes

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-05 21:43:02 -08:00
parent 7b83238cb5
commit acef6bd58d
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/router.py
View file

@ -8,7 +8,7 @@
# Thank you ! We ❤️ you! - Krrish & Ishaan # Thank you ! We ❤️ you! - Krrish & Ishaan
from datetime import datetime from datetime import datetime
from typing import Dict, List, Optional, Union, Literal from typing import Dict, List, Optional, Union, Literal, Any
import random, threading, time, traceback import random, threading, time, traceback
import litellm, openai import litellm, openai
from litellm.caching import RedisCache, InMemoryCache, DualCache from litellm.caching import RedisCache, InMemoryCache, DualCache
@ -651,10 +651,10 @@ class Router:
if k != "metadata": if k != "metadata":
previous_model[k] = v previous_model[k] = v
elif k == "metadata" and isinstance(v, dict): elif k == "metadata" and isinstance(v, dict):
previous_model[k] = {} previous_model["metadata"] = {} # type: ignore
for metadata_k, metadata_v in kwargs['metadata'].items(): for metadata_k, metadata_v in kwargs['metadata'].items():
if metadata_k != "previous_models": if metadata_k != "previous_models":
previous_model[k][metadata_k] = metadata_v previous_model[k][metadata_k] = metadata_v # type: ignore
self.previous_models.append(previous_model) self.previous_models.append(previous_model)
kwargs["metadata"]["previous_models"] = self.previous_models kwargs["metadata"]["previous_models"] = self.previous_models
return kwargs return kwargs