diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py index 6a0cc5ac327..20f9a6380de 100644 --- a/litellm/tests/test_completion.py +++ b/litellm/tests/test_completion.py @@ -1916,7 +1916,7 @@ def test_completion_gemini(): pytest.fail(f"Error occurred: {e}") -test_completion_gemini() +# test_completion_gemini() # Palm tests @@ -2002,6 +2002,22 @@ def test_completion_together_ai_stream(): # test_completion_together_ai_stream() +# Cloud flare AI tests +def test_completion_cloudflare(): + try: + litellm.set_verbose = True + response = completion( + model="cloudflare/@cf/meta/llama-2-7b-chat-int8", + messages=[{"content": "what llm are you", "role": "user"}], + ) + print(response) + + except Exception as e: + pytest.fail(f"Error occurred: {e}") + + +test_completion_cloudflare() + # async def get_response(generator): # async for elem in generator: # print(elem)