handle codestral@2405 instability

tests/local_testing/test_amazing_vertex_completion.py

@@ -993,6 +993,8 @@ async def test_partner_models_httpx(model, sync_mode):
         pass
     except litellm.InternalServerError as e:
         pass
+    except litellm.APIConnectionError as e:
+        pass
     except Exception as e:
         if "429 Quota exceeded" in str(e):
             pass

tests/local_testing/test_text_completion.py

@@ -4190,6 +4190,9 @@ async def test_completion_codestral_fim_api_stream(model):
         # cost = litellm.completion_cost(completion_response=response)
         # print("cost to make mistral completion=", cost)
         # assert cost > 0.0
+    except litellm.APIConnectionError as e:
+        print(e)
+        pass
     except Exception as e:
         pytest.fail(f"Error occurred: {e}")