nebu 0.1.70__py3-none-any.whl → 0.1.72__py3-none-any.whl

nebu/orign.py

@@ -0,0 +1,70 @@
+import os
+from dataclasses import dataclass
+from typing import Optional
+
+import yaml
+
+
+@dataclass
+class OrignServerConfig:
+    """
+    Python equivalent of the Rust ServerConfig struct.
+    """
+
+    name: Optional[str] = None
+    api_key: Optional[str] = None
+    server: Optional[str] = None
+    auth_server: Optional[str] = None
+
+
+def get_orign_server() -> Optional[str]:
+    """
+    Get the Orign server URL from environment variables.
+    """
+    env = os.environ.get("ORIGN_SERVER") or os.environ.get("ORIGIN_SERVER")
+    if env:
+        return env
+
+    path = _get_config_file_path()
+    path_exists = os.path.exists(path)
+
+    # Load from disk or create a default
+    if path_exists:
+        try:
+            with open(path, "r") as yaml_file:
+                data = yaml.safe_load(yaml_file) or {}
+            # Convert each server entry into a ServerConfig
+            servers_data = data.get("servers", [])
+            # Ensure data is a list and elements are dicts before comprehension
+            if isinstance(servers_data, list):
+                servers = [
+                    OrignServerConfig(**srv)
+                    for srv in servers_data
+                    if isinstance(srv, dict)
+                ]
+            else:
+                servers = []  # Or handle error appropriately
+            current_server = data.get("current_server")
+
+            for srv in servers:
+                if srv.name == current_server:
+                    return srv.server
+
+            return None
+        except (yaml.YAMLError, TypeError, FileNotFoundError) as e:
+            # Handle potential issues during loading or parsing
+            print(
+                f"Warning: Could not load or parse config file '{path}': {e}. Starting with default config."
+            )
+            return None
+
+    return None
+
+
+def _get_config_file_path() -> str:
+    """
+    Return the path to ~/.agentsea/orign.yaml
+    """
+    home = os.path.expanduser("~")
+    config_dir = os.path.join(home, ".agentsea")
+    return os.path.join(config_dir, "orign.yaml")

nebu/processors/decorate.py

@@ -37,6 +37,7 @@ from nebu.containers.models import (
 )
 from nebu.data import Bucket
 from nebu.meta import V1ResourceMetaRequest
