Update error message and change argument parsing

neuml · Dec 9, 2023 · a515d30 · a515d30
1 parent 3dab4d8
commit a515d30
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/python/txtai/pipeline/llm/litellm.py b/src/python/txtai/pipeline/llm/litellm.py
@@ -46,7 +46,7 @@ def __init__(self, path, template=None, **kwargs):
         super().__init__(path, template, **kwargs)
 
         if not LITELLM:
-            raise ImportError('LiteLLM is not available - install "llm" extra to enable')
+            raise ImportError('LiteLLM is not available - install "pipeline" extra to enable')
 
         # Register prompt template
         self.register(path)

diff --git a/src/python/txtai/pipeline/llm/llama.py b/src/python/txtai/pipeline/llm/llama.py
@@ -40,13 +40,13 @@ def __init__(self, path, template=None, **kwargs):
         super().__init__(path, template, **kwargs)
 
         if not LLAMA_CPP:
-            raise ImportError('llama.cpp is not available - install "llm" extra to enable')
+            raise ImportError('llama.cpp is not available - install "pipeline" extra to enable')
 
         # Check if this is a local path, otherwise download from the HF Hub
         path = path if os.path.exists(path) else self.download(path)
 
         # Create llama.cpp instance
-        self.llm = Llama(path, verbose=kwargs.get("verbose", False), **kwargs)
+        self.llm = Llama(path, verbose=kwargs.pop("verbose", False), **kwargs)
 
     def execute(self, texts, maxlength, **kwargs):
         results = []