fix: import undefined types

pull/9184/head
Yeuoly 2 years ago
parent 7b76b1ff82
commit 635a53ea38
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61

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

Loading…
Cancel
Save