test(test_llm_guard.py): fix llm guard integration

2025-04-27 11:43:54 +00:00 · 2024-03-21 08:31:11 -07:00 · 2024-03-21 08:31:11 -07:00 · 7f6a2691bd
commit 7f6a2691bd
parent 859a2af3ed
1 changed files with 2 additions and 0 deletions
--- 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: