From 7e26304e94672bbe10ce0dfca10062cc895f8d7a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=98=A5=E6=B2=B3=E6=99=B4?= Date: Wed, 23 Apr 2025 14:39:51 +0900 Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8=E6=97=A7=E5=BC=8F=E8=81=94?= =?UTF-8?q?=E5=90=88=E4=BD=93=E5=86=99=E6=B3=95=E4=BB=A5=E5=85=BC=E5=AE=B9?= =?UTF-8?q?python3.9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/plugins/knowledge/src/ie_process.py | 4 ++-- src/plugins/knowledge/src/qa_manager.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/plugins/knowledge/src/ie_process.py b/src/plugins/knowledge/src/ie_process.py index 3e53e4b2d..0bbe21698 100644 --- a/src/plugins/knowledge/src/ie_process.py +++ b/src/plugins/knowledge/src/ie_process.py @@ -1,6 +1,6 @@ import json import time -from typing import List +from typing import List, Union from .global_logger import logger from . import prompt_template @@ -64,7 +64,7 @@ def _rdf_triple_extract(llm_client: LLMClient, paragraph: str, entities: list) - def info_extract_from_str( llm_client_for_ner: LLMClient, llm_client_for_rdf: LLMClient, paragraph: str -) -> tuple[None, None] | tuple[list[str], list[list[str]]]: +) -> Union[tuple[None, None], tuple[list[str], list[list[str]]]]: try_count = 0 while True: try: diff --git a/src/plugins/knowledge/src/qa_manager.py b/src/plugins/knowledge/src/qa_manager.py index 22460651a..9cb5c0186 100644 --- a/src/plugins/knowledge/src/qa_manager.py +++ b/src/plugins/knowledge/src/qa_manager.py @@ -1,5 +1,5 @@ import time -from typing import Tuple, List, Dict +from typing import Tuple, List, Dict, Optional from .global_logger import logger @@ -28,7 +28,7 @@ class QAManager: "qa": llm_client_qa, } - def process_query(self, question: str) -> Tuple[List[Tuple[str, float, float]], Dict[str, float] | None]: + def process_query(self, question: str) -> Tuple[List[Tuple[str, float, float]], Optional[Dict[str, float]]]: """处理查询""" # 生成问题的Embedding