Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion graphrag_sdk/ontology.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@
from typing import Optional, Union
from graphrag_sdk.source import AbstractSource
from graphrag_sdk.models import GenerativeModel
from graphrag_sdk.steps.create_ontology_step import CreateOntologyStep
from .attribute import Attribute, AttributeType

logger = logging.getLogger(__name__)
Expand Down
1 change: 0 additions & 1 deletion graphrag_sdk/steps/create_ontology_step.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@
from threading import Lock
from typing import Optional
from graphrag_sdk.steps.Step import Step
from graphrag_sdk.document import Document
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

After removing Document, _process_source’s docstring still documents a document (Document) parameter that isn’t in the signature; consider updating the docstring to match (source: AbstractSource) to avoid misleading API docs.

Fix This in Augment

🤖 Was this useful? React with 👍 or 👎

from graphrag_sdk.ontology import Ontology
from graphrag_sdk.helpers import extract_json
from ratelimit import limits, sleep_and_retry
Expand Down
Loading