AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 07:35:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: typo in dataset docstore (#1711)

Browse Source
This commit is contained in:
Pascal M 2023-12-07 02:24:52 +01:00 committed by GitHub
parent 5d10cf0fe6
commit bc54cdc537
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/docstore/dataset_docstore.py
View File

@ -8,7 +8,7 @@ from extensions.ext_database import db
from models.dataset import Dataset, DocumentSegment from models.dataset import Dataset, DocumentSegment
class DatesetDocumentStore: class DatasetDocumentStore:
def __init__( def __init__(
self, self,
dataset: Dataset, dataset: Dataset,
@ -20,7 +20,7 @@ class DatesetDocumentStore:
self._document_id = document_id self._document_id = document_id
@classmethod @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) return cls(**config_dict)
def to_dict(self) -> Dict[str, Any]: def to_dict(self) -> Dict[str, Any]:

4
api/core/indexing_runner.py
View File

@ -15,7 +15,7 @@ from sqlalchemy.orm.exc import ObjectDeletedError
from core.data_loader.file_extractor import FileExtractor from core.data_loader.file_extractor import FileExtractor
from core.data_loader.loader.notion import NotionLoader 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.generator.llm_generator import LLMGenerator
from core.index.index import IndexBuilder from core.index.index import IndexBuilder
from core.model_providers.error import ProviderTokenNotInitError from core.model_providers.error import ProviderTokenNotInitError
@ -475,7 +475,7 @@ class IndexingRunner:
) )
# save node to document segment # save node to document segment
doc_store = DatesetDocumentStore( doc_store = DatasetDocumentStore(
dataset=dataset, dataset=dataset,
user_id=dataset_document.created_by, user_id=dataset_document.created_by,
document_id=dataset_document.id document_id=dataset_document.id