mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-07-12 14:12:53 +02:00
Bugfix Vespa Deletion Script (#3998)
This commit is contained in:
@ -772,6 +772,7 @@ def main() -> None:
|
||||
"update",
|
||||
"delete",
|
||||
"get_acls",
|
||||
"delete-all-documents",
|
||||
],
|
||||
required=True,
|
||||
help="Action to perform",
|
||||
@ -786,12 +787,19 @@ def main() -> None:
|
||||
parser.add_argument(
|
||||
"--fields", help="Fields to update, in JSON format (for update)"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--count",
|
||||
type=int,
|
||||
help="Maximum number of documents to delete (for delete-all-documents)",
|
||||
)
|
||||
|
||||
args = parser.parse_args()
|
||||
vespa_debug = VespaDebugging(args.tenant_id)
|
||||
|
||||
if args.action == "delete-all-documents":
|
||||
vespa_debug.delete_documents_for_tenant(args.count)
|
||||
if not args.tenant_id:
|
||||
parser.error("--tenant-id is required for delete-all-documents action")
|
||||
vespa_debug.delete_documents_for_tenant(count=args.count)
|
||||
elif args.action == "config":
|
||||
vespa_debug.print_config()
|
||||
elif args.action == "connect":
|
||||
|
Reference in New Issue
Block a user