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

fix(utils.py): fix pre call rules

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-21 07:10:04 -08:00
parent b991a35ebe
commit 2a681c1874
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/tests/test_completion.py
View file

@ -139,7 +139,7 @@ def test_completion_gpt4_vision():
pass pass
except Exception as e: except Exception as e:
pytest.fail(f"Error occurred: {e}") pytest.fail(f"Error occurred: {e}")
# test_completion_gpt4_vision() test_completion_gpt4_vision()
def test_completion_perplexity_api(): def test_completion_perplexity_api():
try: try:

2
litellm/utils.py
View file

@ -1154,7 +1154,7 @@ def client(original_function):
elif kwargs.get("messages", None): elif kwargs.get("messages", None):
messages = kwargs["messages"] messages = kwargs["messages"]
### PRE-CALL RULES ### ### PRE-CALL RULES ###
rules_obj.pre_call_rules(input="".join(m["content"] for m in messages), model=model) rules_obj.pre_call_rules(input="".join(m["content"] for m in messages if isinstance(m["content"], str)), model=model)
elif call_type == CallTypes.embedding.value: elif call_type == CallTypes.embedding.value:
messages = args[1] if len(args) > 1 else kwargs["input"] messages = args[1] if len(args) > 1 else kwargs["input"]
stream = True if "stream" in kwargs and kwargs["stream"] == True else False stream = True if "stream" in kwargs and kwargs["stream"] == True else False