Spaces:
Sleeping
Sleeping
Commit
·
1b1e5f4
1
Parent(s):
99e165e
Add schema for deepeval endpoint
Browse files- svc/schemas.py +6 -1
svc/schemas.py
CHANGED
@@ -27,8 +27,13 @@ class LMHarnessTaskRequest(BaseModel):
|
|
27 |
model_name: str
|
28 |
model_sha: str
|
29 |
|
|
|
|
|
|
|
|
|
|
|
30 |
|
31 |
-
class
|
32 |
results: Json # dict[Any,Any]
|
33 |
|
34 |
class LoadModelRequest(BaseModel):
|
|
|
27 |
model_name: str
|
28 |
model_sha: str
|
29 |
|
30 |
+
class DeepEvalSuiteRequest(BaseModel):
|
31 |
+
model_name: str
|
32 |
+
tasks: Optional[List[str]] = None
|
33 |
+
|
34 |
+
|
35 |
|
36 |
+
class TaskResponse(BaseModel):
|
37 |
results: Json # dict[Any,Any]
|
38 |
|
39 |
class LoadModelRequest(BaseModel):
|