From 94ce524c632f01761b77d253ac3d4ba0d67a6265 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 18 Jan 2024 17:54:16 -0800
Subject: [PATCH] test(test_completion.py): handle together ai timeout

---
 litellm/tests/test_completion.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/litellm/tests/test_completion.py b/litellm/tests/test_completion.py
index 71884bde7..57388fad7 100644
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@@ -1269,6 +1269,8 @@ def test_completion_together_ai():
             "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}")