summaryrefslogblamecommitdiffstats
path: root/openai_rev/forefront/models.py
blob: 23e9090313e4924d565aae1c583e20ef857c0252 (plain) (tree)

























                                   
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