From 0e4310b56978be403a69ad09b35fbc53ed3c9be5 Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Mon, 24 Jun 2024 16:00:28 +0800 Subject: [PATCH] [dev] modify model response format --- memory_scope/models/response.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/memory_scope/models/response.py b/memory_scope/models/response.py index 34fed4df..9a506293 100644 --- a/memory_scope/models/response.py +++ b/memory_scope/models/response.py @@ -6,14 +6,14 @@ from memory_scope.enumeration.model_enum import ModelEnum class ModelResponse(BaseModel): - text: str = Field("", description="") + text: str = Field("", description="generation model result") - embedding_results: List[List[float]] | List[float] = Field([], description="embedding result") + embedding_results: List[List[float]] | List[float] = Field([], description="embedding vector") rank_scores: List[Dict[int, float]] = Field([], description="The rank scores of each documents. " "key: index, value: rank score") - model_type: ModelEnum = Field("", description="One of LLM, EMB, RANK.") + model_type: ModelEnum = Field(ModelEnum.GENERATION_MODEL, description="One of LLM, EMB, RANK.") status: bool = Field(True, description="Indicates whether the model call was successful.")