hot fix litellm import

This commit is contained in:
ishaan-jaff 2023-08-17 11:39:27 -07:00
parent 489a31adb7
commit 32a3b98c0a
2 changed files with 26 additions and 12 deletions

View file

@ -68,10 +68,22 @@ class ModelResponse:
}
def __getitem__(self, key):
return getattr(self, key)
return getattr(self, key)
def __setitem__(self, key, value):
setattr(self, key, value)
setattr(self, key, value)
def __iter__(self):
return iter(vars(self))
def __str__(self):
result = f"""
{self.choices}\n
{self.created}\n
{self.model}\n
{self.usage}\n
"""
return result
def print_verbose(print_statement):