Merge pull request #46 from SubstrateLabs/chris/openai-and-anthropic-api-keys

Secrets field for OpenAI and Anthropic keys

Author: csreesan

substrate/GEN_VERSION

@@ -1 +1 @@
-20240617.20240727
+20240617.20240802

substrate/__init__.py

@@ -1,7 +1,7 @@
 """
 𐃏 Substrate Python SDK
 
-20240617.20240727
+20240617.20240802
 """
 
 from .nodes import (
@@ -51,11 +51,12 @@
 )
 from .core.sb import sb
 from ._version import __version__
-from .substrate import Substrate, SubstrateResponse
+from .substrate import Secrets, Substrate, SubstrateResponse
 from .run_python import RunPython
 
 __all__ = [
     "__version__",
+    "Secrets",
     "SubstrateResponse",
     "sb",
     "Substrate",

substrate/core/sb.py

@@ -1,7 +1,8 @@
 """
 CORE ꩜ SUBSTRATE
 """
-from typing import Any, Union
+from re import template
+from typing import Any, Dict, Union
 
 from .client.future import Future
 from .future_directive import (

substrate/nodes.py

@@ -6,6 +6,7 @@
 
 import warnings
 
+from .substrate import SubstrateResponse
 from .core.corenode import CoreNode
 
 # filter pydantic v2 deprecation warnings
@@ -17,6 +18,7 @@
         CLIPOut,
         JinaV2Out,
         EmbedTextOut,
+        RunPythonOut,
         EmbedImageOut,
         EraseImageOut,
         ComputeJSONOut,
@@ -57,6 +59,7 @@
         StableDiffusionXLControlNetOut,
     )
 from typing import Any, Dict, List, Optional
+from dataclasses import dataclass
 from typing_extensions import Literal
 
 from .future_dataclass_models import (
@@ -65,6 +68,7 @@
     FutureCLIPOut,
     FutureJinaV2Out,
     FutureEmbedTextOut,
+    FutureRunPythonOut,
     FutureEmbedImageOut,
     FutureEraseImageOut,
     FutureComputeJSONOut,

substrate/substrate.py

@@ -1,6 +1,7 @@
 import json
 import zlib
 import base64
+import dataclasses
 from typing import Any, Dict, Optional
 
 from substrate.streaming import SubstrateStreamingResponse
@@ -11,6 +12,12 @@
 from .substrate_response import SubstrateResponse
 
 
+@dataclasses.dataclass
+class Secrets:
+    openai: Optional[str] = None
+    anthropic: Optional[str] = None
+
+
 class Substrate:
     """
     Substrate client.
@@ -21,6 +28,7 @@ def __init__(
         api_key: str,
         base_url: str = "https://api.substrate.run",
         timeout: float = 60 * 5.0,
+        secrets: Optional[Secrets] = None,
         additional_headers: Optional[Dict[str, Any]] = None,
     ):
         """
@@ -29,6 +37,11 @@ def __init__(
         if additional_headers is None:
             additional_headers = {}
         self.api_key = api_key
+        if secrets is not None:
+            if secrets.openai is not None:
+                additional_headers["x-substrate-openai-api-key"] = secrets.openai
+            if secrets.anthropic is not None:
+                additional_headers["x-substrate-anthropic-api-key"] = secrets.anthropic
         self._client = APIClient(
             api_key=api_key,
             base_url=base_url,