Update api/models.py
Browse files- api/models.py +2 -2
api/models.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
# api/models.py
|
| 2 |
|
| 3 |
-
from typing import List, Optional, Union, Dict
|
| 4 |
from pydantic import BaseModel
|
| 5 |
|
| 6 |
class ImageContent(BaseModel):
|
|
@@ -8,7 +8,7 @@ class ImageContent(BaseModel):
|
|
| 8 |
|
| 9 |
class Message(BaseModel):
|
| 10 |
role: str
|
| 11 |
-
content: Union[str, List[Dict[str, Any]]]
|
| 12 |
|
| 13 |
class ChatRequest(BaseModel):
|
| 14 |
model: str
|
|
|
|
| 1 |
# api/models.py
|
| 2 |
|
| 3 |
+
from typing import List, Optional, Union, Dict, Any # Ensure 'Any' is imported
|
| 4 |
from pydantic import BaseModel
|
| 5 |
|
| 6 |
class ImageContent(BaseModel):
|
|
|
|
| 8 |
|
| 9 |
class Message(BaseModel):
|
| 10 |
role: str
|
| 11 |
+
content: Union[str, List[Dict[str, Any]]] # 'Any' is now defined
|
| 12 |
|
| 13 |
class ChatRequest(BaseModel):
|
| 14 |
model: str
|