fix(low-code cdk): pass state_migrations when creating concurrent cursor in _group_streams

[darynaishchenko]

Added missing state migration to create_concurrent_cursor_from_datetime_based_cursor when creating cursor in ConcurrentDeclarativeSource

Summary by CodeRabbit

• Bug Fixes
  • Improved handling of stream state migrations during concurrent incremental syncs using datetime-based cursors.

[coderabbitai]

📝 Walkthrough

Walkthrough

This change updates the _group_streams method in concurrent_declarative_source.py to ensure that the stream_state_migrations attribute from a declarative_stream is now passed to the create_concurrent_cursor_from_datetime_based_cursor method. This adjustment is applied specifically in the code path where a datetime-based cursor is used for incremental syncs, and no other logic or public interfaces are modified.

Changes

File(s)	Change Summary
airbyte_cdk/sources/declarative/concurrent_declarative_source.py	Passes stream_state_migrations from declarative_stream to create_concurrent_cursor_from_datetime_based_cursor within _group_streams when handling datetime-based cursors. No other logic or public signatures changed.

Sequence Diagram(s)

sequenceDiagram
    participant DeclarativeStream
    participant ConcurrentDeclarativeSource
    participant CursorFactory

    DeclarativeStream->>ConcurrentDeclarativeSource: Provide stream with state migrations
    ConcurrentDeclarativeSource->>CursorFactory: create_concurrent_cursor_from_datetime_based_cursor(..., stream_state_migrations)
    CursorFactory-->>ConcurrentDeclarativeSource: Return cursor with migrations

Loading

Possibly related PRs

• chore: fix typing issues #135: Both PRs modify the _group_streams method in ConcurrentDeclarativeSource, focusing on incremental sync and stream state handling.
• feat: add IncrementingCountCursor #346: Both PRs involve changes to cursor creation logic in _group_streams, with attention to datetime-based and count-based cursors and state migration integration.
• fix: (low code)run state migrations for concurrent streams #316: Both PRs address passing and handling of stream_state_migrations within concurrent streams, modifying the same method and file. Wdyt about reviewing these together for consistency?

Suggested labels

bug

Suggested reviewers

• maxi297

✨ 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

🪧 Tips

Chat

There are 3 ways to chat with CodeRabbit:

‼️ IMPORTANT
Auto-reply has been disabled for this repository in the CodeRabbit settings. The CodeRabbit bot will not respond to your replies unless it is explicitly tagged.

• Files and specific lines of code (under the "Files changed" tab): Tag @coderabbitai in a new review comment at the desired location with your query. Examples:
  • @coderabbitai generate unit testing code for this file.
  • @coderabbitai modularize this function.
• PR comments: Tag @coderabbitai in a new PR comment to ask questions about the PR branch. For the best results, please provide a very specific query, as very limited context is provided in this mode. Examples:
  • @coderabbitai gather interesting stats about this repository and render them as a table. Additionally, render a pie chart showing the language distribution in the codebase.
  • @coderabbitai read src/utils.ts and generate unit testing code.
  • @coderabbitai read the files in the src/scheduler package and generate a class diagram using mermaid and a README in the markdown format.
  • @coderabbitai help me debug CodeRabbit configuration file.

Note: Be mindful of the bot's finite context window. It's strongly recommended to break down tasks such as reading entire modules into smaller chunks. For a focused discussion, use review comments to chat about specific files and their changes, instead of using the PR comments.

CodeRabbit Commands (Invoked using PR comments)

• @coderabbitai pause to pause the reviews on a PR.
• @coderabbitai resume to resume the paused reviews.
• @coderabbitai review to trigger an incremental review. This is useful when automatic reviews are disabled for the repository.
• @coderabbitai full review to do a full review from scratch and review all the files again.
• @coderabbitai summary to regenerate the summary of the PR.
• @coderabbitai generate docstrings to generate docstrings for this PR.
• @coderabbitai generate sequence diagram to generate a sequence diagram of the changes in this PR.
• @coderabbitai resolve resolve all the CodeRabbit review comments.
• @coderabbitai configuration to show the current CodeRabbit configuration for the repository.
• @coderabbitai help to get help.

Other keywords and placeholders

• Add @coderabbitai ignore anywhere in the PR description to prevent this PR from being reviewed.
• Add @coderabbitai summary to generate the high-level summary at a specific location in the PR description.
• Add @coderabbitai anywhere in the PR title to generate the title automatically.

CodeRabbit Configuration File (.coderabbit.yaml)

• You can programmatically configure CodeRabbit by adding a .coderabbit.yaml file to the root of your repository.
• Please see the configuration documentation for more information.
• If your editor has YAML language server enabled, you can add the path at the top of this file to enable auto-completion and validation: # yaml-language-server: $schema=https://coderabbit.ai/integrations/schema.v2.json

Documentation and Community

• Visit our Documentation for detailed information on how to use CodeRabbit.
• Join our Discord Community to get help, request features, and share feedback.
• Follow us on X/Twitter for updates and announcements.

[coderabbitai]

Actionable comments posted: 1

📜 Review details

Configuration used: CodeRabbit UI
Review profile: CHILL
Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between 4a96a2a and 8dada35.

📒 Files selected for processing (1)

• airbyte_cdk/sources/declarative/concurrent_declarative_source.py (1 hunks)

⏰ Context from checks skipped due to timeout of 90000ms (11)

• GitHub Check: Analyze (python)
• GitHub Check: Analyze (actions)
• GitHub Check: Ruff Format Check
• GitHub Check: Pytest (All, Python 3.11, Ubuntu)
• GitHub Check: Build and Inspect Python Package
• GitHub Check: SDM Docker Image Build
• GitHub Check: Pytest (Fast)
• GitHub Check: MyPy Check
• GitHub Check: Pytest (All, Python 3.10, Ubuntu)
• GitHub Check: preview_docs
• GitHub Check: Dependency Analysis with Deptry