Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enhanced server startup logs (API URL, Docs URL, ect) #61

Merged
merged 1 commit into from
Jul 26, 2024
Merged

Enhanced server startup logs (API URL, Docs URL, ect) #61

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 20 additions & 2 deletions guardrails_api/app.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,8 @@
from guardrails_api.clients.postgres_client import postgres_is_enabled
from guardrails_api.otel import otel_is_disabled, initialize
from guardrails_api.clients.cache_client import CacheClient

from rich.console import Console
from rich.rule import Rule

# TODO: Move this to a separate file
class OverrideJsonProvider(DefaultJSONProvider):
Expand Down Expand Up @@ -50,6 +51,9 @@ def register_config(config: Optional[str] = None):
def create_app(
env: Optional[str] = None, config: Optional[str] = None, port: Optional[int] = None
):
# used to print user-facing messages during server startup
console = Console()

if os.environ.get("APP_ENVIRONMENT") != "production":
from dotenv import load_dotenv

Expand Down Expand Up @@ -96,9 +100,23 @@ def create_app(
cache_client.initialize(app)

from guardrails_api.blueprints.root import root_bp
from guardrails_api.blueprints.guards import guards_bp
from guardrails_api.blueprints.guards import guards_bp, guard_client

app.register_blueprint(root_bp)
app.register_blueprint(guards_bp)

console.print(
f"\n:rocket: Guardrails API is available at {self_endpoint}"
)
console.print(f":book: Visit {self_endpoint}/docs to see available API endpoints.\n")

console.print(":green_circle: Active guards and OpenAI compatible endpoints:")

for g in guard_client.get_guards():
g = g.to_dict()
console.print(f"- Guard: [bold white]{g.get('name')}[/bold white] {self_endpoint}/guards/{g.get('name')}/openai/v1")

console.print("")
console.print(Rule("[bold grey]Server Logs[/bold grey]", characters="=", style="white"))

return app