From 7f6a2691bd84b99bbb299d50dbb64f7cc553374c Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Thu, 21 Mar 2024 08:31:11 -0700 Subject: [PATCH] test(test_llm_guard.py): fix llm guard integration --- litellm/tests/test_llm_guard.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/litellm/tests/test_llm_guard.py b/litellm/tests/test_llm_guard.py index 0f9fad9a4f..c0f7b065f3 100644 --- a/litellm/tests/test_llm_guard.py +++ b/litellm/tests/test_llm_guard.py @@ -54,6 +54,7 @@ async def test_llm_guard_valid_response(): } ] }, + call_type="completion", ) except Exception as e: pytest.fail(f"An exception occurred - {str(e)}") @@ -89,6 +90,7 @@ async def test_llm_guard_error_raising(): } ] }, + call_type="completion", ) pytest.fail(f"Should have failed - {str(e)}") except Exception as e: