nebu 0.1.68__py3-none-any.whl → 0.1.71__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,
@@ -450,6 +451,32 @@ def processor(
                 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}")
             raise RuntimeError(

{nebu-0.1.68.dist-info → nebu-0.1.71.dist-info}/METADATA

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

{nebu-0.1.68.dist-info → nebu-0.1.71.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=rzasgz_kq6yQosoABawu5yhkBj7i12Ah9ntBIVxEyrk,52928
+nebu/processors/decorate.py,sha256=03K9iC1ivt1lcFWCFdyhYZNAKfCwCskao4ZfolDMdwM,53979
 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.68.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-nebu-0.1.68.dist-info/METADATA,sha256=6f0SgdnkR7YgCe-gWXAzXGZDTM-iKZ3z8tAk-IVpSO8,1731
-nebu-0.1.68.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
-nebu-0.1.68.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
-nebu-0.1.68.dist-info/RECORD,,
+nebu-0.1.71.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+nebu-0.1.71.dist-info/METADATA,sha256=5_UsU30p99wxKSRQ6XCUCtnQkN-zJ8GdUDYp7Um4D-4,1731
+nebu-0.1.71.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+nebu-0.1.71.dist-info/top_level.txt,sha256=uLIbEKJeGSHWOAJN5S0i5XBGwybALlF9bYoB1UhdEgQ,5
+nebu-0.1.71.dist-info/RECORD,,