Add models for OpenAPI Response

Author: code-yeongyu

aishell/models/__init__.py

@@ -0,0 +1 @@
+from .open_ai_response_model import OpenAIResponseModel as OpenAIResponseModel

aishell/models/open_ai_response_model.py

@@ -0,0 +1,24 @@
+from typing import List, Optional
+
+from pydantic import BaseModel
+
+
+class OpenAIResponseModel(BaseModel):
+
+    class Choice(BaseModel):
+        finish_reason: str
+        index: int
+        logprobs: Optional[None]
+        text: Optional[str]
+
+    class Usage(BaseModel):
+        completion_tokens: int
+        prompt_tokens: int
+        total_tokens: int
+
+    choices: Optional[List[Choice]]
+    created: int
+    id: str
+    model: str
+    object: str
+    usage: Usage