diff --git a/backend/app/database.py b/backend/app/database.py index e574f05..5e442c5 100644 --- a/backend/app/database.py +++ b/backend/app/database.py @@ -21,7 +21,7 @@ from app.models import ( Settings, WritingStyle, ProjectDefaultStyle, RelationshipType, CharacterRelationship, Organization, OrganizationMember, StoryMemory, PlotAnalysis, AnalysisTask, BatchGenerationTask, - RegenerationTask + RegenerationTask, Career, CharacterCareer ) # 引擎缓存:每个用户一个引擎 diff --git a/backend/app/models/__init__.py b/backend/app/models/__init__.py index ded9609..f077796 100644 --- a/backend/app/models/__init__.py +++ b/backend/app/models/__init__.py @@ -14,6 +14,7 @@ from app.models.project_default_style import ProjectDefaultStyle from app.models.mcp_plugin import MCPPlugin from app.models.user import User, UserPassword from app.models.regeneration_task import RegenerationTask +from app.models.career import Career, CharacterCareer __all__ = [ "Project", @@ -35,5 +36,7 @@ __all__ = [ "MCPPlugin", "User", "UserPassword", - "RegenerationTask" + "RegenerationTask", + "Career", + "CharacterCareer" ] \ No newline at end of file