fixed str enum

This commit is contained in:
Yannick Stephan
2025-02-17 18:26:07 +01:00
parent 70ea5ec878
commit 3b30b7985b
2 changed files with 6 additions and 6 deletions

View File

@@ -1,7 +1,7 @@
from __future__ import annotations
from abc import ABC, abstractmethod
from enum import StrEnum
from enum import Enum
import os
from dotenv import load_dotenv
from dataclasses import dataclass, field
@@ -205,7 +205,7 @@ class BaseGraphStorage(StorageNameSpace, ABC):
"""Retrieve a subgraph of the knowledge graph starting from a given node."""
class DocStatus(StrEnum):
class DocStatus(str, Enum):
"""Document processing status"""
PENDING = "pending"