From 01e6c7ed89101c8eadf7699135494a7e3f0a6788 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 22 Feb 2024 21:33:34 -0800
Subject: [PATCH] test(test_completion.py): remove deprecated together ai model

---
 litellm/tests/test_completion.py | 29 -----------------------------
 1 file changed, 29 deletions(-)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 7816c391890..68821f8e37c 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -1306,35 +1306,6 @@ def test_replicate_custom_prompt_dict():
 
 
 ######## Test TogetherAI ########
-def test_completion_together_ai():
-    model_name = "together_ai/togethercomputer/CodeLlama-13b-Instruct"
-    try:
-        messages = [
-            {"role": "user", "content": "Who are you"},
-            {"role": "assistant", "content": "I am your helpful assistant."},
-            {"role": "user", "content": "Tell me a joke"},
-        ]
-        response = completion(
-            model=model_name,
-            messages=messages,
-            max_tokens=256,
-            n=1,
-            logger_fn=logger_fn,
-        )
-        # Add any assertions here to check the response
-        print(response)
-        cost = completion_cost(completion_response=response)
-        assert cost > 0.0
-        print(
-            "Cost for completion call together-computer/llama-2-70b: ",
-            f"${float(cost):.10f}",
-        )
-    except litellm.Timeout as e:
-        pass
-    except Exception as e:
-        pytest.fail(f"Error occurred: {e}")
-
-
 @pytest.mark.skip(reason="Skip flaky test")
 def test_completion_together_ai_mixtral():
     model_name = "together_ai/DiscoResearch/DiscoLM-mixtral-8x7b-v2"