File size: 775 Bytes
3943768 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from pydantic import BaseModel
class ReturnType(BaseModel):
reply: str | list[str] | None
reply_final: str | list[str] | None = None
prompt_raw: str | None = None
actual_llm: str | None = None
text_context_list: list[str] | None = []
input_tokens: int = 0
output_tokens: int = 0
tokens_per_second: float = 0.0
time_to_first_token: float = 0.0
trial: int = 0
vision_visible_model: str | None = None
vision_batch_input_tokens: int = 0
vision_batch_output_tokens: int = 0
vision_batch_tokens_per_second: float = 0.0
files: list[str] | list[dict[str, str]] | None = []
files_pdf: list[str] | list[dict[str, str]] | None = []
chat_history: list[dict[str, str]] | None = []
chat_history_md: str | None = ""
|