diff --git a/model-providers/model_providers/core/model_runtime/model_providers/xinference/llm/llm.py b/model-providers/model_providers/core/model_runtime/model_providers/xinference/llm/llm.py
index c2551230..8f405a35 100644
--- a/model-providers/model_providers/core/model_runtime/model_providers/xinference/llm/llm.py
+++ b/model-providers/model_providers/core/model_runtime/model_providers/xinference/llm/llm.py
@@ -1,5 +1,5 @@
 from collections.abc import Generator, Iterator
-from typing import Dict, List, Union, cast
+from typing import Dict, List, Union, cast, Type
 
 from openai import (
     APIConnectionError,