feat(add perform_multi_step_scan):

This commit is contained in:
Alexander Myasoedov
2024-11-29 23:26:44 +02:00
parent 03dcf8c644
commit fa27f8e70e
2 changed files with 269 additions and 105 deletions
+13 -1
View File
@@ -23,11 +23,23 @@ class Scan(BaseModel):
class ScanResult(BaseModel):
module: str
tokens: int
tokens: float | int
cost: float
progress: float
status: bool = False
failureRate: float = 0.0
@classmethod
def status_msg(cls, msg: str) -> str:
return cls(
module=msg,
tokens=0,
cost=0,
progress=0,
failureRate=0,
status=True,
).model_dump_json()
class Probe(BaseModel):
prompt: str