scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-07 08:26:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

[dev] modify model response format

Browse source
This commit is contained in:
jinli.yl 2024-06-24 16:00:28 +08:00
parent 31316a9080
commit 0e4310b569
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
memory_scope/models/response.py
View file

@ -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.")