src/inference.py

from .tokenizer import SimpleTokenizer
from .intent_extractor import IntentExtractor
from .graph_builder import GraphBuilder

class IntentGraphPipeline:
    def __init__(self):
        self.tokenizer = SimpleTokenizer()
        self.extractor = IntentExtractor()
        self.builder = GraphBuilder()

    def __call__(self, text: str):
        tokens = self.tokenizer.tokenize(text)
        intents = self.extractor.extract(tokens)
        graph = self.builder.build(intents)
        return {'intents': intents, 'nodes': list(graph.nodes), 'edges': list(graph.edges)}