Skip to content

Commit

Permalink
fix linting
Browse files Browse the repository at this point in the history
  • Loading branch information
anakin87 committed Nov 22, 2024
1 parent 2f63292 commit 9408bf2
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from .document_embedder import OllamaDocumentEmbedder
from .text_embedder import OllamaTextEmbedder

__all__ = ["OllamaTextEmbedder", "OllamaDocumentEmbedder"]
__all__ = ["OllamaDocumentEmbedder", "OllamaTextEmbedder"]
Original file line number Diff line number Diff line change
Expand Up @@ -122,7 +122,7 @@ def run(self, documents: List[Document], generation_kwargs: Optional[Dict[str, A
- `documents`: Documents with embedding information attached
- `meta`: The metadata collected during the embedding process
"""
if not isinstance(documents, list) or documents and not isinstance(documents[0], Document):
if not isinstance(documents, list) or (documents and not isinstance(documents[0], Document)):
msg = (
"OllamaDocumentEmbedder expects a list of Documents as input."
"In case you want to embed a list of strings, please use the OllamaTextEmbedder."
Expand Down
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from .chat.chat_generator import OllamaChatGenerator
from .generator import OllamaGenerator

__all__ = ["OllamaGenerator", "OllamaChatGenerator"]
__all__ = ["OllamaChatGenerator", "OllamaGenerator"]

0 comments on commit 9408bf2

Please sign in to comment.