Skip to content

Commit

Permalink
Add support for --verbose flag.
Browse files Browse the repository at this point in the history
  • Loading branch information
drazvan committed Dec 20, 2023
1 parent e12c2b8 commit f19a85f
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 0 deletions.
2 changes: 2 additions & 0 deletions nemoguardrails/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,3 +18,5 @@
from .rails import LLMRails, RailsConfig

patch_asyncio.apply()

__version__ = "0.6.1"
16 changes: 16 additions & 0 deletions nemoguardrails/cli/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@
import uvicorn
from fastapi import FastAPI

from nemoguardrails import __version__
from nemoguardrails.actions_server import actions_server
from nemoguardrails.cli.chat import run_chat
from nemoguardrails.eval.cli import evaluate
Expand Down Expand Up @@ -139,3 +140,18 @@ def action_server(
"""Start a NeMo Guardrails actions server."""

uvicorn.run(actions_server.app, port=port, log_level="info", host="0.0.0.0")


def version_callback(value: bool):
if value:
typer.echo(__version__)
raise typer.Exit()


@app.callback()
def cli(
_: Optional[bool] = typer.Option(
None, "-v", "--version", callback=version_callback, is_eager=True
)
):
pass

0 comments on commit f19a85f

Please sign in to comment.