Преглед на файлове

refactor: typo in dataset docstore (#1711)

tags/0.3.33
Pascal M преди 1 година
родител
ревизия
bc54cdc537
No account linked to committer's email address
променени са 2 файла, в които са добавени 4 реда и са изтрити 4 реда
  1. 2
    2
      api/core/docstore/dataset_docstore.py
  2. 2
    2
      api/core/indexing_runner.py

+ 2
- 2
api/core/docstore/dataset_docstore.py Целия файл

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,
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]:

+ 2
- 2
api/core/indexing_runner.py Целия файл



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
) )


# 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

Loading…
Отказ
Запис