diff --git a/civitai/models.py b/civitai/models.py index a79411d..ef5a48c 100644 --- a/civitai/models.py +++ b/civitai/models.py @@ -1,4 +1,5 @@ from enum import Enum +from typing import List from pydantic import BaseModel, Field class ResourceTypes(str, Enum): @@ -53,7 +54,7 @@ class CommandActivitiesList(Command): class CommandResourcesList(Command): type: CommandTypes = Field(default=CommandTypes.ResourcesList, title="Type", description="The type of command to execute.") - types: list[ResourceTypes] = Field(default=[], title="Types", description="The types of resources to list.") + types: List[ResourceTypes] = Field(default=[], title="Types", description="The types of resources to list.") class CommandResourcesAdd(Command): type: CommandTypes = Field(default=CommandTypes.ResourcesAdd, title="Type", description="The type of command to execute.") diff --git a/scripts/main.py b/scripts/main.py index 9601467..0f8e0bb 100644 --- a/scripts/main.py +++ b/scripts/main.py @@ -1,6 +1,7 @@ # main ui from datetime import datetime, timezone import time +from typing import List import gradio as gr import socketio import os @@ -29,8 +30,8 @@ def on_activities_clear(payload: CommandActivitiesList): command_response(payload) report_interval = 1 -processing_activites: list[str] = [] -should_cancel_activity: list[str] = [] +processing_activites: List[str] = [] +should_cancel_activity: List[str] = [] def on_resources_add(payload: CommandResourcesAdd): resource = payload['resource'] payload['status'] = 'processing'