remove embedding moderation

Author: Dev Agent

aigateway/component/moderation.go

@@ -26,7 +26,7 @@ const (
 	// cache ttl
 	cacheTTL = 24 * time.Hour
 	// moderation cache prefix
-	moderationCachePrefix = "moderation:"
+	moderationCachePrpmptPrefix = "moderation:prompt:"
 )
 
 type Moderation interface {
@@ -94,7 +94,7 @@ func splitContentIntoChunksByWindow(content string) []string {
 func (modImpl *moderationImpl) checkSingleChunk(ctx context.Context, content, key string) (*rpc.CheckResult, error) {
 	if modImpl.cacheClient != nil {
 		chunkHash := md5.Sum([]byte(content))
-		cacheKey := moderationCachePrefix + fmt.Sprintf("%x", chunkHash)
+		cacheKey := moderationCachePrpmptPrefix + fmt.Sprintf("%x", chunkHash)
 		cached, err := modImpl.cacheClient.Get(ctx, cacheKey)
 		if err == nil {
 			var result rpc.CheckResult
@@ -112,7 +112,7 @@ func (modImpl *moderationImpl) checkSingleChunk(ctx context.Context, content, ke
 
 	if modImpl.cacheClient != nil {
 		// Cache the result for the single chunk
-		cacheKey := moderationCachePrefix + content
+		cacheKey := moderationCachePrpmptPrefix + content
 		resultBytes, err := json.Marshal(result)
 		if err == nil {
 			err := modImpl.cacheClient.SetEx(ctx, cacheKey, string(resultBytes), cacheTTL)
@@ -143,7 +143,7 @@ func (modImpl *moderationImpl) checkBuffer(
 		// cache each chunk in the current buffer
 		for _, chunk := range currentBufferChunks {
 			chunkHash := md5.Sum([]byte(chunk))
-			cacheKey := moderationCachePrefix + fmt.Sprintf("%x", chunkHash)
+			cacheKey := moderationCachePrpmptPrefix + fmt.Sprintf("%x", chunkHash)
 			resultBytes, err := json.Marshal(result)
 			if err == nil {
 				err := modImpl.cacheClient.SetEx(ctx, cacheKey, string(resultBytes), cacheTTL)
@@ -236,7 +236,7 @@ func (modImpl *moderationImpl) checkLLMPrompt(ctx context.Context, content, key
 		// Check if chunk is in cache
 		if modImpl.cacheClient != nil {
 			chunkHash := md5.Sum([]byte(chunk))
-			cacheKey := moderationCachePrefix + fmt.Sprintf("%x", chunkHash)
+			cacheKey := moderationCachePrpmptPrefix + fmt.Sprintf("%x", chunkHash)
 			cached, err := modImpl.cacheClient.Get(ctx, cacheKey)
 			if err == nil {
 				var result rpc.CheckResult
@@ -265,7 +265,7 @@ func (modImpl *moderationImpl) checkLLMPrompt(ctx context.Context, content, key
 	for _, chunk := range unCheckedChunks {
 		if modImpl.cacheClient != nil {
 			chunkHash := md5.Sum([]byte(chunk))
-			cacheKey := moderationCachePrefix + fmt.Sprintf("%x", chunkHash)
+			cacheKey := moderationCachePrpmptPrefix + fmt.Sprintf("%x", chunkHash)
 			cached, err := modImpl.cacheClient.Get(ctx, cacheKey)
 			if err == nil {
 				var result rpc.CheckResult

aigateway/component/moderation_test.go

@@ -369,7 +369,7 @@ func TestModerationImpl_CheckChatNonStreamResponse(t *testing.T) {
 	})
 }
 
-// TestModerationImpl_CheckLLMPrompt_CacheCheck 测试 moderation.go 中第231-255行的缓存检查逻辑
+// TestModerationImpl_CheckLLMPrompt_CacheCheck tests the cache checking logic in moderation.go
 func TestModerationImpl_CheckLLMPrompt_CacheCheck(t *testing.T) {
 	ctx := context.Background()
 	key := "test-key"
@@ -388,7 +388,7 @@ func TestModerationImpl_CheckLLMPrompt_CacheCheck(t *testing.T) {
 		testContent := sensitiveChunk + ". " + safeContent
 
 		chunkHash := md5.Sum([]byte(sensitiveChunk))
-		cacheKey := moderationCachePrefix + fmt.Sprintf("%x", chunkHash)
+		cacheKey := moderationCachePrpmptPrefix + fmt.Sprintf("%x", chunkHash)
 
 		sensitiveResult := &rpc.CheckResult{IsSensitive: true, Reason: "contains inappropriate content"}
 		resultJSON, _ := json.Marshal(sensitiveResult)
@@ -425,8 +425,8 @@ func TestModerationImpl_CheckLLMPrompt_CacheCheck(t *testing.T) {
 		testContent := testChunk + ". " + strings.Repeat("y", slidingWindowSize*2)
 
 		chunkHash := md5.Sum([]byte(testChunk))
-		cacheKey1 := moderationCachePrefix + fmt.Sprintf("%x", chunkHash)
-		cacheKey2 := moderationCachePrefix + fmt.Sprintf("%x", md5.Sum([]byte(strings.Repeat("y", slidingWindowSize))))
+		cacheKey1 := moderationCachePrpmptPrefix + fmt.Sprintf("%x", chunkHash)
+		cacheKey2 := moderationCachePrpmptPrefix + fmt.Sprintf("%x", md5.Sum([]byte(strings.Repeat("y", slidingWindowSize))))
 
 		mockCacheClient.EXPECT().Get(mock.Anything, cacheKey1).Return("", errors.New("cache error"))
 

aigateway/handler/requests.go

@@ -70,7 +70,7 @@ type ChatMessageHistoryResponse struct {
 
 // EmbeddingRequest represents an embedding request structure
 type EmbeddingRequest struct {
-	Input          string `json:"input"`           // Input text content
-	Model          string `json:"model"`           // Model name used (e.g., "text-embedding-ada-002")
-	EncodingFormat string `json:"encoding_format"` // Encoding format (e.g., "float")
+	Input          string `json:"input"`                     // Input text content
+	Model          string `json:"model"`                     // Model name used (e.g., "text-embedding-ada-002")
+	EncodingFormat string `json:"encoding_format,omitempty"` // Encoding format (e.g., "float")
 }