Niansuh commited on
Commit
6c37ac6
·
verified ·
1 Parent(s): 3073084

Update api/models.py

Browse files
Files changed (1) hide show
  1. 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