BUG: fix client import (#1905)

Co-authored-by: wuzhaoxin <[email protected]>
xorbitsai · Jul 19, 2024 · 880929c · 880929c
1 parent 35c3008
commit 880929c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 12 deletions.
diff --git a/xinference/api/restful_api.py b/xinference/api/restful_api.py
@@ -52,11 +52,7 @@
 
 from .._compat import BaseModel, Field
 from .._version import get_versions
-from ..constants import (
-    XINFERENCE_AUDIO_SPEECH_DEFAULT_STREAM,
-    XINFERENCE_DEFAULT_ENDPOINT_PORT,
-    XINFERENCE_DISABLE_METRICS,
-)
+from ..constants import XINFERENCE_DEFAULT_ENDPOINT_PORT, XINFERENCE_DISABLE_METRICS
 from ..core.event import Event, EventCollectorActor, EventType
 from ..core.supervisor import SupervisorActor
 from ..core.utils import json_dumps
@@ -133,7 +129,7 @@ class SpeechRequest(BaseModel):
     voice: Optional[str]
     response_format: Optional[str] = "mp3"
     speed: Optional[float] = 1.0
-    stream: Optional[bool] = XINFERENCE_AUDIO_SPEECH_DEFAULT_STREAM
+    stream: Optional[bool] = False
 
 
 class RegisterModelRequest(BaseModel):

diff --git a/xinference/client/restful/restful_client.py b/xinference/client/restful/restful_client.py
@@ -17,7 +17,6 @@
 
 import requests
 
-from ...constants import XINFERENCE_AUDIO_SPEECH_DEFAULT_STREAM
 from ..common import streaming_response_iterator
 
 if TYPE_CHECKING:
@@ -768,7 +767,7 @@ def speech(
         voice: str = "",
         response_format: str = "mp3",
         speed: float = 1.0,
-        stream: bool = XINFERENCE_AUDIO_SPEECH_DEFAULT_STREAM,
+        stream: bool = False,
     ):
         """
         Generates audio from the input text.

diff --git a/xinference/constants.py b/xinference/constants.py
@@ -30,7 +30,6 @@
 XINFERENCE_ENV_ENABLE_SGLANG = "XINFERENCE_ENABLE_SGLANG"
 XINFERENCE_ENV_DISABLE_METRICS = "XINFERENCE_DISABLE_METRICS"
 XINFERENCE_ENV_TRANSFORMERS_ENABLE_BATCHING = "XINFERENCE_TRANSFORMERS_ENABLE_BATCHING"
-XINFERENCE_ENV_AUDIO_SPEECH_DEFAULT_STREAM = "XINFERENCE_AUDIO_SPEECH_DEFAULT_STREAM"
 
 
 def get_xinference_home() -> str:
@@ -81,6 +80,3 @@ def get_xinference_home() -> str:
 XINFERENCE_TRANSFORMERS_ENABLE_BATCHING = bool(
     int(os.environ.get(XINFERENCE_ENV_TRANSFORMERS_ENABLE_BATCHING, 0))
 )
-XINFERENCE_AUDIO_SPEECH_DEFAULT_STREAM = bool(
-    int(os.environ.get(XINFERENCE_ENV_AUDIO_SPEECH_DEFAULT_STREAM, 0))
-)