🐛 Bug: Fix the bug of undefined review content.

main.py

@@ -470,14 +470,24 @@ class StatsMiddleware(BaseHTTPMiddleware):
                 model = request_model.model
                 current_info["model"] = model
 
+                moderated_content = None
                 if request_model.request_type == "chat":
                     moderated_content = request_model.get_last_text_message()
                 elif request_model.request_type == "image":
                     moderated_content = request_model.prompt
+                elif request_model.request_type == "moderation":
+                    pass
+                elif request_model.request_type == "embedding":
+                    if isinstance(request_model.input, list) and len(request_model.input) > 0 and isinstance(request_model.input[0], str):
+                        moderated_content = "\n".join(request_model.input)
+                    else:
+                        moderated_content = request_model.input
+                else:
+                    logger.error(f"Unknown request type: {request_model.request_type}")
+
                 if moderated_content:
                     current_info["text"] = moderated_content
 
-
                 if enable_moderation and moderated_content:
                     moderation_response = await self.moderate_content(moderated_content, api_index)
                     is_flagged = moderation_response.get('results', [{}])[0].get('flagged', False)

models.py

@@ -115,6 +115,8 @@ class EmbeddingRequest(BaseRequest):
     input: Union[str, List[Union[str, int, List[int]]]]  # 支持字符串或数组
     model: str
     encoding_format: Optional[str] = "float"
+    dimensions: Optional[int] = None
+    user: Optional[str] = None
     stream: bool = False
 
 class AudioTranscriptionRequest(BaseRequest):
@@ -130,7 +132,7 @@ class AudioTranscriptionRequest(BaseRequest):
         arbitrary_types_allowed = True
 
 class ModerationRequest(BaseRequest):
-    input: str
+    input: Union[str, List[str]]
     model: Optional[str] = "text-moderation-latest"
     stream: bool = False
 
@@ -150,12 +152,12 @@ class UnifiedRequest(BaseModel):
             elif "file" in values:
                 values["data"] = AudioTranscriptionRequest(**values)
                 values["data"].request_type = "audio"
+            elif "text-embedding" in values.get("model", ""):
+                values["data"] = EmbeddingRequest(**values)
+                values["data"].request_type = "embedding"
             elif "input" in values:
                 values["data"] = ModerationRequest(**values)
                 values["data"].request_type = "moderation"
-            elif "input" in values:
-                values["data"] = EmbeddingRequest(**values)
-                values["data"].request_type = "embedding"
             else:
                 raise ValueError("无法确定请求类型")
         return values