diff --git a/private_gpt/components/embedding/embedding_component.py b/private_gpt/components/embedding/embedding_component.py index 4f0ae93b2b..945ba1f904 100644 --- a/private_gpt/components/embedding/embedding_component.py +++ b/private_gpt/components/embedding/embedding_component.py @@ -69,7 +69,9 @@ def __init__(self, settings: Settings) -> None: ) case "fireworks": try: - from llama_index.embeddings.fireworks import FireworksEmbedding + from llama_index.embeddings.fireworks import ( # type: ignore + FireworksEmbedding, + ) except ImportError as e: raise ImportError( "FireworksEmbedding dependencies not found, install with `poetry install --extras embeddings-fireworks`" diff --git a/pyproject.toml b/pyproject.toml index bfa1370596..3651845235 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -203,9 +203,6 @@ exclude = ["tests"] [tool.mypy-llama-index] ignore_missing_imports = true -[tool.mypy-llama_index.embeddings.fireworks] -ignore_missing_imports = true - [tool.pytest.ini_options] asyncio_mode = "auto" testpaths = ["tests"]