ecemumutlu commited on
Commit
1b1e5f4
·
1 Parent(s): 99e165e

Add schema for deepeval endpoint

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