Skip to content

Commit 9780471

Browse files
fix: change StrEnum to support lower python versions (#192)
1 parent c81a190 commit 9780471

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed
Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
from enum import StrEnum
1+
from enum import Enum
22

33

4-
class DatabaseEngineType(StrEnum):
4+
class DatabaseEngineType(str, Enum):
55
POSTGRESQL = "postgresql"
66
MYSQL = "mysql"
77
SQLITE = "sqlite"

llmstack/datasources/handlers/databases/sql.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,12 +6,12 @@
66
from typing_extensions import Literal
77

88
from llmstack.common.blocks.base.schema import BaseSchema as _Schema
9+
from llmstack.common.blocks.data.store.database.constants import DatabaseEngineType
910
from llmstack.common.blocks.data.store.database.database_reader import (
1011
DatabaseReader,
1112
DatabaseReaderInput,
1213
)
1314
from llmstack.common.blocks.data.store.database.utils import (
14-
DatabaseEngineType,
1515
get_database_configuration_class,
1616
)
1717
from llmstack.common.blocks.data.store.vectorstore import Document

0 commit comments

Comments
 (0)