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

minor fixes in add screen

Browse source
This commit is contained in:
rajeevrajeshuni 2025-11-30 08:16:36 +05:30
parent 192c2b7363
commit 1b8ac854dd
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
cognee/cli/tui/add_screen.py
View file

@ -1,5 +1,4 @@
import asyncio import asyncio
import cognee
from textual.app import ComposeResult from textual.app import ComposeResult
from textual.widgets import Input, Label, Static, TextArea from textual.widgets import Input, Label, Static, TextArea
from textual.containers import Container, Vertical from textual.containers import Container, Vertical
@ -34,7 +33,7 @@ class AddTUIScreen(BaseTUIScreen):
with Container(classes="tui-title-wrapper"): with Container(classes="tui-title-wrapper"):
yield Static("📥 Add Data to Cognee", classes="tui-title-bordered") yield Static("📥 Add Data to Cognee", classes="tui-title-bordered")
with Vertical(classes="tui-form"): with Vertical(classes="tui-form"):
yield Label("Data (text, file path, URL, or S3 path):", classes="tui-label-spaced") yield Label("Data (text, file path (/path/to/doc), URL, or S3 path (s3://bucket)):", classes="tui-label-spaced")
yield TextArea( yield TextArea(
"", "",
id="data-input", id="data-input",
@ -105,6 +104,8 @@ class AddTUIScreen(BaseTUIScreen):
status = self.query_one(".tui-status", Static) status = self.query_one(".tui-status", Static)
try: try:
import cognee
await cognee.add(data=data, dataset_name=dataset_name) await cognee.add(data=data, dataset_name=dataset_name)
status.update(f"[green]✓ Successfully added data to dataset '{dataset_name}'[/green]") status.update(f"[green]✓ Successfully added data to dataset '{dataset_name}'[/green]")