move import

This commit is contained in:
Classic298 2025-08-12 22:06:10 +02:00 committed by GitHub
parent 2818b4643a
commit adda47ab04
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -4,6 +4,7 @@ import os
import shutil import shutil
import json import json
import re import re
import sqlite3
from typing import Optional, Set from typing import Optional, Set
from pathlib import Path from pathlib import Path
@ -320,7 +321,6 @@ def cleanup_orphaned_vector_collections(
uuid_to_collection = {} uuid_to_collection = {}
try: try:
import sqlite3
with sqlite3.connect(str(chroma_db_path)) as conn: with sqlite3.connect(str(chroma_db_path)) as conn:
collection_id_to_name = {} collection_id_to_name = {}
@ -660,7 +660,6 @@ async def prune_data(form_data: PruneDataForm, user=Depends(get_admin_user)):
chroma_db_path = Path(CACHE_DIR).parent / "vector_db" / "chroma.sqlite3" chroma_db_path = Path(CACHE_DIR).parent / "vector_db" / "chroma.sqlite3"
if chroma_db_path.exists(): if chroma_db_path.exists():
try: try:
import sqlite3
with sqlite3.connect(str(chroma_db_path)) as conn: with sqlite3.connect(str(chroma_db_path)) as conn:
conn.execute("VACUUM") conn.execute("VACUUM")