feat(rag): improve multi-worker support and add search type options

[larryro]

Summary

This PR improves the RAG service's multi-worker support and adds configurable search type options.

Changes

Multi-Worker Lock Conflict Fix

• Add _patch_remote_kuzu_adapter() to override RemoteKuzuAdapter.__init__ and skip local Kuzu database creation that causes file lock conflicts when running multiple uvicorn workers
• Increase RAG_WORKERS from 1 to 2 in Dockerfile now that the lock conflict issue is resolved
• The patch initializes only the minimal attributes needed for the remote adapter without calling the parent's _initialize_connection()

Search Type Support

• Add SearchType enum with CHUNKS, GRAPH_COMPLETION, RAG_COMPLETION, SUMMARIES, GRAPH_SUMMARY_COMPLETION, and TEMPORAL options
• Update search endpoint to accept search_type parameter (defaults to CHUNKS)
• Map API search types to Cognee SearchType for proper query routing

Ingestion Timeout Handling

• Add configurable ingestion_timeout_seconds setting (default: 3 hours)
• Wrap cognee.add() and cognee.cognify() with asyncio.wait_for timeout
• Improve logging for ingestion progress and search operations

Files Changed

• services/rag/Dockerfile
• services/rag/app/config.py
• services/rag/app/models.py
• services/rag/app/routers/search.py
• services/rag/app/services/cognee/config.py
• services/rag/app/services/cognee/service.py

---

Pull Request opened by Augment Code with guidance from the PR author

Summary by CodeRabbit

Release Notes

• New Features

  • Multiple search strategies now available: comprehensive text chunks, knowledge graph-based reasoning, concise AI-generated answers, document summaries, and insights.
  • Configurable document ingestion timeout (default 3 hours).

• Improvements

  • Enhanced stability for multi-worker deployments with improved timeout handling during document ingestion.

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

[coderabbitai]

Caution

Review failed

The pull request is closed.

📝 Walkthrough

Walkthrough

This pull request introduces multi-search-type support and timeout-based document ingestion protection to the RAG service. Changes include: increasing Docker default workers from 1 to 2, adding a configurable ingestion timeout setting, introducing a SearchType enum to support different query strategies (CHUNKS, GRAPH_COMPLETION, RAG_COMPLETION, SUMMARIES, INSIGHTS), implementing asyncio-based timeout wrapping around Cognee ingestion operations, patching RemoteKuzuAdapter to prevent file-lock conflicts in multi-worker setups, and plumbing search type mappings through the service layer.

Estimated code review effort

🎯 3 (Moderate) | ⏱️ ~25 minutes

Areas requiring extra attention:

• services/rag/app/services/cognee/config.py — The RemoteKuzuAdapter patching mechanism with the _tale_patched guard flag requires verification that the patch is applied correctly and idempotently before Cognee import
• services/rag/app/services/cognee/service.py — The timeout handling logic, particularly the remaining-time calculation between add and cognify operations, needs validation to ensure timeouts propagate correctly and the 60-second floor is appropriate
• services/rag/app/services/cognee/service.py — The mapping from API SearchType to Cognee query_type values should be cross-referenced against Cognee's expected input to prevent runtime failures
• services/rag/app/routers/search.py — Verify that the search_type parameter is correctly threaded from the request through to the service layer and that all search type values are documented

---

📜 Recent review details

Configuration used: Organization UI

Review profile: ASSERTIVE

Plan: Pro (Legacy)

📥 Commits

Reviewing files that changed from the base of the PR and between 211baa2 and 7c43d96.

📒 Files selected for processing (6)

• services/rag/Dockerfile (1 hunks)
• services/rag/app/config.py (1 hunks)
• services/rag/app/models.py (1 hunks)
• services/rag/app/routers/search.py (1 hunks)
• services/rag/app/services/cognee/config.py (2 hunks)
• services/rag/app/services/cognee/service.py (6 hunks)

---

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