mirror of
https://github.com/OpenSPG/openspg.git
synced 2025-08-07 00:12:45 +00:00
24 lines
931 B
Python
24 lines
931 B
Python
from typing import List
|
|
|
|
from knext.client.search import SearchClient
|
|
from knext.operator.op import PredictOp
|
|
from knext.operator.spg_record import SPGRecord
|
|
|
|
|
|
class IndicatorPredict(PredictOp):
|
|
|
|
bind_to = ("Financial.State", "derivedFrom", "Financial.Indicator")
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.search_client = SearchClient("Financial.Indicator")
|
|
|
|
def invoke(self, subject_record: SPGRecord) -> List[SPGRecord]:
|
|
print("##########IndicatorPredict###########")
|
|
query = {"match": {"name": subject_record.get_property("name", "")}}
|
|
recall_records = self.search_client.search(query, start=0, size=10)
|
|
if recall_records is not None and len(recall_records) > 0:
|
|
rerank_record = SPGRecord("Financial.Indicator", {"id": recall_records[0].doc_id, "name": recall_records[0].properties.get("name", "")})
|
|
return [rerank_record]
|
|
return []
|