summaryrefslogtreecommitdiffstats
path: root/openai_rev/forefront/models.py
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--openai_rev/forefront/models.py26
1 files changed, 26 insertions, 0 deletions
diff --git a/openai_rev/forefront/models.py b/openai_rev/forefront/models.py
new file mode 100644
index 00000000..23e90903
--- /dev/null
+++ b/openai_rev/forefront/models.py
@@ -0,0 +1,26 @@
+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