beamlit 0.0.21__py3-none-any.whl → 0.0.22rc11__py3-none-any.whl

beamlit/authentication/credentials.py

@@ -43,6 +43,28 @@ class Config:
         if self.context is None:
             self.context = ContextConfig()
 
+    def to_json(self) -> dict:
+        return {
+            "workspaces": [
+                {
+                    "name": ws.name,
+                    "credentials": {
+                        "api_key": ws.credentials.api_key,
+                        "access_token": ws.credentials.access_token,
+                        "refresh_token": ws.credentials.refresh_token,
+                        "expires_in": ws.credentials.expires_in,
+                        "device_code": ws.credentials.device_code,
+                        "client_credentials": ws.credentials.client_credentials,
+                    },
+                }
+                for ws in self.workspaces
+            ],
+            "context": {
+                "workspace": self.context.workspace,
+                "environment": self.context.environment,
+            },
+        }
+
 
 def load_config() -> Config:
     config = Config()
@@ -68,23 +90,6 @@ def load_config() -> Config:
 
 
 def save_config(config: Config):
-    data = {
-        "workspaces": [
-            {
-                "name": ws.name,
-                "credentials": {
-                    "access_token": ws.credentials.access_token,
-                    "api_key": ws.credentials.api_key,
-                },
-            }
-            for ws in config.workspaces
-        ],
-        "context": {
-            "workspace": config.context.workspace,
-            "environment": config.context.environment,
-        },
-    }
-
     home_dir = Path.home()
     if not home_dir:
         raise RuntimeError("Could not determine home directory")
@@ -93,9 +98,8 @@ def save_config(config: Config):
     config_file = config_dir / "config.yaml"
 
     config_dir.mkdir(mode=0o700, parents=True, exist_ok=True)
-
     with open(config_file, "w", encoding="utf-8") as f:
-        yaml.dump(data, f)
+        yaml.dump(config.to_json(), f)
 
 
 def list_workspaces() -> List[str]:

beamlit/common/settings.py

@@ -59,7 +59,7 @@ class Settings(BaseSettings):
     )
 
     workspace: str
-    environment: str
+    environment: str = Field(default="production")
     type: str = Field(default="agent")
     name: str = Field(default="beamlit-agent")
     base_url: str = Field(default="https://api.beamlit.dev/v0")

{beamlit-0.0.21.dist-info → beamlit-0.0.22rc11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: beamlit
-Version: 0.0.21
+Version: 0.0.22rc11
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12

{beamlit-0.0.21.dist-info → beamlit-0.0.22rc11.dist-info}/RECORD

@@ -132,13 +132,13 @@ beamlit/authentication/__init__.py,sha256=wiXqRbc7E-ulrH_ueA9duOGFvXeo7-RvhSD1Xb
 beamlit/authentication/apikey.py,sha256=jnz1FMRauI5qAInqeeDER8aCONx4O8ZPZGedvi3Ap_o,659
 beamlit/authentication/authentication.py,sha256=om26AteY2cCV9ctqbOCynX6PgS8YO-aCreNOFSnnWKc,3121
 beamlit/authentication/clientcredentials.py,sha256=6kbfTjwUkXUArJX8XZLe9ZzbEicQc19tSXBvsTpiXMk,3954
-beamlit/authentication/credentials.py,sha256=_Bjj49jGeo-JTvO2GPS1yXxh5vS1NmX0haxTfg0KiEk,4965
+beamlit/authentication/credentials.py,sha256=DBw598T6it7EgXKnS_qfsBM0mkC33iRDadWA70bg5X0,5349
 beamlit/authentication/device_mode.py,sha256=oQVBCDsq-pdeXF31WSTAAEdaX6eACV7SYcOSyf3ea_Q,3728
 beamlit/common/__init__.py,sha256=yDoMJDKj-xjTGl7U1YI59KpWxiOV65HSiUulgO8xdTA,277
 beamlit/common/generate.py,sha256=VJ_MiRDulXdQdnlKdM4_Bod6CO6DOGlFiosGXOLuLGs,7227
 beamlit/common/logger.py,sha256=ayabnsoHS8ncXm8EpBS01FkvSe0XRcaNdQjKVuPI5z4,1025
 beamlit/common/secrets.py,sha256=sid81bOe3LflkMKDHwBsBs9nIju8bp5-v9qU9gkyNMc,212
-beamlit/common/settings.py,sha256=Ylfw1L4a3TdYlUXwoYLYbS7qZT1zi2Hm-uLzhSZ1jcI,5222
+beamlit/common/settings.py,sha256=7PDfLPzTPN5Yeh2NnmAlSzpBi_u-qza2lUIX-i6N_lg,5252
 beamlit/common/utils.py,sha256=jouz5igBvT37Xn_e94-foCHyQczVim-UzVcoIF6RWJ4,657
 beamlit/functions/__init__.py,sha256=_RPG1Bfg54JGdIPnViAU6n9zD7E1cDNsdXi8oYGskzE,138
 beamlit/functions/decorator.py,sha256=-2newMBztweIgFuh0ABKOdxCfUzWaRxf0ym-YAgggJI,3168
@@ -296,6 +296,6 @@ beamlit/serve/app.py,sha256=-CvUU7pqFDigndHpA01IR4_mxXaT8c0VsnVjMfIw6K8,2399
 beamlit/serve/middlewares/__init__.py,sha256=1dVmnOmhAQWvWktqHkKSIX-YoF6fmMU8xkUQuhg_rJU,148
 beamlit/serve/middlewares/accesslog.py,sha256=wM52-hcwtO-_hdM1pnsEJzerzJf1MzEyN5m85BdDccE,609
 beamlit/serve/middlewares/processtime.py,sha256=lDAaIasZ4bwvN-HKHvZpaD9r-yrkVNZYx4abvbjbrCg,411
-beamlit-0.0.21.dist-info/METADATA,sha256=EsyU4e1T5fySulg09x5vy9XZ9BNPVfrYZX8dHY5l-b0,2023
-beamlit-0.0.21.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-beamlit-0.0.21.dist-info/RECORD,,
+beamlit-0.0.22rc11.dist-info/METADATA,sha256=AOUr4a-4i_HzlZwKbn12EyaSvGR2BJFEFdzs4WXPqRw,2027
+beamlit-0.0.22rc11.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+beamlit-0.0.22rc11.dist-info/RECORD,,