Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix: type validation.
Browse files Browse the repository at this point in the history
undo76 committed Dec 2, 2024
1 parent 0c247ea commit 851311c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/raglite/_eval.py
Original file line number Diff line number Diff line change
@@ -234,13 +234,13 @@ def evaluate(
verbose=llm.verbose,
)
else:
lc_llm = ChatLiteLLM(model=config.llm)
lc_llm = ChatLiteLLM(model=config.llm) # type: ignore[call-arg]
# Load the embedder.
if not config.embedder.startswith("llama-cpp-python"):
error_message = "Currently, only `llama-cpp-python` embedders are supported."
raise NotImplementedError(error_message)
embedder = LlamaCppPythonLLM().llm(model=config.embedder, embedding=True)
lc_embedder = LlamaCppEmbeddings(
lc_embedder = LlamaCppEmbeddings( # type: ignore[call-arg]
model_path=embedder.model_path,
n_batch=embedder.n_batch,
n_ctx=embedder.n_ctx(),

0 comments on commit 851311c

Please sign in to comment.