Commit d0644875 authored by Jan Reimes's avatar Jan Reimes
Browse files

chore(args): remove GraphQueryArgument and ai_graph command 

- Deleted GraphQueryArgument from args.py
- Removed ai_graph command from ai.py
- Simplified CLI by eliminating unused graph query functionality
parent 778bf818

src/tdoc_crawler/cli/ai.py

+0 −28
Original line number Diff line number Diff line
@@ -50,7 +50,6 @@ from tdoc_crawler.cli.args import ( 
    EmbeddingBackendOption,

    EmbeddingTopKOption,

    EndDateOption,

    GraphQueryArgument,

    JsonOutputOption,

    ProcessAllOption,

    ProcessForceOption,
@@ -323,33 +322,6 @@ def ai_status( 
            console.print(table)





@ai_app.command("graph")

def ai_graph(

    query_text: GraphQueryArgument = None,

    workspace: WorkspaceNameOption = None,

    json_output: JsonOutputOption = False,

) -> None:

    """Query the knowledge graph for a workspace."""

    if not query_text:

        console.print("[red]Error: query is required (positional or --query).[/red]")

        raise typer.Exit(1)



    _get_cache_manager()  # Ensure cache manager is registered before config loads

    # Tests expect None as default, so don't convert to "default"

    result = query_graph(query_text, workspace=workspace)



    if json_output:

        typer.echo(json.dumps(result))

    else:

        answer = result.get("answer", "")

        node_count = len(result.get("nodes", []))

        edge_count = len(result.get("edges", []))



        console.print(f"[bold]Query:[/bold] {query_text}")

        console.print(f"[bold]Answer:[/bold] {answer}")

        console.print(f"[dim](nodes: {node_count}, edges: {edge_count})[/dim]")





# Workspace management subcommands

_workspace_app = typer.Typer(help="Manage GraphRAG workspaces")

src/tdoc_crawler/cli/args.py

+0 −2
Original line number Diff line number Diff line
@@ -185,8 +185,6 @@ ProcessLimitOption = Annotated[int | None, typer.Option("--limit", "-l", help="L 


StatusTDocIdOption = Annotated[str | None, typer.Option("--tdoc-id", "-t", help="TDoc ID to check status for")]



GraphQueryArgument = Annotated[str | None, typer.Argument(help="Graph query string")]



WorkspaceNameArgument = Annotated[str, typer.Argument(help="Workspace name")]

WorkspaceActivateArgument = Annotated[str, typer.Argument(help="Workspace name to activate")]

WorkspaceAutoBuildOption = Annotated[