From 4344ae66b8352efffd613e2a0a38f24753123935 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 6 Nov 2023 09:10:48 -0800 Subject: [PATCH] (test) text completion --- litellm/tests/test_text_completion.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/litellm/tests/test_text_completion.py b/litellm/tests/test_text_completion.py index 51f3e260823..c0ba750a3a5 100644 --- a/litellm/tests/test_text_completion.py +++ b/litellm/tests/test_text_completion.py @@ -31,7 +31,7 @@ def test_completion_openai_prompt(): #print(response.choices[0].text) except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_completion_openai_prompt() +test_completion_openai_prompt() def test_completion_openai_prompt(): @@ -47,7 +47,7 @@ def test_completion_openai_prompt(): #print(response.choices[0].text) except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_completion_openai_prompt() +test_completion_openai_prompt() def test_text_completion_basic(): @@ -64,7 +64,7 @@ def test_text_completion_basic(): response_str = response["choices"][0]["text"] except Exception as e: pytest.fail(f"Error occurred: {e}") -# test_text_completion_basic() +test_text_completion_basic() # def test_completion_hf_prompt_array(): # try: @@ -101,14 +101,15 @@ def test_completion_text_003_prompt_array(): test_completion_text_003_prompt_array() +# not including this in our ci cd pipeline, since we don't want to fail tests due to an unstable replit # def test_text_completion_with_proxy(): # try: # litellm.set_verbose=True # response = text_completion( # model="facebook/opt-125m", # prompt='Write a tagline for a traditional bavarian tavern', -# api_base="https://openai-proxy.berriai.repl.co//v1", -# custom_llm_provider="custom_openai", +# api_base="https://openai-proxy.berriai.repl.co/v1", +# custom_llm_provider="openai", # temperature=0, # max_tokens=10, # )