Skip to content

Commit 4281f57

Browse files
committed
feat:qa knowledge recommend question add error code
1 parent a698c45 commit 4281f57

2 files changed

Lines changed: 7 additions & 2 deletions

File tree

src/backend/bisheng/api/services/knowledge_imp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
from bisheng.common.errcode import BaseErrorCode
4141
from bisheng.common.errcode.knowledge import KnowledgeSimilarError, KnowledgeFileDeleteError, KnowledgeFileEmptyError, \
4242
KnowledgeFileChunkMaxError, KnowledgeLLMError, KnowledgeFileDamagedError, KnowledgeFileNotSupportedError, \
43-
KnowledgeEtl4lmTimeoutError, KnowledgeFileFailedError, KnowledgeExcelChunkMaxError
43+
KnowledgeEtl4lmTimeoutError, KnowledgeFileFailedError, KnowledgeExcelChunkMaxError, KnowledgeRecommendQuestionError
4444
from bisheng.common.schemas.telemetry.event_data_schema import FileParseEventData
4545
from bisheng.common.services import telemetry_service
4646
from bisheng.common.services.config_service import settings
@@ -1567,7 +1567,7 @@ def recommend_question(invoke_user_id: int, question: str, answer: str, number:
15671567
return []
15681568
except Exception as exc:
15691569
logger.error("recommend_question json.loads error:{}", gen_question)
1570-
raise ValueError(gen_question) from exc
1570+
raise KnowledgeRecommendQuestionError(exception=exc, message=gen_question)
15711571

15721572

15731573
def extract_code_blocks(markdown_code_block: str):

src/backend/bisheng/common/errcode/knowledge.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,11 @@ class KnowledgeNotQAError(BaseErrorCode):
9999
Msg: str = 'Is notQAThe knowledge base upon'
100100

101101

102+
class KnowledgeRecommendQuestionError(BaseErrorCode):
103+
Code: int = 10961
104+
Msg: str = 'The model returned an incorrect format: {message}'
105+
106+
102107
# Knowledge base does not exist
103108
class KnowledgeNotExistError(BaseErrorCode):
104109
Code: int = 10970

0 commit comments

Comments
 (0)