diff --git a/private_gpt/components/llm/llm_component.py b/private_gpt/components/llm/llm_component.py index 9b9c46ce9..bea2637c3 100644 --- a/private_gpt/components/llm/llm_component.py +++ b/private_gpt/components/llm/llm_component.py @@ -108,7 +108,9 @@ def __init__(self, settings: Settings) -> None: ) case "ollama": try: - from private_gpt.components.llm.custom.ollama import CustomOllama # type: ignore + from private_gpt.components.llm.custom.ollama import ( + CustomOllama, # type: ignore + ) except ImportError as e: raise ImportError( "Ollama dependencies not found, install with `poetry install --extras llms-ollama`"