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