mirror of
https://github.com/OpenSPG/openspg.git
synced 2025-07-29 20:12:52 +00:00
39 lines
1.4 KiB
Python
39 lines
1.4 KiB
Python
from typing import List
|
|
|
|
from knext.client.search import SearchClient
|
|
from knext.operator.op import FuseOp
|
|
from knext.operator.spg_record import SPGRecord
|
|
|
|
|
|
class StateFuse(FuseOp):
|
|
|
|
bind_to = "Financial.State"
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.search_client = SearchClient("Financial.State")
|
|
|
|
def invoke(self, subject_records: List[SPGRecord]) -> List[SPGRecord]:
|
|
print("####################StateFuse(状态融合)#####################")
|
|
print("StateFuse(Input): ")
|
|
print("----------------------")
|
|
[print(r) for r in subject_records]
|
|
fused_records = []
|
|
for record in subject_records:
|
|
query = {"match": {"name": 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.State",
|
|
{
|
|
"id": recall_records[0].doc_id,
|
|
"name": recall_records[0].properties.get("name", ""),
|
|
},
|
|
)
|
|
rerank_record.update_property("name", record.get_property("name"))
|
|
fused_records.append(rerank_record)
|
|
print("StateFuse(Output): ")
|
|
print("----------------------")
|
|
[print(r) for r in fused_records]
|
|
return fused_records
|