gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

code review changes

Browse source
This commit is contained in:
rajeevrajeshuni 2025-12-29 20:18:50 +05:30
parent 7d3586e1b2
commit ac8d130af2
3 changed files with 13 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/cli/commands/tui_command.py
View file

@ -3,7 +3,6 @@ from cognee.cli import SupportsCliCommand
from cognee.cli.config import DEFAULT_DOCS_URL from cognee.cli.config import DEFAULT_DOCS_URL
import cognee.cli.echo as fmt import cognee.cli.echo as fmt
from cognee.cli.exceptions import CliCommandException from cognee.cli.exceptions import CliCommandException
from cognee.cli.tui.home_screen import HomeScreen
class TuiCommand(SupportsCliCommand): class TuiCommand(SupportsCliCommand):
@ -26,6 +25,7 @@ class TuiCommand(SupportsCliCommand):
def execute(self, args: argparse.Namespace) -> None: def execute(self, args: argparse.Namespace) -> None:
try: try:
from textual.app import App from textual.app import App
from cognee.cli.tui.home_screen import HomeScreen
from cognee.shared.logging_utils import setup_logging from cognee.shared.logging_utils import setup_logging
class CogneeTUI(App): class CogneeTUI(App):
"""Main TUI application for cognee.""" """Main TUI application for cognee."""

14
cognee/cli/tui/delete_screen.py
View file

@ -126,16 +126,18 @@ class DeleteTUIScreen(BaseTUIScreen):
"""Async function to delete data.""" """Async function to delete data."""
status = self.query_one(".tui-status", Static) status = self.query_one(".tui-status", Static)
try: try:
if user_id is None:
user = await get_default_user()
resolved_user_id = user.id
else:
resolved_user_id = UUID(user_id)
if dataset_name: if dataset_name:
if user_id is None:
user = await get_default_user()
resolved_user_id = user.id
else:
resolved_user_id = UUID(user_id)
await delete_dataset_by_name(dataset_name, resolved_user_id) await delete_dataset_by_name(dataset_name, resolved_user_id)
status.update(f"✓ Successfully deleted dataset '{dataset_name}'.")
else: else:
await delete_data_by_user(resolved_user_id) await delete_data_by_user(resolved_user_id)
status.update(f"✓ Successfully deleted dataset '{dataset_name}'.") status.update(f"✓ Successfully deleted all data for user {resolved_user_id}.")
except Exception as e: except Exception as e:
status.update(f"✗ Error: {str(e)}") status.update(f"✗ Error: {str(e)}")
finally: finally:

8
tests/reproduce_tui_search_logic.py
View file

@ -139,11 +139,11 @@ async def test_empty_graph_handling():
# Check that notify was called with warning # Check that notify was called with warning
# We allow flexible matching for the exact message but check 'warning' severity # We allow flexible matching for the exact message but check 'warning' severity
assert screen.notify.called, "notify was not called"
args = screen.notify.call_args args = screen.notify.call_args
if args: assert args is not None
assert args[1].get('severity') == 'warning' or 'Knowledge graph is empty' in args[0][0] # Assert specific conditions on args
else: assert args[1].get('severity') == 'warning' or 'Knowledge graph is empty' in args[0][0]
print("FAIL: notify was not called")
print("SUCCESS: EntityNotFoundError was caught and handled correctly.") print("SUCCESS: EntityNotFoundError was caught and handled correctly.")