🔄 refactor(credentials): update JwtCredential to ApiCredential usage

LlmKira · Jan 5, 2025 · f33e507 · f33e507
1 parent 3cc47a0
commit f33e507
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/playground/augment-image.png b/playground/augment-image.png
diff --git a/playground/augment-image.py b/playground/augment-image.py
@@ -32,7 +32,7 @@ async def generate(
         agent = AugmentImageInfer.build(
             req_type=request_type,
             image=image,
-            mood=Moods.Shy,
+            # mood=Moods.Shy,
             prompt="",
             defry=0,
         )
@@ -45,7 +45,7 @@ async def generate(
         print(f"Error: {e.message}")
         return None
     else:
-        print(f"Meta: {result.meta}")
+        print(f"Meta: {result.meta.endpoint}")
     _res: ImageGenerateResp
     file = result.files[0]
     with open(f"{pathlib.Path(__file__).stem}.png", "wb") as f:

diff --git a/playground/generate_image.py b/playground/generate_image.py
@@ -91,7 +91,7 @@ async def direct_use():
     that's pydantic!
     :return:
     """
-    credential = JwtCredential(jwt_token=SecretStr("pst-5555"))
+    credential = ApiCredential(api_token=SecretStr("pst-5555"))
     result = await GenerateImageInfer(
         input="1girl",
         model=Model.NAI_DIFFUSION_4_CURATED_PREVIEW,