ソースを参照

fix: import undefined types

Yeuoly 7 ヶ月 前
コミット
635a53ea38
共有1 個のファイルを変更した6 個の追加5 個の削除を含む
  1. 6 5
      api/core/plugin/manager/model.py

+ 6 - 5
api/core/plugin/manager/model.py

@@ -10,13 +10,14 @@ from core.model_runtime.entities.text_embedding_entities import TextEmbeddingRes
 from core.model_runtime.utils.encoders import jsonable_encoder
 from core.plugin.entities.plugin_daemon import (
     PluginBasicBooleanResponse,
+    PluginDaemonInnerError,
     PluginModelProviderEntity,
     PluginModelSchemaEntity,
     PluginNumTokensResponse,
     PluginStringResultResponse,
     PluginVoicesResponse,
 )
-from core.plugin.manager.base import BasePluginManager, PluginDaemonRespError
+from core.plugin.manager.base import BasePluginManager
 
 
 class PluginModelManager(BasePluginManager):
@@ -179,8 +180,8 @@ class PluginModelManager(BasePluginManager):
 
         try:
             yield from response
-        except PluginDaemonRespError as e:
-            raise ValueError(e.resp_message + str(e.code))
+        except PluginDaemonInnerError as e:
+            raise ValueError(e.message + str(e.code))
 
     def get_llm_num_tokens(
         self,
@@ -395,8 +396,8 @@ class PluginModelManager(BasePluginManager):
             for result in response:
                 hex_str = result.result
                 yield binascii.unhexlify(hex_str)
-        except PluginDaemonRespError as e:
-            raise ValueError(e.resp_message + str(e.code))
+        except PluginDaemonInnerError as e:
+            raise ValueError(e.message + str(e.code))
 
     def get_tts_model_voices(
         self,