Skip to content

Commit

Permalink
feat: Add document id and chunk ids to segments fixing order.
Browse files Browse the repository at this point in the history
  • Loading branch information
undo76 committed Nov 25, 2024
1 parent e85ed19 commit 65b8927
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
Expand Up @@ -233,13 +233,13 @@ def evaluate(
verbose=llm.verbose,
)
else:
lc_llm = ChatLiteLLM(model=config.llm)
lc_llm = ChatLiteLLM(model=config.llm) # 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( # ignore[call-arg]
model_path=embedder.model_path,
n_batch=embedder.n_batch,
n_ctx=embedder.n_ctx(),
Expand Down

0 comments on commit 65b8927

Please sign in to comment.