+from nebu.orign import get_orign_server
 from nebu.processors.models import (
     Message,
     V1Scale,
@@ -444,31 +445,38 @@ def processor(
         # --- Get API Key ---
         print("[DEBUG Decorator] Loading Nebu configuration...")
         try:
-            config = GlobalConfig.read()
-            current_server = config.get_current_server_config()
-            if not current_server or not current_server.api_key:
-                raise ValueError("Nebu server configuration or API key not found.")
-            api_key = current_server.api_key
-            print("[DEBUG Decorator] Nebu API key loaded successfully.")
-
-            # Add additional environment variables from current configuration
-            all_env.append(V1EnvVar(key="AGENTSEA_API_KEY", value=api_key))
-
-            # Get server URL and auth server URL from current configuration
-            server_url = (
-                current_server.server
-                if current_server
-                else GlobalConfig.get_server_url()
-            )
-            auth_server_url = current_server.auth_server if current_server else None
-
-            if server_url:
-                all_env.append(V1EnvVar(key="NEBULOUS_SERVER", value=server_url))
-
-            if auth_server_url:
-                all_env.append(
-                    V1EnvVar(key="AGENTSEA_AUTH_SERVER", value=auth_server_url)
-                )
+            pass
+            # config = GlobalConfig.read()
+            # current_server = config.get_current_server_config()
+            # if not current_server or not current_server.api_key:
+            #     raise ValueError("Nebu server configuration or API key not found.")
+            # api_key = current_server.api_key
+            # print("[DEBUG Decorator] Nebu API key loaded successfully.")
+
+            # # Add additional environment variables from current configuration
+            # all_env.append(V1EnvVar(key="AGENTSEA_API_KEY", value=api_key))
+
+            # # Get server URL and auth server URL from current configuration
+            # server_url = (
+            #     current_server.server
+            #     if current_server
+            #     else GlobalConfig.get_server_url()
+            # )
+            # auth_server_url = current_server.auth_server if current_server else None
+
+            # if server_url:
+            #     all_env.append(V1EnvVar(key="NEBULOUS_SERVER", value=server_url))
+
+            # if auth_server_url:
+            #     all_env.append(
+            #         V1EnvVar(key="AGENTSEA_AUTH_SERVER", value=auth_server_url)
+            #     )
+
+            # orign_server = get_orign_server()
+            # if orign_server:
+            #     all_env.append(V1EnvVar(key="ORIGN_SERVER", value=orign_server))
+            # else:
+            #     print("[DEBUG Decorator] No Orign server found. Not setting...")
 
         except Exception as e:
             print(f"ERROR: Failed to load Nebu configuration or API key: {e}")

{nebu-0.1.70.dist-info → nebu-0.1.72.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nebu
-Version: 0.1.70
+Version: 0.1.72
 Summary: A globally distributed container runtime
 Requires-Python: >=3.10.14
 Description-Content-Type: text/markdown

{nebu-0.1.70.dist-info → nebu-0.1.72.dist-info}/RECORD

@@ -4,6 +4,7 @@ nebu/cache.py,sha256=jmluqvWnE9N8uNq6nppXSxEJK7DKWaB79GicaGg9KmY,4718
 nebu/config.py,sha256=C5Jt9Bd0i0HrgzBSVNJ-Ml3KwX_gaYbYYZEtNL2gvJg,7031
 nebu/data.py,sha256=X0aAJYuHNVcTCRHpIDDm546HwMqIZpv40lGrozlL41A,39797
 nebu/meta.py,sha256=CzFHMND9seuewzq9zNNx9WTr6JvrCBExe7BLqDSr7lM,745
+nebu/orign.py,sha256=SkVfHgpadwik58KCZCrjdV5EHY0dhpEhDvijzLxY11Y,2052
 nebu/builders/builder.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nebu/builders/models.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nebu/containers/container.py,sha256=yb7KaPTVXnEEAlrpdlUi4HNqF6P7z9bmwAILGlq6iqU,13502
@@ -13,15 +14,15 @@ nebu/containers/server.py,sha256=yFa2Y9PzBn59E1HftKiv0iapPonli2rbGAiU6r-wwe0,251
 nebu/namespaces/models.py,sha256=EqUOpzhVBhvJw2P92ONDUbIgC31M9jMmcaG5vyOrsWg,497
 nebu/namespaces/namespace.py,sha256=Q_EDH7BgQrTkaDh_l4tbo22qpq-uARfIk8ZPBLjITGY,4967
 nebu/processors/consumer.py,sha256=wFxPwLXCrRM8eD4nd6pQsGW46e06rbkoF5YZihodjVk,33857
-nebu/processors/decorate.py,sha256=1Z0kQSxB1EjzKzC7BnYfiYxx3n66vERE7MiX_Aj33JM,53683
+nebu/processors/decorate.py,sha256=U_ogaebtVrQP9WqpSx32R-Sw4yLVl16Jajg5kIGz90I,54048
 nebu/processors/default.py,sha256=W4slJenG59rvyTlJ7gRp58eFfXcNOTT2Hfi6zzJAobI,365
 nebu/processors/models.py,sha256=y40HoW-MEzDWB2dm_tsYlUy3Nf3s6eiLC0iGO9BoNog,3956
 nebu/processors/processor.py,sha256=PsLs-Oo0bcvqoDKHErpOaic25y8uvTQ8KxtyFwLptW0,16165
 nebu/processors/remote.py,sha256=TeAIPGEMqnDIb7H1iett26IEZrBlcbPB_-DSm6jcH1E,1285
 nebu/redis/models.py,sha256=coPovAcVXnOU1Xh_fpJL4PO3QctgK9nBe5QYoqEcnxg,1230
 nebu/services/service.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nebu-0.1.70.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.70.dist-info/METADATA,sha256=-qnWpIjYFUrm2goKDFdfswoAI8IH_onT5YWALGxR7MI,1731
-nebu-0.1.70.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-nebu-0.1.70.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.70.dist-info/RECORD,,
+nebu-0.1.72.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.72.dist-info/METADATA,sha256=_ny8BqaWyiBInGJNxNMYgX7utIMTFIZL5a6h7MTNI6o,1731
+nebu-0.1.72.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+nebu-0.1.72.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.72.dist-info/RECORD,,