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 import asyncio
from cognee.modules.search.types import SearchType
from textual.app import ComposeResult from textual.app import ComposeResult
from textual.widgets import Input, Label, Static, Select from textual.widgets import Input, Label, Static, Select
from textual.containers import Container, Vertical, ScrollableContainer from textual.containers import Container, Vertical, ScrollableContainer
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
class SearchTUIScreen(BaseTUIScreen): class SearchTUIScreen(BaseTUIScreen):
"""Simple search screen with query input and results display.""" """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 def _async_search(self, query_text: str, query_type: str) -> None:
"""Async search operation.""" """Async search operation."""
try: try:
import cognee
from cognee.modules.search.types import SearchType
# Convert string to SearchType enum # Convert string to SearchType enum
search_type = SearchType[query_type] search_type = SearchType[query_type]
import cognee
# Perform search # Perform search
results = await cognee.search( results = await cognee.search(
query_text=query_text, query_text=query_text,
@ -157,7 +157,7 @@ class SearchTUIScreen(BaseTUIScreen):
if query_type in ["GRAPH_COMPLETION", "RAG_COMPLETION"]: if query_type in ["GRAPH_COMPLETION", "RAG_COMPLETION"]:
formatted = "\n\n".join([f"📝 {result}" for result in results]) formatted = "\n\n".join([f"📝 {result}" for result in results])
elif query_type == "CHUNKS": 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: else:
formatted = "\n\n".join([f"• {result}" for result in results]) formatted = "\n\n".join([f"• {result}" for result in results])