From b8927704e30b8b91023e43e0ccf11387b25880ca Mon Sep 17 00:00:00 2001 From: rajeevrajeshuni Date: Sun, 30 Nov 2025 08:18:17 +0530 Subject: [PATCH] change import to avoid circular imports --- cognee/cli/tui/cognify_screen.py | 4 +--- cognee/cli/tui/search_screen.py | 5 +---- 2 files changed, 2 insertions(+), 7 deletions(-) diff --git a/cognee/cli/tui/cognify_screen.py b/cognee/cli/tui/cognify_screen.py index 844cb345b..1108d6a7d 100644 --- a/cognee/cli/tui/cognify_screen.py +++ b/cognee/cli/tui/cognify_screen.py @@ -1,6 +1,4 @@ import asyncio - -import cognee from cognee.modules.chunking.TextChunker import TextChunker from textual.app import ComposeResult from textual.widgets import Input, Label, Static, Checkbox, RadioSet, RadioButton @@ -138,7 +136,7 @@ class CognifyTUIScreen(BaseTUIScreen): # Prepare datasets parameter datasets = [dataset_name] if dataset_name else None - + import cognee await cognee.cognify( datasets=datasets, chunker=chunker_class, diff --git a/cognee/cli/tui/search_screen.py b/cognee/cli/tui/search_screen.py index e04165b60..831178842 100644 --- a/cognee/cli/tui/search_screen.py +++ b/cognee/cli/tui/search_screen.py @@ -1,12 +1,9 @@ import asyncio - -import cognee from cognee.modules.search.types import SearchType from textual.app import ComposeResult from textual.widgets import Input, Label, Static, Select from textual.containers import Container, Vertical, ScrollableContainer from textual.binding import Binding - from cognee.cli.tui.base_screen import BaseTUIScreen @@ -141,7 +138,7 @@ class SearchTUIScreen(BaseTUIScreen): try: # Convert string to SearchType enum search_type = SearchType[query_type] - + import cognee # Perform search results = await cognee.search( query_text=query_text,