github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-07-07 04:18:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

added # type: ignore for embeddings.fireworks

Browse Source
This commit is contained in:
Somashekar B R 2024-09-25 01:18:28 +05:30
parent b8cb49a359
commit 2052ff49b8
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
private_gpt/components/embedding/embedding_component.py
View File

@ -69,7 +69,9 @@ class EmbeddingComponent:
) )
case "fireworks": case "fireworks":
try: try:
from llama_index.embeddings.fireworks import FireworksEmbedding from llama_index.embeddings.fireworks import ( # type: ignore
FireworksEmbedding,
)
except ImportError as e: except ImportError as e:
raise ImportError( raise ImportError(
"FireworksEmbedding dependencies not found, install with `poetry install --extras embeddings-fireworks`" "FireworksEmbedding dependencies not found, install with `poetry install --extras embeddings-fireworks`"

3
pyproject.toml
View File

@ -203,9 +203,6 @@ exclude = ["tests"]
[tool.mypy-llama-index] [tool.mypy-llama-index]
ignore_missing_imports = true ignore_missing_imports = true
[tool.mypy-llama_index.embeddings.fireworks]
ignore_missing_imports = true
[tool.pytest.ini_options] [tool.pytest.ini_options]
asyncio_mode = "auto" asyncio_mode = "auto"
testpaths = ["tests"] testpaths = ["tests"]