Unverified Commit bc54cdc5 authored by Pascal M's avatar Pascal M Committed by GitHub

refactor: typo in dataset docstore (#1711)

parent 5d10cf0f
......@@ -8,7 +8,7 @@ from extensions.ext_database import db
from models.dataset import Dataset, DocumentSegment
class DatesetDocumentStore:
class DatasetDocumentStore:
def __init__(
self,
dataset: Dataset,
......@@ -20,7 +20,7 @@ class DatesetDocumentStore:
self._document_id = document_id
@classmethod
def from_dict(cls, config_dict: Dict[str, Any]) -> "DatesetDocumentStore":
def from_dict(cls, config_dict: Dict[str, Any]) -> "DatasetDocumentStore":
return cls(**config_dict)
def to_dict(self) -> Dict[str, Any]:
......
......@@ -15,7 +15,7 @@ from sqlalchemy.orm.exc import ObjectDeletedError
from core.data_loader.file_extractor import FileExtractor
from core.data_loader.loader.notion import NotionLoader
from core.docstore.dataset_docstore import DatesetDocumentStore
from core.docstore.dataset_docstore import DatasetDocumentStore
from core.generator.llm_generator import LLMGenerator
from core.index.index import IndexBuilder
from core.model_providers.error import ProviderTokenNotInitError
......@@ -475,7 +475,7 @@ class IndexingRunner:
)
# save node to document segment
doc_store = DatesetDocumentStore(
doc_store = DatasetDocumentStore(
dataset=dataset,
user_id=dataset_document.created_by,
document_id=dataset_document.id
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment