feat: add managed llm summaries and memory timeline

2026-08-30 10:51:32 +08:00
parent 47f8a164fa
commit 962dd1a063
22 changed files with 1417 additions and 86 deletions
@@ -0,0 +1,133 @@
+package httpapi
+
+import (
+	"errors"
+	"net/http"
+	"strings"
+
+	"data-terminal-cloud/internal/concurrency"
+	"data-terminal-cloud/internal/contentmodel"
+)
+
+func ListContentModelsHandler(service *contentmodel.Service) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		principal, ok := webPrincipal(w, r)
+		if !ok {
+			return
+		}
+		tenantID := strings.TrimSpace(r.Header.Get("X-Tenant-Id"))
+		if !validUUID(tenantID) {
+			writeValidation(w, r, "租户参数不正确", FieldError{Field: "X-Tenant-Id", Message: "必须是有效UUID"})
+			return
+		}
+		items, err := service.List(r.Context(), principal, tenantID)
+		if err != nil {
+			writeContentModelError(w, r, err)
+			return
+		}
+		w.Header().Set("Cache-Control", "no-store")
+		WriteJSON(w, r, http.StatusOK, map[string]any{"items": items})
+	})
+}
+
+func CreateContentModelHandler(service *contentmodel.Service) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		principal, ok := webPrincipal(w, r)
+		if !ok {
+			return
+		}
+		var input contentmodel.CreateInput
+		if !decodeWebJSON(w, r, &input, "大模型配置参数不正确") {
+			return
+		}
+		input.TenantID = strings.TrimSpace(r.Header.Get("X-Tenant-Id"))
+		item, err := service.Create(r.Context(), principal, input, RequestID(r.Context()))
+		if err != nil {
+			writeContentModelError(w, r, err)
+			return
+		}
+		w.Header().Set("Cache-Control", "no-store")
+		w.Header().Set("ETag", concurrency.FormatETag(item.Version))
+		WriteJSON(w, r, http.StatusCreated, item)
+	})
+}
+
+func TestContentModelHandler(service *contentmodel.Service) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		principal, ok := resourcePrincipal(w, r, "id")
+		if !ok {
+			return
+		}
+		item, err := service.Test(r.Context(), principal, strings.TrimSpace(r.Header.Get("X-Tenant-Id")), r.PathValue("id"), RequestID(r.Context()))
+		if err != nil {
+			writeContentModelError(w, r, err)
+			return
+		}
+		w.Header().Set("Cache-Control", "no-store")
+		w.Header().Set("ETag", concurrency.FormatETag(item.Version))
+		WriteJSON(w, r, http.StatusOK, item)
+	})
+}
+
+func ActivateContentModelHandler(service *contentmodel.Service) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		principal, ok := resourcePrincipal(w, r, "id")
+		if !ok {
+			return
+		}
+		version, err := concurrency.ParseIfMatch(r)
+		if err != nil {
+			writeETagError(w, r, err)
+			return
+		}
+		item, err := service.Activate(r.Context(), principal, strings.TrimSpace(r.Header.Get("X-Tenant-Id")), r.PathValue("id"), RequestID(r.Context()), version)
+		if err != nil {
+			writeContentModelError(w, r, err)
+			return
+		}
+		w.Header().Set("Cache-Control", "no-store")
+		w.Header().Set("ETag", concurrency.FormatETag(item.Version))
+		WriteJSON(w, r, http.StatusOK, item)
+	})
+}
+
+func RevokeContentModelHandler(service *contentmodel.Service) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		principal, ok := resourcePrincipal(w, r, "id")
+		if !ok {
+			return
+		}
+		version, err := concurrency.ParseIfMatch(r)
+		if err != nil {
+			writeETagError(w, r, err)
+			return
+		}
+		item, err := service.Revoke(r.Context(), principal, strings.TrimSpace(r.Header.Get("X-Tenant-Id")), r.PathValue("id"), RequestID(r.Context()), version)
+		if err != nil {
+			writeContentModelError(w, r, err)
+			return
+		}
+		w.Header().Set("Cache-Control", "no-store")
+		w.Header().Set("ETag", concurrency.FormatETag(item.Version))
+		WriteJSON(w, r, http.StatusOK, item)
+	})
+}
+
+func writeContentModelError(w http.ResponseWriter, r *http.Request, err error) {
+	switch {
+	case errors.Is(err, contentmodel.ErrInvalid):
+		writeValidation(w, r, "大模型配置参数不正确", FieldError{Field: "model", Message: "请检查供应商、地址、模型ID、密钥和生成参数"})
+	case errors.Is(err, contentmodel.ErrAccessDenied):
+		WriteError(w, r, http.StatusForbidden, "CONTENT_MODEL_ACCESS_DENIED", "只有家庭负责人或管理员可以管理总结模型")
+	case errors.Is(err, contentmodel.ErrNotFound):
+		WriteError(w, r, http.StatusNotFound, "RESOURCE_NOT_FOUND", "大模型配置不存在")
+	case errors.Is(err, contentmodel.ErrVersion):
+		WriteError(w, r, http.StatusPreconditionFailed, "VERSION_CONFLICT", "配置已更新，请刷新后重试")
+	case errors.Is(err, contentmodel.ErrTestRequired):
+		WriteError(w, r, http.StatusConflict, "MODEL_TEST_REQUIRED", "请先通过连通性测试再激活")
+	case errors.Is(err, contentmodel.ErrActiveRevoke):
+		WriteError(w, r, http.StatusConflict, "ACTIVE_MODEL_CANNOT_REMOVE", "请先激活另一个模型，再移除当前模型")
+	default:
+		WriteError(w, r, http.StatusInternalServerError, "INTERNAL_ERROR", "大模型配置处理失败")
+	}
+}
@@ -8,6 +8,7 @@ import (
 	"data-terminal-cloud/internal/alerts"
 	"data-terminal-cloud/internal/billing"
 	"data-terminal-cloud/internal/buildinfo"
+	"data-terminal-cloud/internal/contentmodel"
 	"data-terminal-cloud/internal/device"
 	"data-terminal-cloud/internal/diagnostics"
 	"data-terminal-cloud/internal/health"
@@ -42,6 +43,7 @@ type RouterOptions struct {
 	DiagnosticService               *diagnostics.Service
 	OpsAccessService                *opsaccess.Service
 	OpsDeviceService                *opsdevice.Service
+	ContentModelService             *contentmodel.Service
 	ConversationCreator             ConversationCreator
 	ConversationActivator           ConversationActivator
 	DeviceConversationStateReader   DeviceConversationStateReader
@@ -164,6 +166,13 @@ func NewRouter(options RouterOptions) http.Handler {
 			mux.Handle("POST /api/v1/ops/devices/{id}/credentials/rotate", authenticated(RotateOpsDeviceCredentialHandler(options.OpsDeviceService)))
 			mux.Handle("POST /api/v1/ops/devices/{id}/claims", authenticated(IssueOpsDeviceClaimHandler(options.OpsDeviceService)))
 		}
+		if options.ContentModelService != nil {
+			mux.Handle("GET /api/v1/content-models", authenticated(ListContentModelsHandler(options.ContentModelService)))
+			mux.Handle("POST /api/v1/content-models", authenticated(CreateContentModelHandler(options.ContentModelService)))
+			mux.Handle("POST /api/v1/content-models/{id}/test", authenticated(TestContentModelHandler(options.ContentModelService)))
+			mux.Handle("POST /api/v1/content-models/{id}/activate", authenticated(ActivateContentModelHandler(options.ContentModelService)))
+			mux.Handle("DELETE /api/v1/content-models/{id}", authenticated(RevokeContentModelHandler(options.ContentModelService)))
+		}
 		mux.Handle("GET /api/v1/households", authenticated(ListHouseholdsHandler(options.SaaSService)))
 		mux.Handle("POST /api/v1/households", authenticated(CreateHouseholdHandler(options.SaaSService)))
 		mux.Handle("GET /api/v1/households/{id}", authenticated(GetHouseholdHandler(options.SaaSService)))
@@ -226,6 +235,7 @@ func NewRouter(options RouterOptions) http.Handler {
 		}
 		mux.Handle("GET /api/v1/conversations/{id}/turns", authenticated(ListConversationTurnsHandler(options.SaaSService)))
 		mux.Handle("GET /api/v1/conversations/{id}/summaries", authenticated(ListConversationSummariesHandler(options.SaaSService)))
+		mux.Handle("POST /api/v1/conversations/{id}/summaries:regenerate", authenticated(RegenerateConversationSummaryHandler(options.SaaSService)))
 		mux.Handle("GET /api/v1/subjects/{id}/stories", authenticated(ListSubjectStoriesHandler(options.SaaSService)))
 		mux.Handle("GET /api/v1/stories/{id}", authenticated(GetStoryHandler(options.SaaSService)))
 		mux.Handle("PATCH /api/v1/stories/{id}", authenticated(PatchStoryHandler(options.SaaSService)))
@@ -1044,6 +1044,21 @@ func ListConversationSummariesHandler(service *saas.Service) http.Handler {
 	})
 }

