Update api/models.py

api/models.py

@@ -1,34 +1,14 @@
-# api/models.py
-
-from typing import List, Optional, Union
+from typing import List, Optional
 from pydantic import BaseModel
 
 class Message(BaseModel):
     role: str
-    content: Union[str, List[dict]]
+    content: str | list
 
 class ChatRequest(BaseModel):
     model: str
     messages: List[Message]
-    proxy: Optional[str] = None  # Assuming proxy might be needed
     stream: Optional[bool] = False
     temperature: Optional[float] = 0.7
     top_p: Optional[float] = 0.9
     max_tokens: Optional[int] = 99999999
-
-class ImageResponseModel(BaseModel):
-    images: List[str]
-    alt: str
-
-class ChatCompletionChoice(BaseModel):
-    index: int
-    message: dict
-    finish_reason: Optional[str]
-
-class ChatCompletionResponse(BaseModel):
-    id: str
-    object: str
-    created: int
-    model: str
-    choices: List[ChatCompletionChoice]
-    usage: Optional[dict] = None