mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
nodes: convert InvocationQueueItem to Pydantic class
This commit is contained in:
parent
6f6de402ad
commit
7fc5fbd4ce
@ -2,16 +2,16 @@
|
||||
|
||||
import time
|
||||
from abc import ABC, abstractmethod
|
||||
from dataclasses import dataclass
|
||||
from pydantic import BaseModel, Field
|
||||
from queue import Queue
|
||||
|
||||
|
||||
@dataclass
|
||||
class InvocationQueueItem:
|
||||
class InvocationQueueItem(BaseModel):
|
||||
graph_execution_state_id: str
|
||||
invocation_id: str
|
||||
invoke_all: bool
|
||||
timestamp: float = time.time()
|
||||
timestamp: float = Field(default_factory=time.time)
|
||||
|
||||
|
||||
class InvocationQueueABC(ABC):
|
||||
"""Abstract base class for all invocation queues"""
|
||||
|
Loading…
Reference in New Issue
Block a user