File size: 538 Bytes
27b774b 5e8a58c 27b774b 5e8a58c 27b774b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
# import libraries
from dotenv import find_dotenv, load_dotenv
# import functions
from src.test.eval_rag import evaluate_rag
from src.generation.generate_response import get_qa_chain_eval
def main():
# find .env automatically by walking up directories until it's found, then
# load up the .env entries as environment variables
load_dotenv(find_dotenv())
# get the qa chain
qa_chain = get_qa_chain_eval()
# evaluate the qa chain
evaluate_rag("qa_chain", qa_chain)
if __name__ == "__main__":
main() |