SystemSh0cker/private-gpt
1
0
Fork 0
mirror of https://github.com/zylon-ai/private-gpt.git synced 2025-12-22 20:12:55 +01:00
Code Issues Projects Releases Packages Activity

Fix mypy

Browse source
This commit is contained in:
imartinez 2024-02-29 15:20:26 +01:00
parent 34d48d7b4d
commit 63d3b9f936
7 changed files with 39 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

8
private_gpt/components/embedding/embedding_component.py
View file

@ -20,7 +20,9 @@ class EmbeddingComponent:
match embedding_mode:
case "local":
try:
from llama_index.embeddings.huggingface import HuggingFaceEmbedding
from llama_index.embeddings.huggingface import ( # type: ignore
HuggingFaceEmbedding,
)
except ImportError as e:
raise ImportError(
"Local dependencies not found, install with `poetry install --extras local`"
@ -45,7 +47,9 @@ class EmbeddingComponent:
)
case "openai":
try:
from llama_index.embeddings.openai import OpenAIEmbedding
from llama_index.embeddings.openai import ( # type: ignore
OpenAIEmbedding,
)
except ImportError as e:
raise ImportError(
"OpenAI dependencies not found, install with `poetry install --extras openai`"