From 6cbb5289156d4ed62577eb7c9ddd866c53b15b5f Mon Sep 17 00:00:00 2001
From: KAAANG <79990647+SAKURA-CAT@users.noreply.github.com>
Date: Mon, 9 Dec 2024 18:35:57 +0800
Subject: [PATCH] refactor: sys-model

---
 pyproject.toml                 |  2 +-
 swankit/callback/__init__.py   |  2 +-
 swankit/callback/models/key.py | 13 +++++++++++--
 3 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index f4c3963..eb22b1f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,7 +9,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "swankit"
-version = "0.1.2b2"
+version = "0.1.2b3"
 dynamic = ["readme", "dependencies"]
 description = "Base toolkit for SwanLab"
 license = "Apache-2.0"
diff --git a/swankit/callback/__init__.py b/swankit/callback/__init__.py
index 339472d..6a8bd8e 100644
--- a/swankit/callback/__init__.py
+++ b/swankit/callback/__init__.py
@@ -107,4 +107,4 @@ def __str__(self) -> str:
         pass
 
 
-__all__ = ["SwanKitCallback", "models"]
+__all__ = ["SwanKitCallback", "ColumnInfo", "MetricInfo", "ColumnConfig", "OperateErrorInfo", "RuntimeInfo"]
diff --git a/swankit/callback/models/key.py b/swankit/callback/models/key.py
index 6c33cb6..cf1c00f 100644
--- a/swankit/callback/models/key.py
+++ b/swankit/callback/models/key.py
@@ -18,12 +18,21 @@
 YRange = Optional[Tuple[Optional[float], Optional[float]]]
 
 
-class ColumnConfig(TypedDict):
+class ColumnConfig:
     """
     列信息配置
     """
 
-    y_range: YRange
+    def __init__(self, y_range: YRange = None, chart_name: Optional[str] = None, metric_name: Optional[str] = None):
+        """
+        生成的列信息配置对象
+        :param y_range: y轴范围
+        :param chart_name: 图表名称
+        :param metric_name: 指标名称
+        """
+        self.y_range: YRange = y_range
+        self.chart_name: Optional[str] = chart_name
+        self.metric_name: Optional[str] = metric_name
 
 
 class ColumnInfo: