fix deprecated refs

This commit is contained in:
Adam Wilson
2025-05-15 10:10:41 -06:00
parent fab3633ec6
commit 52819e34e0
2 changed files with 5 additions and 3 deletions
@@ -32,5 +32,7 @@ jobs:
pip install accelerate
pip install optimum[exporters,onnxruntime]
pip install faiss-cpu
pip install hf_xet
python -m tests.llm.llm_rag
+3 -3
View File
@@ -7,8 +7,8 @@ from typing import List
import numpy as np
# LangChain imports
from langchain_community.llms import HuggingFacePipeline
from langchain_community.embeddings import HuggingFaceEmbeddings
from langchain_huggingface import HuggingFacePipeline
from langchain_huggingface import HuggingFaceEmbeddings
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain_community.vectorstores import FAISS
from langchain.chains import RetrievalQA
@@ -160,7 +160,7 @@ def ask_rag(question: str):
print(f"\nQuestion: {question}")
# Get response from the chain
response = qa_chain({"query": question})
response = qa_chain.invoke({"query": question})
# Print the answer
print("\nAnswer:")