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