change import to avoid circular imports
This commit is contained in:
parent
29959ada7f
commit
ad6bdd2be3
1 changed files with 4 additions and 4 deletions
|
|
@ -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,
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue