Revert "add delay to avoid rate limits" & Fix curricular import

[galshubeli]

Summary by CodeRabbit

• Chores
  • Adjusted internal imports and removed an unused import to simplify module structure.
• Bug Fixes / Behavior
  • Source processing now runs continuously without an optional pause between items.
  • Simplified processing flow by removing a redundant post-processing step that re-added processed sources.

_{✏️ Tip: You can customize this high-level summary in your review settings.}

[coderabbitai]

📝 Walkthrough

Walkthrough

Removed unused imports and simplified source processing: moved a CreateOntologyStep import into a local scope, dropped an unused Document import, and removed the delay parameter plus per-source sleep and re-adding behavior in process_sources.

Changes

Cohort / File(s)	Summary
Ontology import cleanup graphrag_sdk/ontology.py, graphrag_sdk/steps/create_ontology_step.py	Moved CreateOntologyStep import into from_sources to avoid circular import; removed unused Document import. No API surface changes.
KnowledgeGraph processing change graphrag_sdk/kg.py	Removed delay parameter from process_sources, eliminated time.sleep(delay) and the post-processing step that re-added processed sources; docstring/signature updated.

Sequence Diagram(s)

(omitted)

Estimated code review effort

🎯 3 (Moderate) | ⏱️ ~20 minutes

Poem

🐰 I hopped through bytes and trimmed a thread,
Moved a snagged import to where it led.
Sleep removed from the source-bound run,
Fewer hops now—my work is done.
Crunch, nibble, ship—hooray, let's have fun! 🥕

🚥 Pre-merge checks | ✅ 2 | ❌ 1

❌ Failed checks (1 inconclusive)

Check name	Status	Explanation	Resolution
Title check	❓ Inconclusive	The title partially addresses the changeset—it mentions reverting a delay feature and fixing imports, but contains a typo ('curricular' instead of 'circular') and doesn't fully clarify that the delay removal is a revert while imports are a fix.	Correct the typo to 'circular' and clarify whether both changes are equally important or if one is the primary focus (e.g., 'Fix circular imports and remove delay parameter').

✅ Passed checks (2 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Docstring Coverage	✅ Passed	Docstring coverage is 100.00% which is sufficient. The required threshold is 80.00%.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

✨ Finishing touches

• 📝 Generate docstrings

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

[augmentcode]

🤖 Augment PR Summary

Summary: Cleans up imports to prevent unused/circular import issues.
Changes: Removes direct imports of CreateOntologyStep and Document, relying on the package export and local usage instead.

_{🤖 Was this summary useful? React with 👍 or 👎}

[augmentcode]

Review completed. 1 suggestions posted.

Comment augment review to trigger a new review at any time.

[augmentcode]

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.

_{🤖 Was this useful? React with 👍 or 👎}

[coderabbitai]

Actionable comments posted: 1

🤖 Fix all issues with AI agents

In `@graphrag_sdk/kg.py`:
- Around line 204-214: The public method process_sources currently removed the
delay parameter which can break callers; restore backwards compatibility by
reintroducing a delay parameter (e.g., delay: Optional[float] = None) to the
process_sources signature in class/def process_sources and, inside the method,
accept and ignore it while issuing a DeprecationWarning (or logging a warning)
when delay is not None; ensure the warning text references process_sources and
suggests the new behavior so callers have a migration path.