scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-09 22:31:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

[dev] modify model response delta

Browse source
This commit is contained in:
jinli.yl 2024-06-24 18:24:40 +08:00
parent ffc1c5261d
commit a1c67d9e40
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
memory_scope/models/response.py
View file

@ -8,6 +8,8 @@ from memory_scope.enumeration.model_enum import ModelEnum
class ModelResponse(BaseModel):
text: str = Field("", description="generation model result")
delta: str = Field("", description="New text that just streamed in (only used when streaming)")
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. "
@ -22,6 +24,5 @@ class ModelResponse(BaseModel):
raw: Any = Field("", description="Raw response from model call")
delta: str = Field("", description="New text that just streamed in (only used when streaming)")
ModelResponseGen = Generator[ModelResponse, None, None]