Create api/schemas.py
Browse files- api/schemas.py +2 -1
api/schemas.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
from pydantic import BaseModel, Field
|
| 2 |
from typing import List, Optional
|
| 3 |
|
|
@@ -9,7 +10,7 @@ class AskResponse(BaseModel):
|
|
| 9 |
answer: Optional[str]
|
| 10 |
confidence: float
|
| 11 |
warning: Optional[str] = None
|
| 12 |
-
weights: Optional[
|
| 13 |
|
| 14 |
class HallucinationRequest(BaseModel):
|
| 15 |
prompt: str
|
|
|
|
| 1 |
+
# api/schemas.py
|
| 2 |
from pydantic import BaseModel, Field
|
| 3 |
from typing import List, Optional
|
| 4 |
|
|
|
|
| 10 |
answer: Optional[str]
|
| 11 |
confidence: float
|
| 12 |
warning: Optional[str] = None
|
| 13 |
+
weights: Optional[List[float]] = None
|
| 14 |
|
| 15 |
class HallucinationRequest(BaseModel):
|
| 16 |
prompt: str
|