proscenium 0.0.6__tar.gz → 0.0.7__tar.gz

{proscenium-0.0.6 → proscenium-0.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: proscenium
-Version: 0.0.6
+Version: 0.0.7
 Summary: Frame AI Agents
 License: ASFv2
 Author: Adam Pingel

proscenium-0.0.7/proscenium/util/__init__.py

@@ -0,0 +1,29 @@
+import logging
+import os
+
+logging.getLogger(__name__).addHandler(logging.NullHandler())
+
+log = logging.getLogger(__name__)
+
+
+def get_secret(key: str, default: str = None) -> str:
+    try:
+        from google.colab import userdata
+
+        try:
+            value = userdata.get(key)
+            print(
+                f"Using {key} from colab userdata and setting corresponding os.environ value"
+            )
+            os.environ[key] = value
+            return value
+        except userdata.SecretNotFoundError:
+            print(f"Using default value for {key}")
+            return default
+    except ImportError:
+        if key in os.environ:
+            print(f"Not in colab. Using {key} from environment")
+            return os.environ.get(key, default)
+        else:
+            print(f"Using default value for {key}")
+            return default

{proscenium-0.0.6 → proscenium-0.0.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "proscenium"
-version = "0.0.6"
+version = "0.0.7"
 description = "Frame AI Agents"
 authors = ["Adam Pingel <oss@pingel.org>"]
 license = "ASFv2"