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
|