Update jsondocstatus_storage.py
Browse files
lightrag/storage/jsondocstatus_storage.py
CHANGED
@@ -48,28 +48,17 @@ Usage:
|
|
48 |
|
49 |
"""
|
50 |
|
51 |
-
|
52 |
-
import asyncio
|
53 |
-
import html
|
54 |
import os
|
55 |
-
from tqdm.asyncio import tqdm as tqdm_async
|
56 |
from dataclasses import dataclass
|
57 |
-
from typing import
|
58 |
-
import numpy as np
|
59 |
-
|
60 |
-
import time
|
61 |
|
62 |
from lightrag.utils import (
|
63 |
logger,
|
64 |
load_json,
|
65 |
write_json,
|
66 |
-
compute_mdhash_id,
|
67 |
)
|
68 |
|
69 |
from lightrag.base import (
|
70 |
-
BaseGraphStorage,
|
71 |
-
BaseKVStorage,
|
72 |
-
BaseVectorStorage,
|
73 |
DocStatus,
|
74 |
DocProcessingStatus,
|
75 |
DocStatusStorage,
|
|
|
48 |
|
49 |
"""
|
50 |
|
|
|
|
|
|
|
51 |
import os
|
|
|
52 |
from dataclasses import dataclass
|
53 |
+
from typing import Union, Dict
|
|
|
|
|
|
|
54 |
|
55 |
from lightrag.utils import (
|
56 |
logger,
|
57 |
load_json,
|
58 |
write_json,
|
|
|
59 |
)
|
60 |
|
61 |
from lightrag.base import (
|
|
|
|
|
|
|
62 |
DocStatus,
|
63 |
DocProcessingStatus,
|
64 |
DocStatusStorage,
|