mirror of https://github.com/OpenSPG/KAG
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright 2023 OpenSPG Authors
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
|
|
# in compliance with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software distributed under the License
|
|
# is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
|
|
# or implied.
|
|
|
|
from abc import ABC, abstractmethod
|
|
from typing import List
|
|
|
|
from knext.builder.operator.base import BaseOp
|
|
|
|
from kag.builder.model.sub_graph import Node, SubGraph
|
|
|
|
|
|
class FuseOpABC(BaseOp, ABC):
|
|
"""
|
|
Interface for fusing mapped sub graphs with data in storage.
|
|
|
|
It is usually used in mapping component for SPG builder.
|
|
"""
|
|
|
|
@abstractmethod
|
|
def link(self, source: SubGraph) -> List[SubGraph]:
|
|
raise NotImplementedError(
|
|
f"{self.__class__.__name__} need to implement `link` method."
|
|
)
|
|
|
|
@abstractmethod
|
|
def merge(self, source: SubGraph, target: List[SubGraph]) -> List[SubGraph]:
|
|
raise NotImplementedError(
|
|
f"{self.__class__.__name__} need to implement `merge` method."
|
|
)
|
|
|
|
def invoke(self, source: SubGraph) -> List[SubGraph]:
|
|
target = self.link(source)
|
|
return self.merge(source, target)
|
|
|
|
|
|
class LinkOpABC(BaseOp, ABC):
|
|
"""
|
|
Interface for recall nodes in storage by mapped properties.
|
|
|
|
It is usually used in mapping component for SPG builder.
|
|
"""
|
|
|
|
@abstractmethod
|
|
def invoke(self, source: Node, prop_value: str, target_type: str) -> List[Node]:
|
|
raise NotImplementedError(
|
|
f"{self.__class__.__name__} need to implement `invoke` method."
|
|
)
|