+func RegenerateConversationSummaryHandler(service *saas.Service) http.Handler {
+	return http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
+		principal, ok := resourcePrincipal(writer, request, "id")
+		if !ok {
+			return
+		}
+		version, err := service.RegenerateConversationSummary(request.Context(), principal, request.PathValue("id"), RequestID(request.Context()))
+		if err != nil {
+			writeSaaSError(writer, request, err)
+			return
+		}
+		WriteJSON(writer, request, http.StatusAccepted, map[string]any{"generation_version": version, "status": "queued"})
+	})
+}
+
 func ListSubjectStoriesHandler(service *saas.Service) http.Handler {
 	return http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
 		principal, ok := resourcePrincipal(writer, request, "id")
@@ -1921,6 +1936,8 @@ func writeSaaSError(writer http.ResponseWriter, request *http.Request, err error
 		WriteError(writer, request, http.StatusConflict, "STORY_REVIEW_STATE_CONFLICT", "当前故事审核状态不允许此操作")
 	case errors.Is(err, saas.ErrFactReviewStateConflict):
 		WriteError(writer, request, http.StatusConflict, "FACT_REVIEW_STATE_CONFLICT", "当前事实候选状态不允许此操作")
+	case errors.Is(err, saas.ErrConversationSummaryNotReady):
+		WriteError(writer, request, http.StatusConflict, "CONVERSATION_SUMMARY_NOT_READY", "聊天尚未结束或没有可用于总结的文字内容")
 	case errors.Is(err, pagination.ErrInvalidCursor):
 		writeValidation(writer, request, "分页游标无效", FieldError{Field: "cursor", Message: "游标无效、已损坏或不是本服务签发"})
 	default: