refactor: Refactor TRPG NER model SDK: restructure codebase into base_model_trpgner package, implement training and inference modules, and add model download functionality. Remove legacy training and utils modules. Enhance documentation and examples for better usability.

5 files changed, 9 insertions, 9 deletions

diff --git a/src/basemodel/__init__.py b/src/base_model_trpgner/__init__.py
index 7287df4..9796c83 100644
--- a/src/basemodel/__init__.py
+++ b/src/base_model_trpgner/__init__.py
@@ -1,28 +1,28 @@
 """
-base-model - HydroRoll TRPG NER 模型 SDK
+base-model-trpgner - HydroRoll TRPG NER 模型 SDK
 
 这是一个用于 TRPG（桌上角色扮演游戏）日志命名实体识别的 Python SDK。
 
 基本用法:
-    >>> from basemodel import TRPGParser
+    >>> from base_model_trpgner import TRPGParser
     >>> parser = TRPGParser()
     >>> result = parser.parse("风雨 2024-06-08 21:44:59 剧烈的疼痛...")
     >>> print(result)
     {'metadata': {'speaker': '风雨', 'timestamp': '2024-06-08 21:44:59'}, 'content': [...]}
 
 训练功能（需要额外安装）:
-    >>> pip install base-model[train]
-    >>> from basemodel.training import train_ner_model
+    >>> pip install base-model-trpgner[train]
+    >>> from base_model_trpgner.training import train_ner_model
     >>> train_ner_model(conll_data="./data", output_dir="./model")
 """
 
-from basemodel.inference import TRPGParser, parse_line, parse_lines
+from base_model_trpgner.inference import TRPGParser, parse_line, parse_lines
 
 try:
     from importlib.metadata import version
-    __version__ = version("base-model")
+    __version__ = version("base_model_trpgner")
 except Exception:
-    __version__ = "0.1.0.dev"
+    __version__ = "0.1.1.dev"
 
 __all__ = [
     "__version__",
diff --git a/src/basemodel/download_model.py b/src/base_model_trpgner/download_model.py
index 2d65099..2d65099 100644
--- a/src/basemodel/download_model.py
+++ b/src/base_model_trpgner/download_model.py
diff --git a/src/basemodel/inference/__init__.py b/src/base_model_trpgner/inference/__init__.py
index 93a185f..93a185f 100644
--- a/src/basemodel/inference/__init__.py
+++ b/src/base_model_trpgner/inference/__init__.py
diff --git a/src/basemodel/training/__init__.py b/src/base_model_trpgner/training/__init__.py
index 5671c42..ccf3c03 100644
--- a/src/basemodel/training/__init__.py
+++ b/src/base_model_trpgner/training/__init__.py
@@ -36,7 +36,7 @@ def train_ner_model(
         resume_from_checkpoint: 恢复检查点路径
 
     Examples:
-        >>> from basemodel.training import train_ner_model
+        >>> from basemodeltrpgner.training import train_ner_model
         >>> train_ner_model(
         ...     conll_data="./data",
         ...     output_dir="./my_model",
@@ -59,7 +59,7 @@ def train_ner_model(
         ) from e
 
     # 导入数据处理函数
-    from basemodel.utils.conll import load_conll_dataset, tokenize_and_align_labels
+    from base_model_trpgner.utils.conll import load_conll_dataset, tokenize_and_align_labels
 
     print(f"🚀 Starting training...")
 
diff --git a/src/basemodel/utils/__init__.py b/src/base_model_trpgner/utils/__init__.py
index 12a3ef4..12a3ef4 100644
--- a/src/basemodel/utils/__init__.py
+++ b/src/base_model_trpgner/utils/__init__.py