Skip to content

Expose use_combined_extraction on add_episode_bulk #1803

Description

@alekc

Description

extract_nodes_and_edges_bulk (added in #1432) already dispatches between combined single-call extraction and the separate two-call path via use_combined_extraction, but Graphiti.add_episode_bulk never forwards the parameter, so combined extraction is unreachable from any public entry point. _extract_and_dedupe_nodes_bulk, the only caller, hardcodes the default too.

Possible Solution

Add use_combined_extraction: bool = False to add_episode_bulk and thread it through _extract_and_dedupe_nodes_bulk to extract_nodes_and_edges_bulk. Backward compatible, default unchanged. I have a small patch ready (plus tests for the combined/separate dispatch in extract_nodes_and_edges_bulk, which had no coverage either way before this) and can open a PR against this issue.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions