Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix api typing for scripts

Browse Source
This commit is contained in:
Vladimir Mandic
2023-09-15 10:09:06 -04:00
parent c7f3f57093
commit d3177fc560
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/api/api.py
+5 -3
View File
@@ -2,7 +2,7 @@ import io
import time
import base64
from io import BytesIO
from typing import List, Dict, Any
from typing import List, Dict, Any, Optional
from threading import Lock
from secrets import compare_digest
from fastapi import FastAPI, APIRouter, Depends
@@ -180,10 +180,12 @@ class Api:
i2ilist = [script.name for script in scripts.scripts_img2img.scripts if script.name is not None]
return models.ScriptsList(txt2img = t2ilist, img2img = i2ilist)
def get_script_info(self):
def get_script_info(self, script_name: Optional[str] = None):
res = []
for script_list in [scripts.scripts_txt2img.scripts, scripts.scripts_img2img.scripts]:
res += [script.api_info for script in script_list if script.api_info is not None]
for script in script_list:
if script.api_info is not None and (script_name is None or script_name == script.api_info.name):
res.append(script.api_info)
return res
def get_script(self, script_name, script_runner):