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:33:09 +05:30
parent 29959ada7f
commit ad6bdd2be3
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/cli/tui/search_screen.py
View file

@ -1,12 +1,10 @@
import asyncio
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
class SearchTUIScreen(BaseTUIScreen):
"""Simple search screen with query input and results display."""
@ -136,9 +134,11 @@ class SearchTUIScreen(BaseTUIScreen):
async def _async_search(self, query_text: str, query_type: str) -> None:
"""Async search operation."""
try:
import cognee
from cognee.modules.search.types import SearchType
# Convert string to SearchType enum
search_type = SearchType[query_type]
import cognee
# Perform search
results = await cognee.search(
query_text=query_text,
@ -157,7 +157,7 @@ class SearchTUIScreen(BaseTUIScreen):
if query_type in ["GRAPH_COMPLETION", "RAG_COMPLETION"]:
formatted = "\n\n".join([f"📝 {result}" for result in results])
elif query_type == "CHUNKS":
formatted = "\n\n".join([f"📄 Chunk {i+1}:\n{result}" for i, result in enumerate(results)])
formatted = "\n\n".join([f"📄 Chunk {i + 1}:\n{result}" for i, result in enumerate(results)])
else:
formatted = "\n\n".join([f"• {result}" for result in results])