ifs-rag-assistant-cli 2.2.2__tar.gz → 2.3.0__tar.gz

{ifs_rag_assistant_cli-2.2.2 → ifs_rag_assistant_cli-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ifs-rag-assistant-cli
-Version: 2.2.2
+Version: 2.3.0
 Summary: Command-line interface for IFS RAG Assistant Q&A system
 Home-page: https://github.com/maricaantonacci/ifs-rag-assistant-cli
 Author: Marica Antonacci
@@ -122,6 +122,8 @@ If you're using the ECMWF production environment, setup is simple:
 
 That's it! The configuration is already set up for ECMWF's production environment.
 
+> **Network access disclaimer:** The CLI authentication flow is available only from approved ECMWF networks (CCI Cloud and VDI). From other networks, authentication will fail. For specific business needs, access exceptions can be evaluated with the ECMWF IAM team.
+
 ### For Custom Environments
 
 1. **Initialize and customize configuration**:
@@ -347,6 +349,8 @@ Logout and clear cached session:
 ifs-rag-assistant-cli logout
 ```
 
+> **Network access disclaimer:** Authentication is available only from approved ECMWF networks (CCI Cloud and VDI). From other networks, authentication will fail. For specific business needs, access exceptions can be evaluated with the ECMWF IAM team.
+
 ### Initialize Configuration
 
 Create a configuration file pre-configured for ECMWF production:

{ifs_rag_assistant_cli-2.2.2 → ifs_rag_assistant_cli-2.3.0}/README.md

@@ -93,6 +93,8 @@ If you're using the ECMWF production environment, setup is simple:
 
 That's it! The configuration is already set up for ECMWF's production environment.
 
+> **Network access disclaimer:** The CLI authentication flow is available only from approved ECMWF networks (CCI Cloud and VDI). From other networks, authentication will fail. For specific business needs, access exceptions can be evaluated with the ECMWF IAM team.
+
 ### For Custom Environments
 
 1. **Initialize and customize configuration**:
@@ -318,6 +320,8 @@ Logout and clear cached session:
 ifs-rag-assistant-cli logout
 ```
 
+> **Network access disclaimer:** Authentication is available only from approved ECMWF networks (CCI Cloud and VDI). From other networks, authentication will fail. For specific business needs, access exceptions can be evaluated with the ECMWF IAM team.
+
 ### Initialize Configuration
 
 Create a configuration file pre-configured for ECMWF production:

{ifs_rag_assistant_cli-2.2.2 → ifs_rag_assistant_cli-2.3.0}/ifs_rag_assistant_cli/__init__.py

@@ -2,4 +2,4 @@
 IFS RAG Assistant CLI - Command-line interface for IFS RAG Assistant Q&A system.
 """
 
-__version__ = "2.2.2"
+__version__ = "2.3.0"

{ifs_rag_assistant_cli-2.2.2 → ifs_rag_assistant_cli-2.3.0}/ifs_rag_assistant_cli/backend_client.py

@@ -173,6 +173,30 @@ class BackendClient:
 
         return cast(dict[str, Any], resp.json())
 
+    def get_chunk(self, chunk_id: str) -> dict[str, Any]:
+        """
+        Retrieve a single chunk by its identifier.
+
+        Args:
+            chunk_id: Chunk identifier from a citation
+
+        Returns:
+            Response dictionary with chunk metadata and text
+
+        Raises:
+            RuntimeError: If request fails
+        """
+        try:
+            resp = self.session.get(
+                f"{self.backend_url}/chunk/{chunk_id}",
+                timeout=self.timeout
+            )
+            resp.raise_for_status()
+        except RequestException as e:
+            self._raise_request_error("Get chunk request", e)
+
+        return cast(dict[str, Any], resp.json())
+
     def logout(self) -> None:
         """
         Logout from backend (invalidate session).

{ifs_rag_assistant_cli-2.2.2 → ifs_rag_assistant_cli-2.3.0}/ifs_rag_assistant_cli/cli.py

@@ -14,6 +14,7 @@ import typer
 from click.exceptions import Abort
 from rich.console import Console
 from rich.panel import Panel
+from rich.syntax import Syntax
 from rich.table import Table
 
 from ifs_rag_assistant_cli import __version__
@@ -137,6 +138,103 @@ def format_citations(citations_data: list, plain: bool = False) -> None:
         console.print(table)
 
 
+def guess_chunk_language(path: str, kind: str = "") -> str:
+    """Guess syntax lexer from file path and symbol kind."""
+    path_lower = str(path).lower()
+    if path_lower.endswith((".f90", ".f", ".for", ".f77", ".f95", ".f03", ".f08")):
+        return "fortran"
+    if path_lower.endswith((".md", ".markdown")):
+        return "markdown"
+    if path_lower.endswith((".py",)):
+        return "python"
+    if path_lower.endswith((".yml", ".yaml")):
+        return "yaml"
+    if "markdown" in str(kind).lower():
+        return "markdown"
+    return "text"
+
+
+def display_chunk(chunk_data: dict, plain: bool = False) -> None:
+    """Render chunk content and metadata in terminal."""
+    path = str(chunk_data.get("path", ""))
+    start = int(chunk_data.get("start", 1) or 1)
+    end = int(chunk_data.get("end", start) or start)
+    kind = str(chunk_data.get("kind", ""))
+    sym_name = str(chunk_data.get("sym_name", ""))
+    text = str(chunk_data.get("text", ""))
+
+    if plain:
+        console.print(f"\nChunk: {path}:{start}-{end} ({kind or 'n/a'})")
+        if sym_name:
+            console.print(f"Symbol: {sym_name}")
+        console.print(text if text else "<empty chunk>")
+        return
+
+    title = f"Chunk: {path}:{start}-{end}"
+    subtitle = f"{kind or 'n/a'}"
+    if sym_name:
+        subtitle = f"{subtitle} - {sym_name}"
+
+    syntax = Syntax(
+        text if text else "<empty chunk>",
+        guess_chunk_language(path, kind),
+        line_numbers=True,
+        start_line=start,
+        word_wrap=False,
+    )
+    console.print(Panel(syntax, title=title, subtitle=subtitle, border_style="cyan"))
+
+
+def inspect_citations_interactively(client, citations_data: list, plain: bool = False) -> None:
+    """Allow users to fetch and inspect full chunk content from citations."""
+    indexed_citations: list[tuple[int, Citation]] = []
+    for i, cite_dict in enumerate(citations_data, 1):
+        citation = Citation.from_dict(cite_dict)
+        indexed_citations.append((i, citation))
+
+    inspectable = [(i, c) for i, c in indexed_citations if c.id]
+    if not inspectable:
+        return
+
+    try:
+        if not typer.confirm("\n🔎 Inspect a citation chunk?", default=False):
+            return
+
+        valid_numbers = {i for i, _ in inspectable}
+        choices_text = ", ".join(str(i) for i, _ in inspectable)
+
+        while True:
+            selected = typer.prompt(
+                f"Select citation number ({choices_text}) or 0 to cancel",
+                type=int,
+                default=0,
+            )
+
+            if selected == 0:
+                return
+
+            if selected not in valid_numbers:
+                console.print("[yellow]Invalid citation number[/yellow]")
+                continue
+
+            citation = next(c for i, c in inspectable if i == selected)
+            try:
+                chunk_data = client.backend.get_chunk(str(citation.id))
+                display_chunk(chunk_data, plain=plain)
+            except SessionExpiredError as e:
+                console.print(f"[bold red]❌ Error:[/bold red] {e}")
+                return
+            except RuntimeError as e:
+                console.print(f"[yellow]⚠️  Failed to fetch chunk: {e}[/yellow]")
+                return
+
+            if not typer.confirm("Inspect another citation?", default=False):
+                return
+    except (Abort, KeyboardInterrupt):
+        console.print("\n[dim]Citation inspection cancelled[/dim]")
+        return
+
+
 def show_auth_mode_notice(config) -> None:
     """Show an explicit warning when authentication is disabled."""
     auth_mode = str(getattr(config, "auth_mode", "oauth")).strip().lower()
@@ -230,6 +328,7 @@ def chat(
                     # Show citations if available
                     citations_data = result.get("citations", [])
                     format_citations(citations_data, plain)
+                    inspect_citations_interactively(client, citations_data, plain=plain)
 
                     # Ask for optional feedback
                     if "conversation_id" in result and "answer_id" in result:
@@ -307,6 +406,7 @@ def ask(
                 if citations_data:
                     console.print("\n")
                 format_citations(citations_data, plain)
+                inspect_citations_interactively(client, citations_data, plain=plain)
 
                 # Ask for optional feedback
                 if "conversation_id" in result and "answer_id" in result:
@@ -340,7 +440,7 @@ def search(
         None,
         "--config",
         help="Path to config file"
-    )
+    ),
 ) -> None:
     """
     Search the RAG index.
@@ -364,6 +464,7 @@ def search(
             citations_data = result.get("citations", [])
             if citations_data:
                 format_citations(citations_data, plain=False)
+                inspect_citations_interactively(client, citations_data, plain=False)
             else:
                 console.print("[yellow]No results found[/yellow]")
 

{ifs_rag_assistant_cli-2.2.2 → ifs_rag_assistant_cli-2.3.0}/ifs_rag_assistant_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ifs-rag-assistant-cli
-Version: 2.2.2
+Version: 2.3.0
 Summary: Command-line interface for IFS RAG Assistant Q&A system
 Home-page: https://github.com/maricaantonacci/ifs-rag-assistant-cli
 Author: Marica Antonacci
@@ -122,6 +122,8 @@ If you're using the ECMWF production environment, setup is simple:
 
 That's it! The configuration is already set up for ECMWF's production environment.
 
+> **Network access disclaimer:** The CLI authentication flow is available only from approved ECMWF networks (CCI Cloud and VDI). From other networks, authentication will fail. For specific business needs, access exceptions can be evaluated with the ECMWF IAM team.
+
 ### For Custom Environments
 
 1. **Initialize and customize configuration**:
@@ -347,6 +349,8 @@ Logout and clear cached session:
 ifs-rag-assistant-cli logout
 ```
 
+> **Network access disclaimer:** Authentication is available only from approved ECMWF networks (CCI Cloud and VDI). From other networks, authentication will fail. For specific business needs, access exceptions can be evaluated with the ECMWF IAM team.
+
 ### Initialize Configuration
 
 Create a configuration file pre-configured for ECMWF production:

{ifs_rag_assistant_cli-2.2.2 → ifs_rag_assistant_cli-2.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "ifs-rag-assistant-cli"
-version = "2.2.2"
+version = "2.3.0"
 description = "Command-line interface for IFS RAG Assistant Q&A system"
 readme = "README.md"
 requires-python = ">=3.9"