python 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,7 +242,7 @@ class Config:
     """
 
 
-class GenerateCodeIn(BaseModel):
+class DeepseekIn(BaseModel):
     class Config:
         extra = Extra.allow
 
@@ -251,18 +251,75 @@ class Config:
     Input prompt.
     """
     language: Literal[
-        "python",
-        "java",
+        "c",
         "c++",
+        "c#",
+        "css",
+        "go",
+        "html",
+        "java",
         "javascript",
+        "json",
+        "python",
+        "r",
+        "ruby",
+        "shell",
+        "sql",
+        "tex",
         "typescript",
-        "php",
-        "html",
+    ]
+    """
+    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 GenerateCodeChoice(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    code: str
+    """
+    Code response.
+    """
+
+
+class GenerateCodeIn(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    prompt: str
+    """
+    Input prompt.
+    """
+    language: Literal[
+        "c",
+        "c++",
         "c#",
-        "sql",
+        "css",
+        "go",
+        "html",
+        "java",
+        "javascript",
+        "json",
+        "python",
+        "r",
         "ruby",
-        "tex",
         "shell",
+        "sql",
+        "tex",
+        "typescript",
     ]
     """
     Language of the code.
@@ -287,6 +344,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
@@ -2217,3 +2327,13 @@ class Config:
     """
     Document chunks
     """
+
+
+class DeepseekOut(BaseModel):
+    class Config:
+        extra = Extra.allow
+
+    choices: List[GenerateCodeChoice]
+    """
+    Code response choices.
+    """

substrate/future_dataclass_models.py

@@ -302,9 +302,9 @@ class FutureComputeJSONOut:
 
 
 @dataclass
-class FutureGenerateCodeIn:
+class FutureDeepseekIn:
     """
-    Future reference to FutureGenerateCodeIn
+    Future reference to FutureDeepseekIn
     """
 
     prompt: str
@@ -313,18 +313,85 @@ class FutureGenerateCodeIn:
     Input prompt.
     """
     language: Literal[
-        "python",
-        "java",
+        "c",
         "c++",
+        "c#",
+        "css",
+        "go",
+        "html",
+        "java",
         "javascript",
+        "json",
+        "python",
+        "r",
+        "ruby",
+        "shell",
+        "sql",
+        "tex",
         "typescript",
-        "php",
-        "html",
+    ]
+    """
+    (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 GenerateCodeChoice:
+    """
+    Future reference to GenerateCodeChoice
+    """
+
+    code: str
+    """
+    (Future reference)
+    Code response.
+    """
+
+
+@dataclass
+class FutureGenerateCodeIn:
+    """
+    Future reference to FutureGenerateCodeIn
+    """
+
+    prompt: str
+    """
+    (Future reference)
+    Input prompt.
+    """
+    language: Literal[
+        "c",
+        "c++",
         "c#",
-        "sql",
+        "css",
+        "go",
+        "html",
+        "java",
+        "javascript",
+        "json",
+        "python",
+        "r",
         "ruby",
-        "tex",
         "shell",
+        "sql",
+        "tex",
+        "typescript",
     ]
     """
     (Future reference)
@@ -355,6 +422,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:
     """
@@ -2814,3 +2944,16 @@ class FutureSplitDocumentOut:
     (Future reference)
     Document chunks
     """
+
+
+@dataclass
+class FutureDeepseekOut:
+    """
+    Future reference to FutureDeepseekOut
+    """
+
+    choices: List[GenerateCodeChoice]
+    """
+    (Future reference)
+    Code response choices.
+    """