Merge pull request #38 from tisnik/better-logging

Better logging

Author: tisnik

.github/dependabot.yml

@@ -0,0 +1,11 @@
+# To get started with Dependabot version updates, you'll need to specify which
+# package ecosystems to update and where the package manifests are located.
+# Please see the documentation for all configuration options:
+# https://docs.github.com/code-security/dependabot/dependabot-version-updates/configuration-options-for-the-dependabot.yml-file
+
+version: 2
+updates:
+  - package-ecosystem: "" # See documentation for possible values
+    directory: "/" # Location of package manifests
+    schedule:
+      interval: "daily"

src/app/endpoints/query.py

@@ -51,9 +51,9 @@ def retrieve_response(client: LlamaStackClient, model_id: str, prompt: str) -> s
 
     available_shields = [shield.identifier for shield in client.shields.list()]
     if not available_shields:
-        print(colored("No available shields. Disabling safety.", "yellow"))
+        logger.info("No available shields. Disabling safety")
     else:
-        print(f"Available shields found: {available_shields}")
+        logger.info(f"Available shields found: {available_shields}")
 
     agent = Agent(
         client,