Fix mypy

private_gpt/components/llm/llm_component.py

@@ -113,7 +113,9 @@ class LLMComponent:
                 )
             case "tensorrt":
                 try:
-                    from llama_index.llms.nvidia_tensorrt import LocalTensorRTLLM  # type: ignore
+                    from llama_index.llms.nvidia_tensorrt import (  # type: ignore
+                        LocalTensorRTLLM,
+                    )
                 except ImportError as e:
                     raise ImportError(
                         "Nvidia TensorRTLLM dependencies not found, install with `poetry install --extras llms-nvidia-tensorrt`"

private_gpt/settings/settings.py

@@ -81,7 +81,9 @@ class DataSettings(BaseModel):
 
 
 class LLMSettings(BaseModel):
-    mode: Literal["llamacpp", "openai", "openailike", "sagemaker", "mock", "ollama", "tensorrt"]
+    mode: Literal[
+        "llamacpp", "openai", "openailike", "sagemaker", "mock", "ollama", "tensorrt"
+    ]
     max_new_tokens: int = Field(
         256,
         description="The maximum number of token that the LLM is authorized to generate in one completion.",