diff --git a/api/core/tools/entities/tool_entities.py b/api/core/tools/entities/tool_entities.py index 0e64e45da5..5a2915c36d 100644 --- a/api/core/tools/entities/tool_entities.py +++ b/api/core/tools/entities/tool_entities.py @@ -38,6 +38,7 @@ class ToolLabelEnum(Enum): BUSINESS = "business" ENTERTAINMENT = "entertainment" UTILITIES = "utilities" + RAG = "rag" OTHER = "other" diff --git a/api/core/tools/entities/values.py b/api/core/tools/entities/values.py index f460df7e25..050a16f42b 100644 --- a/api/core/tools/entities/values.py +++ b/api/core/tools/entities/values.py @@ -49,6 +49,9 @@ ICONS = { """, # noqa: E501 ToolLabelEnum.OTHER: """ +""", # noqa: E501 + ToolLabelEnum.RAG: """ + """, # noqa: E501 } @@ -105,6 +108,9 @@ default_tool_label_dict = { ToolLabelEnum.OTHER: ToolLabel( name="other", label=I18nObject(en_US="Other", zh_Hans="其他"), icon=ICONS[ToolLabelEnum.OTHER] ), + ToolLabelEnum.RAG: ToolLabel( + name="rag", label=I18nObject(en_US="RAG", zh_Hans="RAG"), icon=ICONS[ToolLabelEnum.RAG] + ), } default_tool_labels = [v for k, v in default_tool_label_dict.items()] diff --git a/api/migrations/versions/2025_05_15_1558-b35c3db83d09_add_pipeline_info.py b/api/migrations/versions/2025_05_15_1558-b35c3db83d09_add_pipeline_info.py index 5238c8e34f..5406903f05 100644 --- a/api/migrations/versions/2025_05_15_1558-b35c3db83d09_add_pipeline_info.py +++ b/api/migrations/versions/2025_05_15_1558-b35c3db83d09_add_pipeline_info.py @@ -12,7 +12,7 @@ from sqlalchemy.dialects import postgresql # revision identifiers, used by Alembic. revision = 'b35c3db83d09' -down_revision = '0e154742a5fa' +down_revision = 'b45e25c2d166' branch_labels = None depends_on = None