vectorgov-cli 0.3.2__tar.gz → 0.3.4__tar.gz

{vectorgov_cli-0.3.2 → vectorgov_cli-0.3.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vectorgov-cli
-Version: 0.3.2
+Version: 0.3.4
 Summary: CLI para a API VectorGov - Busca semântica em legislação brasileira
 Project-URL: Homepage, https://vectorgov.io
 Project-URL: Documentation, https://vectorgov.io/documentacao
@@ -25,7 +25,7 @@ Requires-Python: >=3.9
 Requires-Dist: pyyaml>=6.0.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: typer>=0.9.0
-Requires-Dist: vectorgov>=0.13.0
+Requires-Dist: vectorgov>=0.19.1
 Provides-Extra: all
 Requires-Dist: anthropic>=0.18.0; extra == 'all'
 Requires-Dist: google-generativeai>=0.3.0; extra == 'all'

{vectorgov_cli-0.3.2 → vectorgov_cli-0.3.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "vectorgov-cli"
-version = "0.3.2"
+version = "0.3.4"
 description = "CLI para a API VectorGov - Busca semântica em legislação brasileira"
 readme = "README.md"
 license = "MIT"
@@ -37,7 +37,7 @@ classifiers = [
     "Topic :: Text Processing :: Linguistic",
 ]
 dependencies = [
-    "vectorgov>=0.13.0",
+    "vectorgov>=0.19.1",
     "typer>=0.9.0",
     "rich>=13.0.0",
     "pyyaml>=6.0.0",

{vectorgov_cli-0.3.2 → vectorgov_cli-0.3.4}/src/vectorgov/cli/__init__.py

@@ -7,7 +7,7 @@ Uso:
     vectorgov auth login
 """
 
-__version__ = "0.3.2"
+__version__ = "0.3.4"
 
 from .main import app
 

{vectorgov_cli-0.3.2 → vectorgov_cli-0.3.4}/src/vectorgov/cli/commands/fs_search.py

@@ -6,8 +6,10 @@ from rich.console import Console
 from rich.table import Table
 from ..utils.config import ConfigManager
 from ..utils.output import (
+    extract_credits,
     extract_request_id,
     get_evidence_links,
+    print_credits_footer,
     print_request_id_footer,
     render_evidence_lines_text,
     render_llm_output,
@@ -100,6 +102,7 @@ def fs_search(
                 "latency_ms": getattr(result, "latency_ms", None),
                 "query_id": getattr(result, "query_id", None),
                 "request_id": extract_request_id(result),
+                "credits": extract_credits(result),
             }
             print(render_llm_output(hits, query, metadata))
             return
@@ -153,6 +156,7 @@ def fs_search(
         if query_id:
             console.print(f"\n[dim]query_id: {query_id}[/dim]")
         print_request_id_footer(console, extract_request_id(result))
+        print_credits_footer(console, result)
 
     except typer.Exit:
         raise

{vectorgov_cli-0.3.2 → vectorgov_cli-0.3.4}/src/vectorgov/cli/commands/grep_cmd.py

@@ -6,8 +6,10 @@ from rich.console import Console
 from rich.table import Table
 from ..utils.config import ConfigManager
 from ..utils.output import (
+    extract_credits,
     extract_request_id,
     get_evidence_links,
+    print_credits_footer,
     print_request_id_footer,
     render_evidence_lines_text,
     render_llm_output,
@@ -92,6 +94,7 @@ def grep(
                 "latency_ms": getattr(result, "latency_ms", None),
                 "query_id": None,
                 "request_id": extract_request_id(result),
+                "credits": extract_credits(result),
             }
             print(render_llm_output(hits, query, metadata))
             return
@@ -132,6 +135,7 @@ def grep(
                 console.print()
 
         print_request_id_footer(console, extract_request_id(result))
+        print_credits_footer(console, result)
 
     except Exception as e:
         console.print(f"[red]Erro:[/red] {e}")

{vectorgov_cli-0.3.2 → vectorgov_cli-0.3.4}/src/vectorgov/cli/commands/hybrid.py

@@ -6,8 +6,10 @@ from rich.console import Console
 from rich.table import Table
 from ..utils.config import ConfigManager
 from ..utils.output import (
+    extract_credits,
     extract_request_id,
     get_evidence_links,
+    print_credits_footer,
     print_request_id_footer,
     render_evidence_lines_text,
     render_llm_output,
@@ -27,6 +29,10 @@ def hybrid(
         help="Tipo de expansão: bidirectional ou forward",
     ),
     token_budget: int = typer.Option(None, "--token-budget", help="Limite de tokens para contexto"),
+    show_stats: bool = typer.Option(
+        False, "--show-stats",
+        help="Mostra estatísticas de expansão (seeds, graph_nodes, tokens) no rodapé"
+    ),
     output: str = typer.Option("table", "--output", "-o", help="Formato: table, json, text, llm"),
     raw: bool = typer.Option(False, "--raw", help="JSON bruto para piping"),
 ):
@@ -38,6 +44,7 @@ def hybrid(
     Exemplo:
         vectorgov hybrid "Critérios de julgamento em licitações"
         vectorgov hybrid "Dispensa de licitação" --hops 2 --top-k 15
+        vectorgov hybrid "ETP" --show-stats             # inclui estatisticas
     """
     try:
         from vectorgov import VectorGov
@@ -132,6 +139,7 @@ def hybrid(
                 "latency_ms": getattr(result, "latency_ms", None),
                 "query_id": getattr(result, "query_id", None),
                 "request_id": extract_request_id(result),
+                "credits": extract_credits(result),
             }
             print(render_llm_output(hits, query, metadata))
             return
@@ -176,10 +184,49 @@ def hybrid(
                 for line in render_evidence_lines_text(h):
                     console.print(f"[dim]{line}[/dim]")
 
+        # Estatisticas de expansao (--show-stats): seeds, graph nodes e
+        # tokens do contexto LLM. Util para entender o quanto o grafo
+        # contribuiu vs busca direta, sem precisar de --raw.
+        if show_stats:
+            stats_obj = getattr(result, "stats", None)
+            seeds = None
+            graph_nodes_count = None
+            total_tokens = None
+            truncated = None
+            if stats_obj is not None:
+                # stats pode ser dict (SDK antigo) ou objeto com atributos
+                if isinstance(stats_obj, dict):
+                    seeds = stats_obj.get("seeds") or stats_obj.get("seeds_count")
+                    graph_nodes_count = stats_obj.get("graph_nodes") or stats_obj.get("nodes")
+                    total_tokens = stats_obj.get("tokens") or stats_obj.get("total_tokens")
+                    truncated = stats_obj.get("truncated")
+                else:
+                    seeds = getattr(stats_obj, "seeds", None) or getattr(stats_obj, "seeds_count", None)
+                    graph_nodes_count = getattr(stats_obj, "graph_nodes", None) or getattr(stats_obj, "nodes", None)
+                    total_tokens = getattr(stats_obj, "tokens", None) or getattr(stats_obj, "total_tokens", None)
+                    truncated = getattr(stats_obj, "truncated", None)
+            # Fallback: contar diretamente dos hits se stats nao veio
+            if seeds is None:
+                seeds = sum(1 for h in direct_hits)
+            if graph_nodes_count is None:
+                graph_nodes_count = sum(1 for h in graph_hits)
+
+            console.print()
+            stats_line = (
+                f"[dim]Stats: seeds={seeds} | graph_nodes={graph_nodes_count}"
+            )
+            if total_tokens is not None:
+                stats_line += f" | tokens={total_tokens}"
+            if truncated is not None:
+                stats_line += f" | truncated={'sim' if truncated else 'nao'}"
+            stats_line += "[/dim]"
+            console.print(stats_line)
+
         query_id = getattr(result, "query_id", None)
         if query_id:
             console.print(f"\n[dim]query_id: {query_id}[/dim]")
         print_request_id_footer(console, extract_request_id(result))
+        print_credits_footer(console, result)
 
     except Exception as e:
         console.print(f"[red]Erro:[/red] {e}")

{vectorgov_cli-0.3.2 → vectorgov_cli-0.3.4}/src/vectorgov/cli/commands/lookup.py

@@ -7,7 +7,9 @@ from rich.panel import Panel
 from ..utils.config import ConfigManager
 from ..utils.output import (
     absolute_url,
+    extract_credits,
     extract_request_id,
+    print_credits_footer,
     print_request_id_footer,
     render_evidence_lines_text,
     render_llm_output,
@@ -425,6 +427,7 @@ def lookup(  # noqa: C901
                 "latency_ms": getattr(result, "latency_ms", None),
                 "query_id": getattr(result, "query_id", None),
                 "request_id": extract_request_id(result, raw_resp),
+                "credits": extract_credits(result),
                 "nota_especialista": nota_especialista,
                 "jurisprudencia_tcu": jurisprudencia_tcu,
                 "acordao_tcu_key": acordao_tcu_key,

{vectorgov_cli-0.3.2 → vectorgov_cli-0.3.4}/src/vectorgov/cli/commands/merged.py

@@ -6,8 +6,10 @@ from rich.console import Console
 from rich.table import Table
 from ..utils.config import ConfigManager
 from ..utils.output import (
+    extract_credits,
     extract_request_id,
     get_evidence_links,
+    print_credits_footer,
     print_request_id_footer,
     render_evidence_lines_text,
     render_llm_output,
@@ -132,6 +134,7 @@ def merged(
                 "latency_ms": getattr(result, "latency_ms", None),
                 "query_id": None,
                 "request_id": extract_request_id(result),
+                "credits": extract_credits(result),
             }
             print(render_llm_output(hits, query, metadata))
             return
@@ -167,6 +170,7 @@ def merged(
                     console.print(f"[dim]{line}[/dim]")
 
         print_request_id_footer(console, extract_request_id(result))
+        print_credits_footer(console, result)
 
     except typer.Exit:
         # typer.Exit(0) usado para "nenhum resultado" nao deve ser tratado

{vectorgov_cli-0.3.2 → vectorgov_cli-0.3.4}/src/vectorgov/cli/commands/smart_search.py

@@ -7,8 +7,10 @@ from rich.table import Table
 from rich.panel import Panel
 from ..utils.config import ConfigManager
 from ..utils.output import (
+    extract_credits,
     extract_request_id,
     get_evidence_links,
+    print_credits_footer,
     print_request_id_footer,
     render_evidence_lines_text,
     render_llm_output,
@@ -97,6 +99,7 @@ def smart_search(
                 "latency_ms": getattr(result, "latency_ms", None),
                 "query_id": getattr(result, "query_id", None),
                 "request_id": extract_request_id(result),
+                "credits": extract_credits(result),
             }
             print(render_llm_output(hits, query, metadata))
             return
@@ -148,6 +151,7 @@ def smart_search(
                 f"(use com 'vectorgov feedback send')[/dim]"
             )
         print_request_id_footer(console, extract_request_id(result))
+        print_credits_footer(console, result)
 
     except Exception as e:
         console.print(f"[red]Erro:[/red] {e}")

{vectorgov_cli-0.3.2 → vectorgov_cli-0.3.4}/src/vectorgov/cli/utils/output.py

@@ -175,6 +175,56 @@ def extract_request_id(result: Any, raw_resp: Optional[Dict[str, Any]] = None) -
     return None
 
 
+def extract_credits(result: Any) -> Optional[Dict[str, Any]]:
+    """Extrai CreditsInfo (cost, balance, endpoint) do result do SDK.
+
+    Requer vectorgov SDK >= 0.19.1 (expoe result.credits). Retorna None
+    se o SDK nao populou (versao antiga ou endpoint sem custo).
+
+    Returns:
+        Dict com keys 'cost' (int), 'balance' (int), 'endpoint' (str)
+        ou None se ausente.
+    """
+    credits = getattr(result, "credits", None)
+    if credits is None:
+        return None
+    return {
+        "cost": getattr(credits, "cost", 0),
+        "balance": getattr(credits, "balance", 0),
+        "endpoint": getattr(credits, "endpoint", ""),
+    }
+
+
+def format_credits_line(result: Any, prefix: str = "") -> Optional[str]:
+    """Formata linha de creditos para o rodape text/llm.
+
+    Retorna None se o SDK nao expos credits (nao renderiza nada).
+
+    Exemplo de saida:
+        'Creditos: 1 (saldo: 68)'
+    """
+    credits = extract_credits(result)
+    if not credits:
+        return None
+    cost = credits.get("cost", 0)
+    balance = credits.get("balance", 0)
+    return f"{prefix}Créditos: {cost} (saldo: {balance})"
+
+
+def print_credits_footer(console: Console, result: Any) -> None:
+    """Imprime rodape formatado 'Creditos: X (saldo: Y)' se disponivel.
+
+    Use no final dos handlers de output text dos comandos de busca,
+    para exibir custo da chamada e saldo restante. Dim style para
+    nao competir com o conteudo principal.
+
+    Silencioso se SDK < 0.19.1 (result.credits e None).
+    """
+    line = format_credits_line(result)
+    if line:
+        console.print(f"[dim]{line}[/dim]")
+
+
 def render_evidence_cell_table(hit: Any) -> str:
     """Célula curta para coluna de tabela. Usa Rich link markup clicável."""
     links = get_evidence_links(hit)
@@ -271,6 +321,11 @@ def render_llm_output(hits: list, query: str, metadata: Optional[dict] = None) -
             parts.append(f"Total: {metadata['total']} hits")
         if metadata.get("latency_ms") is not None:
             parts.append(f"Latencia: {metadata['latency_ms']/1000:.1f}s")
+        credits_info = metadata.get("credits")
+        if credits_info:
+            cost = credits_info.get("cost", 0)
+            balance = credits_info.get("balance", 0)
+            parts.append(f"Creditos: {cost} (saldo: {balance})")
         if metadata.get("query_id"):
             parts.append(f"Query ID: {metadata['query_id']}")
         if metadata.get("request_id"):
@@ -345,6 +400,7 @@ def format_search_results(console: Console, results: Any, output_format: OutputF
         query: Query original
     """
     request_id = extract_request_id(results)
+    credits = extract_credits(results)
 
     if output_format == OutputFormat.json:
         data = {
@@ -354,6 +410,7 @@ def format_search_results(console: Console, results: Any, output_format: OutputF
             "latency_ms": results.latency_ms,
             "query_id": results.query_id,
             "request_id": request_id,
+            "credits": credits,
             "hits": [hit_to_raw_dict(h) for h in results.hits],
         }
         console.print_json(json.dumps(data, ensure_ascii=False))
@@ -407,6 +464,7 @@ def format_search_results(console: Console, results: Any, output_format: OutputF
         console.print(f"[dim]Query ID: {results.query_id}[/dim]")
         if request_id:
             console.print(f"[dim]Request ID: {request_id}[/dim]")
+        print_credits_footer(console, results)
         console.print("[dim]Use 'vectorgov feedback send <query_id> --like' para avaliar[/dim]")
 
     elif output_format == OutputFormat.markdown:
@@ -432,6 +490,7 @@ def format_search_results(console: Console, results: Any, output_format: OutputF
             "latency_ms": results.latency_ms,
             "query_id": results.query_id,
             "request_id": request_id,
+            "credits": credits,
         }
         print(render_llm_output(results.hits, query, metadata))
 
@@ -462,6 +521,7 @@ def format_search_results(console: Console, results: Any, output_format: OutputF
         console.print(f"[dim]Query ID: {results.query_id}[/dim]")
         if request_id:
             console.print(f"[dim]Request ID: {request_id}[/dim]")
+        print_credits_footer(console, results)
 
 
 def format_code_block(console: Console, code: str, language: str = "python"):