From 5fc9524a466f3dd93d5f140abe3314787a70970e Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Fri, 29 Dec 2023 11:34:15 +0530 Subject: [PATCH] (test) test cloudflare completion --- litellm/tests/test_completion.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) 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)