25 lines
409 B
Python
25 lines
409 B
Python
|
from typing import Any, List
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
|
||
|
|
class Choice(BaseModel):
|
||
|
|
text: str
|
||
|
|
index: int
|
||
|
|
logprobs: Any
|
||
|
|
finish_reason: str
|
||
|
|
|
||
|
|
|
||
|
|
class Usage(BaseModel):
|
||
|
|
prompt_tokens: int
|
||
|
|
completion_tokens: int
|
||
|
|
total_tokens: int
|
||
|
|
|
||
|
|
|
||
|
|
class ForeFrontResponse(BaseModel):
|
||
|
|
id: str
|
||
|
|
object: str
|
||
|
|
created: int
|
||
|
|
model: str
|
||
|
|
choices: List[Choice]
|
||
|
|
usage: Usage
|
||
|
|
text: str
|