mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-08-28 05:25:04 +00:00
20 lines
509 B
Python
20 lines
509 B
Python
from typing import Generator, List, Dict
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class ModelResponse(BaseModel):
|
|
text: str = Field("", description="")
|
|
|
|
embedding_results: Dict[int, List[float]] | List[float] = Field([], description="")
|
|
|
|
rank_scores: Dict[int, float] = Field({}, description="")
|
|
|
|
model_type: str = Field("", description="")
|
|
|
|
status: bool = Field(True, description="")
|
|
|
|
details: str = Field("", description="")
|
|
|
|
|
|
ModelResponseGen = Generator[ModelResponse, None, None]
|