python multi generate code

substrate/GEN_VERSION

@@ -1 +1 @@
-20240617.20240814
+20240617.20240815

substrate/__init__.py

@@ -1,7 +1,7 @@
 """
 𐃏 Substrate Python SDK
 
-20240617.20240814
+20240617.20240815
 """
 
 from .nodes import (
@@ -41,6 +41,7 @@
     InterpolateFrames,
     Llama3Instruct70B,
     Mistral7BInstruct,
+    MultiGenerateCode,
     MultiInpaintImage,
     SegmentUnderPoint,
     MultiGenerateImage,
@@ -73,6 +74,7 @@
     "ComputeJSON",
     "MultiComputeJSON",
     "GenerateCode",
+    "MultiGenerateCode",
     "Mistral7BInstruct",
     "Mixtral8x7BInstruct",
     "Llama3Instruct8B",

substrate/core/models.py

@@ -242,6 +242,16 @@ class Config:
     """
 
 
+class GenerateCodeChoice(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    code: str
+    """
+    Code response.
+    """
+
+
 class GenerateCodeIn(BaseModel):
     class Config:
         extra = Extra.allow
@@ -251,18 +261,22 @@ class Config:
     Input prompt.
     """
     language: Literal[
-        "python",
-        "java",
+        "c",
         "c++",
-        "javascript",
-        "typescript",
-        "php",
-        "html",
         "c#",
-        "sql",
+        "css",
+        "go",
+        "html",
+        "java",
+        "javascript",
+        "json",
+        "python",
+        "r",
         "ruby",
-        "tex",
         "shell",
+        "sql",
+        "tex",
+        "typescript",
     ]
     """
     Language of the code.
@@ -287,6 +301,59 @@ class Config:
     """
 
 
+class MultiGenerateCodeIn(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    prompt: str
+    """
+    Input prompt.
+    """
+    language: Literal[
+        "c",
+        "c++",
+        "c#",
+        "css",
+        "go",
+        "html",
+        "java",
+        "javascript",
+        "json",
+        "python",
+        "r",
+        "ruby",
+        "shell",
+        "sql",
+        "tex",
+        "typescript",
+    ]
+    """
+    Language of the code.
+    """
+    num_choices: Annotated[int, Field(ge=1, le=8)] = 1
+    """
+    Number of choices to generate.
+    """
+    temperature: Annotated[Optional[float], Field(ge=0.0, le=1.0)] = None
+    """
+    Higher values make the output more random, lower values make the output more deterministic.
+    """
+    max_tokens: Optional[int] = None
+    """
+    Maximum number of tokens to generate.
+    """
+
+
+class MultiGenerateCodeOut(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    choices: List[GenerateCodeChoice]
+    """
+    Code response choices.
+    """
+
+
 class MultiComputeTextIn(BaseModel):
     class Config:
         extra = Extra.allow

substrate/future_dataclass_models.py

@@ -301,6 +301,19 @@ class FutureComputeJSONOut:
     """
 
 
+@dataclass
+class GenerateCodeChoice:
+    """
+    Future reference to GenerateCodeChoice
+    """
+
+    code: str
+    """
+    (Future reference)
+    Code response.
+    """
+
+
 @dataclass
 class FutureGenerateCodeIn:
     """
@@ -313,18 +326,22 @@ class FutureGenerateCodeIn:
     Input prompt.
     """
     language: Literal[
-        "python",
-        "java",
+        "c",
         "c++",
-        "javascript",
-        "typescript",
-        "php",
-        "html",
         "c#",
-        "sql",
+        "css",
+        "go",
+        "html",
+        "java",
+        "javascript",
+        "json",
+        "python",
+        "r",
         "ruby",
-        "tex",
         "shell",
+        "sql",
+        "tex",
+        "typescript",
     ]
     """
     (Future reference)
@@ -355,6 +372,69 @@ class FutureGenerateCodeOut:
     """
 
 
+@dataclass
+class FutureMultiGenerateCodeIn:
+    """
+    Future reference to FutureMultiGenerateCodeIn
+    """
+
+    prompt: str
+    """
+    (Future reference)
+    Input prompt.
+    """
+    language: Literal[
+        "c",
+        "c++",
+        "c#",
+        "css",
+        "go",
+        "html",
+        "java",
+        "javascript",
+        "json",
+        "python",
+        "r",
+        "ruby",
+        "shell",
+        "sql",
+        "tex",
+        "typescript",
+    ]
+    """
+    (Future reference)
+    Language of the code.
+    """
+    num_choices: int = 1
+    """
+    (Future reference)
+    Number of choices to generate.
+    """
+    temperature: Optional[float] = None
+    """
+    (Future reference)
+    Higher values make the output more random, lower values make the output more deterministic.
+    """
+    max_tokens: Optional[int] = None
+    """
+    (Future reference)
+    Maximum number of tokens to generate.
+    """
+
+
+@dataclass
+class FutureMultiGenerateCodeOut:
+    """
+    Future reference to FutureMultiGenerateCodeOut
+    """
+
+    choices: List[GenerateCodeChoice]
+    """
+    (Future reference)
+    Code response choices.
+    """
+
+
 @dataclass
 class FutureMultiComputeTextIn:
     """

substrate/nodes.py

@@ -50,6 +50,7 @@
         InterpolateFramesOut,
         Llama3Instruct70BOut,
         Mistral7BInstructOut,
+        MultiGenerateCodeOut,
         MultiInpaintImageOut,
         SegmentUnderPointOut,
         MultiGenerateImageOut,
@@ -102,6 +103,7 @@
     FutureInterpolateFramesOut,
     FutureLlama3Instruct70BOut,
     FutureMistral7BInstructOut,
+    FutureMultiGenerateCodeOut,
     FutureMultiInpaintImageOut,
     FutureSegmentUnderPointOut,
     FutureMultiGenerateImageOut,
@@ -328,18 +330,22 @@ def __init__(
         self,
         prompt: str,
         language: Literal[
-            "python",
-            "java",
+            "c",
             "c++",
-            "javascript",
-            "typescript",
-            "php",
-            "html",
             "c#",
-            "sql",
+            "css",
+            "go",
+            "html",
+            "java",
+            "javascript",
+            "json",
+            "python",
+            "r",
             "ruby",
-            "tex",
             "shell",
+            "sql",
+            "tex",
+            "typescript",
         ],
         temperature: Optional[float] = None,
         max_tokens: Optional[int] = None,
@@ -376,6 +382,68 @@ def future(self) -> FutureGenerateCodeOut:  # type: ignore
         return super().future  # type: ignore
 
 
+class MultiGenerateCode(CoreNode[MultiGenerateCodeOut]):
+    """https://substrate.run/nodes#MultiGenerateCode"""
+
+    def __init__(
+        self,
+        prompt: str,
+        language: Literal[
+            "c",
+            "c++",
+            "c#",
+            "css",
+            "go",
+            "html",
+            "java",
+            "javascript",
+            "json",
+            "python",
+            "r",
+            "ruby",
+            "shell",
+            "sql",
+            "tex",
+            "typescript",
+        ],
+        num_choices: int = 1,
+        temperature: Optional[float] = None,
+        max_tokens: Optional[int] = None,
+        hide: bool = False,
+        **kwargs,
+    ):
+        """
+        Args:
+            prompt: Input prompt.
+            language: Language of the code.
+            num_choices: Number of choices to generate.
+            temperature: Higher values make the output more random, lower values make the output more deterministic.
+            max_tokens: Maximum number of tokens to generate.
+
+        https://substrate.run/nodes#MultiGenerateCode
+        """
+        super().__init__(
+            prompt=prompt,
+            language=language,
+            num_choices=num_choices,
+            temperature=temperature,
+            max_tokens=max_tokens,
+            hide=hide,
+            out_type=MultiGenerateCodeOut,
+            **kwargs,
+        )
+        self.node = "MultiGenerateCode"
+
+    @property
+    def future(self) -> FutureMultiGenerateCodeOut:  # type: ignore
+        """
+        Future reference to this node's output.
+
+        https://substrate.run/nodes#MultiGenerateCode
+        """
+        return super().future  # type: ignore
+
+
 class MultiComputeText(CoreNode[MultiComputeTextOut]):
     """https://substrate.run/nodes#MultiComputeText"""