diff --git a/pyproject.toml b/pyproject.toml
index d47172f9..c477fa2f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "reme_ai"
-version = "0.2.0.0"
+dynamic = ["version"]
 description = "Remember Me, Refine Me."
 authors = [
     { name = "jinli.yl", email = "jinli.yl@alibaba-inc.com" },
@@ -32,7 +32,7 @@ classifiers = [
 keywords = ["llm", "memory", "experience", "memoryscope", "ai", "mcp", "http"]
 
 dependencies = [
-    "flowllm[reme]>=0.2.0.0",
+    "flowllm[reme]>=0.2.0a1",
 ]
 
 [project.optional-dependencies]
@@ -52,6 +52,9 @@ reme_ai = [
     "**/*.json",
 ]
 
+[tool.setuptools.dynamic]
+version = { attr = "reme_ai.__version__" }
+
 [project.urls]
 Homepage = "https://github.com/agentscope-ai/ReMe"
 Documentation = "https://reme.agentscope.io/"
@@ -60,4 +63,4 @@ Repository = "https://github.com/agentscope-ai/ReMe"
 [project.scripts]
 reme = "reme_ai.main:main"
 
-# python -m build && twine upload dist/*
\ No newline at end of file
+# python -m build && twine upload dist/*
diff --git a/reme_ai/__init__.py b/reme_ai/__init__.py
index 62450870..9337f93b 100644
--- a/reme_ai/__init__.py
+++ b/reme_ai/__init__.py
@@ -30,4 +30,4 @@ __all__ = [
     "vector_store",
 ]
 
-__version__ = "0.2.0.0"
+__version__ = "0.2.0a1"
diff --git a/reme_ai/schema/__init__.py b/reme_ai/schema/__init__.py
index 5b18eb6d..70255638 100644
--- a/reme_ai/schema/__init__.py
+++ b/reme_ai/schema/__init__.py
@@ -4,7 +4,8 @@ This module provides data structures and schemas for memory management,
 including memory types, tool call results, and conversion utilities.
 """
 
-from flowllm.core.schema import Message, Role, Trajectory  # noqa
+from flowllm.core.enumeration import Role  # noqa
+from flowllm.core.schema import Message, Trajectory  # noqa
 
 from reme_ai.schema.memory import (
     BaseMemory,