novelai-python 0.5.3__tar.gz → 0.6.0__tar.gz

{novelai_python-0.5.3 → novelai_python-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: novelai-python
-Version: 0.5.3
+Version: 0.6.0
 Summary: NovelAI Python Binding With Pydantic
 Author-Email: sudoskys <coldlando@hotmail.com>
 License: Apache License 2.0
@@ -10,7 +10,7 @@ Requires-Python: >=3.9
 Requires-Dist: elara>=0.5.5
 Requires-Dist: loguru>=0.7.2
 Requires-Dist: pydantic-settings>=2.1.0
-Requires-Dist: pydantic>=2.5.3
+Requires-Dist: pydantic>=2.9.0
 Requires-Dist: httpx>=0.26.0
 Requires-Dist: shortuuid>=1.0.11
 Requires-Dist: Pillow>=10.2.0
@@ -41,7 +41,7 @@ Description-Content-Type: text/markdown
 ---
 
 [![PyPI version](https://badge.fury.io/py/novelai-python.svg)](https://badge.fury.io/py/novelai-python)
-[![Downloads](https://pepy.tech/badge/novelai_python)](https://pepy.tech/project/novelai_python)
+[![Downloads](https://static.pepy.tech/badge/novelai_python)](https://static.pepy.tech/project/novelai_python)
 
 ✨ NovelAI api python sdk with Pydantic, modern and user-friendly.
 
@@ -91,14 +91,39 @@ from dotenv import load_dotenv
 from pydantic import SecretStr
 
 from novelai_python import GenerateImageInfer, ImageGenerateResp, ApiCredential
+from novelai_python.sdk.ai.generate_image import Model, Character, Sampler, UCPreset
+from novelai_python.sdk.ai.generate_image.schema import PositionMap
 
 load_dotenv()
-enhance = "year 2023,dynamic angle,  best quality, amazing quality, very aesthetic, absurdres"
 session = ApiCredential(api_token=SecretStr(os.getenv("NOVELAI_JWT")))  # pst-***
 
+prompt = "1girl, year 2023,dynamic angle,  best quality, amazing quality, very aesthetic, absurdres"
+
 
 async def main():
-    gen = await GenerateImageInfer.build(prompt=f"1girl,{enhance}")
+    gen = GenerateImageInfer.build_generate(
+        prompt=prompt,
+        model=Model.NAI_DIFFUSION_4_CURATED_PREVIEW,
+        character_prompts=[
+            Character(
+                prompt="1girl",
+                uc="red hair",
+                center=PositionMap.AUTO
+            ),
+            Character(
+                prompt="1boy",
+                center=PositionMap.E5
+            )
+        ],
+        sampler=Sampler.K_EULER_ANCESTRAL,
+        ucPreset=UCPreset.TYPE0,
+        # Recommended, using preset negative_prompt depends on selected model
+        qualitySuffix=True,
+        qualityToggle=True,
+        decrisp_mode=False,
+        variety_boost=True,
+        # Checkbox in novelai.net
+    )
     cost = gen.calculate_cost(is_opus=True)
     print(f"charge: {cost} if you are vip3")
     resp = gen.request(session=session)

{novelai_python-0.5.3 → novelai_python-0.6.0}/README.md

@@ -3,7 +3,7 @@
 ---
 
 [![PyPI version](https://badge.fury.io/py/novelai-python.svg)](https://badge.fury.io/py/novelai-python)
-[![Downloads](https://pepy.tech/badge/novelai_python)](https://pepy.tech/project/novelai_python)
+[![Downloads](https://static.pepy.tech/badge/novelai_python)](https://static.pepy.tech/project/novelai_python)
 
 ✨ NovelAI api python sdk with Pydantic, modern and user-friendly.
 
@@ -53,14 +53,39 @@ from dotenv import load_dotenv
 from pydantic import SecretStr
 
 from novelai_python import GenerateImageInfer, ImageGenerateResp, ApiCredential
+from novelai_python.sdk.ai.generate_image import Model, Character, Sampler, UCPreset
+from novelai_python.sdk.ai.generate_image.schema import PositionMap
 
 load_dotenv()
-enhance = "year 2023,dynamic angle,  best quality, amazing quality, very aesthetic, absurdres"
 session = ApiCredential(api_token=SecretStr(os.getenv("NOVELAI_JWT")))  # pst-***
 
+prompt = "1girl, year 2023,dynamic angle,  best quality, amazing quality, very aesthetic, absurdres"
+
 
 async def main():
-    gen = await GenerateImageInfer.build(prompt=f"1girl,{enhance}")
+    gen = GenerateImageInfer.build_generate(
+        prompt=prompt,
+        model=Model.NAI_DIFFUSION_4_CURATED_PREVIEW,
+        character_prompts=[
+            Character(
+                prompt="1girl",
+                uc="red hair",
+                center=PositionMap.AUTO
+            ),
+            Character(
+                prompt="1boy",
+                center=PositionMap.E5
+            )
+        ],
+        sampler=Sampler.K_EULER_ANCESTRAL,
+        ucPreset=UCPreset.TYPE0,
+        # Recommended, using preset negative_prompt depends on selected model
+        qualitySuffix=True,
+        qualityToggle=True,
+        decrisp_mode=False,
+        variety_boost=True,
+        # Checkbox in novelai.net
+    )
     cost = gen.calculate_cost(is_opus=True)
     print(f"charge: {cost} if you are vip3")
     resp = gen.request(session=session)

{novelai_python-0.5.3 → novelai_python-0.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "novelai-python"
-version = "0.5.3"
+version = "0.6.0"
 description = "NovelAI Python Binding With Pydantic"
 authors = [
     { name = "sudoskys", email = "coldlando@hotmail.com" },
@@ -9,7 +9,7 @@ dependencies = [
     "elara>=0.5.5",
     "loguru>=0.7.2",
     "pydantic-settings>=2.1.0",
-    "pydantic>=2.5.3",
+    "pydantic>=2.9.0",
     "httpx>=0.26.0",
     "shortuuid>=1.0.11",
     "Pillow>=10.2.0",

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/_response/__init__.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/1/26 上午10:51
 # @Author  : sudoskys
 # @File    : __init__.py.py
-# @Software: PyCharm
+
 from .ai.generate_image import ImageGenerateResp
 from .user.information import InformationResp
 from .user.login import LoginResp

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/_response/ai/__init__.py

@@ -2,4 +2,4 @@
 # @Time    : 2024/1/26 上午11:20
 # @Author  : sudoskys
 # @File    : __init__.py
-# @Software: PyCharm
+

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/_response/ai/generate_image.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/1/26 上午11:16
 # @Author  : sudoskys
 # @File    : text2image.py
-# @Software: PyCharm
+
 from typing import Tuple, List
 
 from pydantic import BaseModel

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/_response/ai/upscale.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/2/13 上午11:29
 # @Author  : sudoskys
 # @File    : upscale.py
-# @Software: PyCharm
+
 from typing import Tuple
 
 from pydantic import BaseModel

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/_response/schema.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/2/13 上午11:30
 # @Author  : sudoskys
 # @File    : schema.py
-# @Software: PyCharm
+
 
 from pydantic import BaseModel
 

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/_response/user/__init__.py

@@ -2,4 +2,4 @@
 # @Time    : 2024/2/7 上午9:57
 # @Author  : sudoskys
 # @File    : __init__.py.py
-# @Software: PyCharm
+

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/_response/user/information.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/2/8 下午3:10
 # @Author  : sudoskys
 # @File    : information.py
-# @Software: PyCharm
+
 
 from pydantic import Field
 

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/_response/user/login.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/2/7 上午11:57
 # @Author  : sudoskys
 # @File    : login.py
-# @Software: PyCharm
+
 
 from ..schema import RespBase
 

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/_response/user/subscription.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/2/7 上午9:57
 # @Author  : sudoskys
 # @File    : subscription.py
-# @Software: PyCharm
+
 from typing import Optional, Dict, Any, List
 
 from pydantic import BaseModel, Field

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/credential/ApiToken.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/2/8 下午3:05
 # @Author  : sudoskys
 # @File    : ApiToken.py
-# @Software: PyCharm
+
 
 import arrow
 import shortuuid
@@ -18,7 +18,7 @@ class ApiCredential(CredentialBase):
     ApiCredential is the base class for all credential.
     """
     api_token: SecretStr = Field(None, description="api token")
-    _x_correlation_id: str = shortuuid.uuid()[0:6]
+    x_correlation_id: str = shortuuid.uuid()[0:6]
 
     async def get_session(self, timeout: int = 180, update_headers: dict = None):
         headers = {
@@ -29,7 +29,7 @@ class ApiCredential(CredentialBase):
             "Content-Type": "application/json",
             "Origin": "https://novelai.net",
             "Referer": "https://novelai.net/",
-            "x-correlation-id": self._x_correlation_id,
+            "x-correlation-id": self.x_correlation_id,
             "x-initiated-at": f"{arrow.utcnow().isoformat()}Z",
         }
 

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/credential/JwtToken.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/1/26 上午11:04
 # @Author  : sudoskys
 # @File    : JwtToken.py
-# @Software: PyCharm
+
 
 import arrow
 import shortuuid
@@ -18,7 +18,7 @@ class JwtCredential(CredentialBase):
     JwtCredential is the base class for all credential.
     """
     jwt_token: SecretStr = Field(None, description="jwt token")
-    _x_correlation_id: str = shortuuid.uuid()[0:6]
+    x_correlation_id: str = shortuuid.uuid()[0:6]
 
     async def get_session(self, timeout: int = 180, update_headers: dict = None):
         headers = {
@@ -29,7 +29,7 @@ class JwtCredential(CredentialBase):
             "Content-Type": "application/json",
             "Origin": "https://novelai.net",
             "Referer": "https://novelai.net/",
-            "x-correlation-id": self._x_correlation_id,
+            "x-correlation-id": self.x_correlation_id,
             "x-initiated-at": f"{arrow.utcnow().isoformat()}Z",
         }
 

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/credential/UserAuth.py

@@ -19,9 +19,9 @@ class LoginCredential(CredentialBase):
     """
     username: str = Field(None, description="username")
     password: SecretStr = Field(None, description="password")
-    _session_headers: dict = {}
-    _update_at: Optional[int] = None
-    _x_correlation_id: str = shortuuid.uuid()[0:6]
+    session_headers: dict = Field(default_factory=dict)
+    update_at: Optional[int] = None
+    x_correlation_id: str = shortuuid.uuid()[0:6]
 
     async def get_session(self, timeout: int = 180, update_headers: dict = None):
         headers = {
@@ -32,19 +32,19 @@ class LoginCredential(CredentialBase):
             "Content-Type": "application/json",
             "Origin": "https://novelai.net",
             "Referer": "https://novelai.net/",
-            "x-correlation-id": self._x_correlation_id,
+            "x-correlation-id": self.x_correlation_id,
             "x-initiated-at": f"{arrow.utcnow().isoformat()}Z",
         }
 
         # 30 天有效期
-        if not self._session_headers or int(time.time()) - self._update_at > 29 * 24 * 60 * 60:
+        if not self.session_headers or int(time.time()) - self.update_at > 29 * 24 * 60 * 60:
             from ..sdk import Login
             resp = await Login.build(user_name=self.username, password=self.password.get_secret_value()).request()
             headers["Authorization"] = f"Bearer {resp.accessToken}"
-            self._session_headers = headers
-            self._update_at = int(time.time())
+            self.session_headers = headers
+            self.update_at = int(time.time())
         else:
-            headers.update(self._session_headers)
+            headers.update(self.session_headers)
 
         if update_headers:
             headers.update(update_headers)

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/credential/__init__.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/1/26 上午10:51
 # @Author  : sudoskys
 # @File    : __init__.py.py
-# @Software: PyCharm
+
 from pydantic import SecretStr
 
 from .ApiToken import ApiCredential

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/credential/_base.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/2/7 下午12:14
 # @Author  : sudoskys
 # @File    : _shema.py
-# @Software: PyCharm
+
 from curl_cffi.requests import AsyncSession
 from fake_useragent import UserAgent
 from pydantic import BaseModel

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/sdk/__init__.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/1/26 上午10:51
 # @Author  : sudoskys
 # @File    : __init__.py.py
-# @Software: PyCharm
+
 
 from .ai.augment_image import AugmentImageInfer  # noqa 401
 from .ai.generate import LLM, LLMResp  # noqa 401

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/sdk/ai/__init__.py

@@ -2,4 +2,4 @@
 # @Time    : 2024/1/26 上午11:20
 # @Author  : sudoskys
 # @File    : __init__.py.py
-# @Software: PyCharm
+

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/sdk/ai/_cost.py

@@ -11,9 +11,9 @@ from novelai_python.sdk.ai._enum import Sampler, ModelGroups, get_model_group, M
 class Args(BaseModel):
     height: int
     width: int
-    image: Optional[str]
+    image: bool
     n_samples: int
-    strength: float
+    strength: Optional[float]
     steps: int
     sm: bool
     sm_dyn: bool
@@ -34,11 +34,10 @@ class CostCalculator:
             width: int,
             height: int,
             steps: int,
-            image: str,
+            image: bool,
             n_samples: int,
             account_tier: int,
-            strength: float,
-
+            strength: Optional[float],
             is_sm_enabled: bool,
             is_sm_dynamic: bool,
             is_account_active: bool,
@@ -138,14 +137,15 @@ class CostCalculator:
             dimension = 65536
 
         base_dimension = 1048576
-        strength_multiplier = params.strength if params.image else 1
+        strength_multiplier = params.strength if (isinstance(params.strength, float) and params.image) else 1
         steps = params.steps
         samples = params.n_samples
 
         if steps <= 28 and dimension <= base_dimension and account_tier >= 3 and is_account_active and not is_tool_active:
             samples -= 1
 
-        if model_group and model_group in [ModelGroups.STABLE_DIFFUSION_XL, ModelGroups.STABLE_DIFFUSION_XL_FURRY]:
+        if model_group and model_group in [ModelGroups.STABLE_DIFFUSION_XL, ModelGroups.STABLE_DIFFUSION_XL_FURRY,
+                                           ModelGroups.V4]:
             cost = calculate_dimension_cost(img_width, img_height, steps, params.sm, params.sm_dyn)
         elif dimension <= base_dimension and params.sampler in [
             Sampler.PLMS, Sampler.DDIM,

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/sdk/ai/_enum.py

@@ -2,7 +2,6 @@
 # @Time    : 2024/2/13 下午8:10
 # @Author  : sudoskys
 # @File    : _enum.py
-# @Software: PyCharm
 from enum import Enum, IntEnum
 from typing import List, Optional, Union
 
@@ -92,6 +91,8 @@ class Resolution(Enum):
 
 
 class Model(Enum):
+    NAI_DIFFUSION_4_CURATED_PREVIEW = "nai-diffusion-4-curated-preview"
+
     NAI_DIFFUSION_3 = "nai-diffusion-3"
     NAI_DIFFUSION_3_INPAINTING = "nai-diffusion-3-inpainting"
     NAI_DIFFUSION_FURRY_3 = "nai-diffusion-furry-3"
@@ -112,6 +113,7 @@ class Model(Enum):
     WAIFU_DIFFUSION = "waifu-diffusion"
     CURATED_DIFFUSION_TEST = "curated-diffusion-test"
     NAI_DIFFUSION_XL = "nai-diffusion-xl"
+    DALLE_MINI = "dalle-mini"
 
 
 class ModelGroups(Enum):
@@ -119,6 +121,7 @@ class ModelGroups(Enum):
     STABLE_DIFFUSION_GROUP_2 = "stable_diffusion_group2"
     STABLE_DIFFUSION_XL = "stable_diffusion_xl"
     STABLE_DIFFUSION_XL_FURRY = "stable_diffusion_xl_furry"
+    V4 = "v4"
 
 
 INPAINTING_MODEL_LIST = [
@@ -155,6 +158,158 @@ ImageBytesTypeAlias = Optional[Union[str, bytes]]
 UCPresetTypeAlias = Optional[Union[UCPreset, int]]
 
 
+@dataclass
+class SupportCondition:
+    controlnet: bool
+    vibetransfer: bool
+    scaleMax: int
+    negativePromptGuidance: bool
+    noiseSchedule: bool
+    inpainting: bool
+    cfgDelay: bool
+    characterPrompts: bool
+    v4Prompts: bool
+    smea: bool
+    text: bool
+
+
+def get_supported_params(model: Model):
+    """
+    Get supported parameters for a given model
+    :param model: Model
+    :return: SupportCondition
+    """
+    if model in [
+        Model.STABLE_DIFFUSION,
+        Model.NAI_DIFFUSION,
+        Model.SAFE_DIFFUSION,
+        Model.WAIFU_DIFFUSION,
+        Model.NAI_DIFFUSION_FURRY,
+        Model.CURATED_DIFFUSION_TEST,
+        Model.NAI_DIFFUSION_INPAINTING,
+        Model.SAFE_DIFFUSION_INPAINTING,
+        Model.FURRY_DIFFUSION_INPAINTING
+    ]:
+        return SupportCondition(
+            controlnet=True,
+            vibetransfer=False,
+            scaleMax=25,
+            negativePromptGuidance=False,
+            noiseSchedule=False,
+            inpainting=True,
+            cfgDelay=False,
+            characterPrompts=False,
+            v4Prompts=False,
+            smea=True,
+            text=False
+        )
+    if model in [Model.NAI_DIFFUSION_2]:
+        return SupportCondition(
+            controlnet=True,
+            vibetransfer=False,
+            scaleMax=25,
+            negativePromptGuidance=True,
+            noiseSchedule=False,
+            inpainting=True,
+            cfgDelay=False,
+            characterPrompts=False,
+            v4Prompts=False,
+            smea=True,
+            text=False
+        )
+    if model in [
+        Model.NAI_DIFFUSION_XL,
+        Model.NAI_DIFFUSION_3,
+        Model.NAI_DIFFUSION_3_INPAINTING,
+        Model.NAI_DIFFUSION_FURRY_3,
+        Model.NAI_DIFFUSION_FURRY_3_INPAINTING
+    ]:
+        return SupportCondition(
+            controlnet=False,
+            vibetransfer=True,
+            scaleMax=10,
+            negativePromptGuidance=True,
+            noiseSchedule=True,
+            inpainting=True,
+            cfgDelay=True,
+            characterPrompts=False,
+            v4Prompts=False,
+            smea=True,
+            text=False
+        )
+    if model in [Model.CUSTOM, Model.NAI_DIFFUSION_4_CURATED_PREVIEW]:
+        return SupportCondition(
+            controlnet=False,
+            vibetransfer=False,
+            scaleMax=10,
+            negativePromptGuidance=True,
+            noiseSchedule=True,
+            inpainting=True,
+            cfgDelay=False,
+            characterPrompts=True,
+            v4Prompts=True,
+            smea=False,
+            text=True
+        )
+    return SupportCondition(
+        controlnet=False,
+        vibetransfer=False,
+        scaleMax=25,
+        negativePromptGuidance=False,
+        noiseSchedule=False,
+        inpainting=False,
+        cfgDelay=False,
+        characterPrompts=False,
+        v4Prompts=False,
+        smea=False,
+        text=False
+    )
+
+
+@dataclass
+class Modifier(object):
+    qualityTags: str
+    suffix: str
+
+
+def get_modifiers(model: Model) -> Modifier:
+    """
+    Get modifiers from model
+    :param model:
+    :return:
+    """
+    if model in [Model.CUSTOM, Model.NAI_DIFFUSION_4_CURATED_PREVIEW]:
+        return Modifier(
+            qualityTags="",
+            suffix=", rating:general, amazing quality, very aesthetic, absurdres"
+        )
+    if model in [Model.NAI_DIFFUSION_2]:
+        return Modifier(
+            qualityTags="very aesthetic, best quality, absurdres, ",
+            suffix=""
+        )
+    if model in [
+        Model.NAI_DIFFUSION_3,
+        Model.NAI_DIFFUSION_3_INPAINTING
+    ]:
+        return Modifier(
+            qualityTags="",
+            suffix=", best quality, amazing quality, very aesthetic, absurdres"
+        )
+    if model in [
+        Model.NAI_DIFFUSION_FURRY_3,
+        Model.NAI_DIFFUSION_FURRY_3_INPAINTING
+    ]:
+        return Modifier(
+            qualityTags="",
+            suffix=", {best quality}, {amazing quality}"
+        )
+    return Modifier(
+        qualityTags="masterpiece, best quality, ",
+        suffix=""
+    )
+
+
 def get_supported_noise_schedule(sample_type: Sampler) -> List[NoiseSchedule]:
     """
     Get supported noise schedule for a given sample type
@@ -224,9 +379,10 @@ def get_model_group(model: ModelTypeAlias) -> ModelGroups:
         "nai-diffusion-xl": ModelGroups.STABLE_DIFFUSION_XL,
         "nai-diffusion-3": ModelGroups.STABLE_DIFFUSION_XL,
         "nai-diffusion-3-inpainting": ModelGroups.STABLE_DIFFUSION_XL,
-        "custom": ModelGroups.STABLE_DIFFUSION_XL,
+        "custom": ModelGroups.V4,
         "nai-diffusion-furry-3": ModelGroups.STABLE_DIFFUSION_XL_FURRY,
         "nai-diffusion-furry-3-inpainting": ModelGroups.STABLE_DIFFUSION_XL_FURRY,
+        "nai-diffusion-4-curated-preview": ModelGroups.V4,
     }
     return mapping.get(model, ModelGroups.STABLE_DIFFUSION)
 

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/sdk/ai/augment_image/__init__.py

@@ -2,7 +2,7 @@
 # @Time    : 2024/2/13 下午8:08
 # @Author  : sudoskys
 # @File    : __init__.py.py
-# @Software: PyCharm
+
 import base64
 import io
 import json
@@ -85,7 +85,7 @@ class AugmentImageInfer(ApiBaseModel):
                     height=self.height,
                     steps=28,
                     model=Model.NAI_DIFFUSION_3,
-                    image=self.image,
+                    image=bool(self.image),
                     n_samples=1,
                     account_tier=1,
                     strength=1,
@@ -100,7 +100,7 @@ class AugmentImageInfer(ApiBaseModel):
                 height=self.height,
                 steps=28,
                 model=Model.NAI_DIFFUSION_3,
-                image=self.image,
+                image=bool(self.image),
                 n_samples=1,
                 account_tier=3 if is_opus else 1,
                 strength=1,

{novelai_python-0.5.3 → novelai_python-0.6.0}/src/novelai_python/sdk/ai/generate/__init__.py

@@ -306,6 +306,7 @@ class LLM(ApiBaseModel):
               ) -> "LLM":
         """
         Generate
+        :param phrase_rep_pen: Phrase_rep_pen
         :param prompt: prompt, exp "Hello, World!"
         :param model: TextLLMModelTypeAlias
         :param advanced_setting: AdvanceLLMSetting, default None means auto-select