gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

change import to avoid circular imports

Browse source
This commit is contained in:
rajeevrajeshuni 2025-11-30 08:26:33 +05:30
parent b8927704e3
commit 29959ada7f
1 changed files with 9 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
cognee/cli/tui/cognify_screen.py
View file

@ -1,23 +1,11 @@
import asyncio
from cognee.modules.chunking.TextChunker import TextChunker
from textual.app import ComposeResult
from textual.widgets import Input, Label, Static, Checkbox, RadioSet, RadioButton
from textual.containers import Container, Vertical
from textual.binding import Binding
from cognee.cli.tui.base_screen import BaseTUIScreen
from cognee.cli.config import CHUNKER_CHOICES
try:
from cognee.modules.chunking.LangchainChunker import LangchainChunker
except ImportError:
LangchainChunker = None
try:
from cognee.modules.chunking.CsvChunker import CsvChunker
except ImportError:
CsvChunker = None
class CognifyTUIScreen(BaseTUIScreen):
"""TUI screen for cognifying data in cognee."""
@ -119,16 +107,24 @@ class CognifyTUIScreen(BaseTUIScreen):
async def _cognify_async(self, dataset_name: str | None, chunker_type: str, run_background: bool) -> None:
"""Async function to cognify data."""
status = self.query_one(".tui-status", Static)
from cognee.modules.chunking.TextChunker import TextChunker
try:
# Get chunker class
chunker_class = TextChunker
if chunker_type == "LangchainChunker":
try:
from cognee.modules.chunking.LangchainChunker import LangchainChunker
except ImportError:
LangchainChunker = None
if LangchainChunker is not None:
chunker_class = LangchainChunker
else:
status.update("[yellow]⚠ LangchainChunker not available, using TextChunker[/yellow]")
elif chunker_type == "CsvChunker":
try:
from cognee.modules.chunking.CsvChunker import CsvChunker
except ImportError:
CsvChunker = None
if CsvChunker is not None:
chunker_class = CsvChunker
else: