outman/dify
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: import undefined types

Browse Source
This commit is contained in:
Yeuoly 2024-09-29 13:23:14 +08:00
parent 7b76b1ff82
commit 635a53ea38
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/core/plugin/manager/model.py
View File

@ -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,