From 102de97960b5b35b5b0a8e10bbe651fc87d01890 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Wed, 6 Dec 2023 11:46:09 -0800
Subject: [PATCH] refactor: fix linting errors

---
 litellm/llms/sagemaker.py     | 6 +++---
 litellm/proxy/proxy_server.py | 5 +++++
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/litellm/llms/sagemaker.py b/litellm/llms/sagemaker.py
index 36324286b32..96d06bc20d4 100644
--- a/litellm/llms/sagemaker.py
+++ b/litellm/llms/sagemaker.py
@@ -5,7 +5,7 @@ import requests
 import time
 from typing import Callable, Optional
 import litellm
-from litellm.utils import ModelResponse, get_secret, Usage
+from litellm.utils import ModelResponse, EmbeddingResponse, get_secret, Usage
 import sys
 from copy import deepcopy
 import httpx
@@ -198,7 +198,7 @@ def completion(
 
 def embedding(model: str,
         input: list,
-        model_response: ModelResponse,
+        model_response: EmbeddingResponse,
         print_verbose: Callable,
         encoding,
         logging_obj,
@@ -262,7 +262,7 @@ def embedding(model: str,
         ContentType="application/json",
         Body={data},
         CustomAttributes="accept_eula=true",
-    )"""
+    )""" # type: ignore
     logging_obj.pre_call(
             input=input,
             api_key="",
diff --git a/litellm/proxy/proxy_server.py b/litellm/proxy/proxy_server.py
index 4747866691f..23b39415d68 100644
--- a/litellm/proxy/proxy_server.py
+++ b/litellm/proxy/proxy_server.py
@@ -1175,6 +1175,11 @@ async def test_endpoint(request: Request):
 async def health_endpoint(request: Request, model: Optional[str] = fastapi.Query(None, description="Specify the model name (optional)")):
     global llm_model_list
 
+    if llm_model_list is None: 
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail={"error": "Model list not initialized"},
+        )
     healthy_endpoints, unhealthy_endpoints = await perform_health_check(llm_model_list, model)
 
     return {