Spaces:
Build error
Build error
File size: 15,019 Bytes
51e9885 b3c91dd 51e9885 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 |
import os
import json
import time
import pandas as pd
import altair as alt
import streamlit as st
from typing import List
from langchain.vectorstores import Pinecone
from langchain.llms import Anthropic
from langchain.chat_models import ChatOpenAI
from langchain.evaluation.qa import QAEvalChain
from langchain.embeddings import HuggingFaceEmbeddings
from langchain.embeddings.openai import OpenAIEmbeddings
from langchain.chains.question_answering import load_qa_chain
from langchain.retrievers.self_query.base import SelfQueryRetriever
from kor_retriever_lex import kor_retriever
from langchain.docstore.document import Document
from langchain.vectorstores import Pinecone
from pinecone.client import Client as Pinecone
from self_query_retriever_lex import metadata_field_info, document_content_description
from prompts import GRADE_DOCS_PROMPT, GRADE_ANSWER_PROMPT, GRADE_ANSWER_PROMPT_FAST, GRADE_ANSWER_PROMPT_BIAS_CHECK, GRADE_ANSWER_PROMPT_OPENAI, QA_CHAIN_PROMPT_LEX, QA_CHAIN_PROMPT_TRAVEL
# Keep dataframe in memory to accumulate experimental results
if "existing_df" not in st.session_state:
summary = pd.DataFrame(columns=['model',
'retriever',
'embedding',
'num_neighbors',
'Latency',
'Retrieval score',
'Answer score'])
st.session_state.existing_df = summary
else:
summary = st.session_state.existing_df
@st.cache_resource
def make_llm(model_version: str):
"""
Make LLM from model version
@param model_version: model_version
@return: LLN
"""
if (model_version == "gpt-3.5-turbo") or (model_version == "gpt-4"):
chosen_model = ChatOpenAI(model_name=model_version, temperature=0)
elif model_version == "anthropic":
chosen_model = Anthropic(temperature=0)
else:
st.warning("Model version not recognized. Using gpt-3.5-turbo", icon="⚠")
chosen_model = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0)
return chosen_model
@st.cache_resource
def make_retriever(retriever_type,embedding_type,pc_api_key,pc_region,pc_index):
"""
Make document retriever
@param retriever_type: retriever type
@param embedding_type: embedding type
@param num_neighbors: number of neighbors for retrieval
@return: Pinecone
"""
st.info("Connecting to Pinecone ...")
# Retriver type
if retriever_type in ("Pinecone","Pinecone w/ metadata filtering"):
return p
elif retriever_type == "Pinecone w/ self-querying":
return SelfQueryRetriever.from_llm(ChatOpenAI(model_name="gpt-3.5-turbo",temperature=0), p, document_content_description, metadata_field_info, verbose=True, k=10)
elif retriever_type == "Kor filtering":
return kor_retriever
def make_chain(llm):
"""
Make retrieval chain
@param retriever: retriever
@param retriever_type: retriever type
@return: QA chain
"""
qa_chain = load_qa_chain(llm, chain_type="stuff", prompt=QA_CHAIN_PROMPT_LEX)
return qa_chain
def grade_model_answer(predicted_dataset: List, predictions: List, grade_answer_prompt: str) -> List:
"""
Grades the distilled answer based on ground truth and model predictions.
@param predicted_dataset: A list of dictionaries containing ground truth questions and answers.
@param predictions: A list of dictionaries containing model predictions for the questions.
@param grade_answer_prompt: The prompt level for the grading. Either "Fast" or "Full".
@return: A list of scores for the distilled answers.
"""
# Grade the distilled answer
st.info("Grading model answer ...")
# Set the grading prompt based on the grade_answer_prompt parameter
if grade_answer_prompt == "Fast":
prompt = GRADE_ANSWER_PROMPT_FAST
elif grade_answer_prompt == "Descriptive w/ bias check":
prompt = GRADE_ANSWER_PROMPT_BIAS_CHECK
elif grade_answer_prompt == "OpenAI grading prompt":
prompt = GRADE_ANSWER_PROMPT_OPENAI
else:
prompt = GRADE_ANSWER_PROMPT
# Create an evaluation chain
eval_chain = QAEvalChain.from_llm(
llm=ChatOpenAI(model_name="gpt-4", temperature=0),
prompt=prompt
)
# Evaluate the predictions and ground truth using the evaluation chain
graded_outputs = eval_chain.evaluate(
predicted_dataset,
predictions,
question_key="question",
prediction_key="result"
)
return graded_outputs
def grade_model_retrieval(gt_dataset: List, predictions: List, grade_docs_prompt: str):
"""
Grades the relevance of retrieved documents based on ground truth and model predictions.
@param gt_dataset: list of dictionaries containing ground truth questions and answers.
@param predictions: list of dictionaries containing model predictions for the questions
@param grade_docs_prompt: prompt level for the grading. Either "Fast" or "Full"
@return: list of scores for the retrieved documents.
"""
# Grade the docs retrieval
st.info("Grading relevance of retrieved docs ...")
# Set the grading prompt based on the grade_docs_prompt parameter
prompt = GRADE_DOCS_PROMPT
# Create an evaluation chain
eval_chain = QAEvalChain.from_llm(
llm=ChatOpenAI(model_name="gpt-4", temperature=0),
prompt=prompt
)
# Evaluate the predictions and ground truth using the evaluation chain
graded_outputs = eval_chain.evaluate(
gt_dataset,
predictions,
question_key="question",
prediction_key="result"
)
return graded_outputs
def run_evaluation(chain, retriever, eval_set, grade_prompt, retriever_type, num_neighbors):
"""
Runs evaluation on a model's performance on a given evaluation dataset.
@param chain: Model chain used for answering questions
@param retriever: Document retriever used for retrieving relevant documents
@param eval_set: List of dictionaries containing questions and corresponding ground truth answers
@param grade_prompt: String prompt used for grading model's performance
@param retriever_type: String specifying the type of retriever used
@param num_neighbors: Number of neighbors to retrieve using the retriever
@return: A tuple of four items:
- answers_grade: A dictionary containing scores for the model's answers.
- retrieval_grade: A dictionary containing scores for the model's document retrieval.
- latencies_list: A list of latencies in seconds for each question answered.
- predictions_list: A list of dictionaries containing the model's predicted answers and relevant documents for each question.
"""
st.info("Running evaluation ...")
predictions_list = []
retrieved_docs = []
gt_dataset = []
latencies_list = []
for data in eval_set:
# Get answer and log latency
start_time = time.time()
# Get docs
if retriever_type == "Pinecone w/ self-querying":
docs = retriever.get_relevant_documents(data["question"])
elif retriever_type == "Pinecone w/ metadata filtering":
### Set metadata here ###
metadata_filter = {'id':"0252"}
docs = retriever.similarity_search(query=data["question"],k=num_neighbors,filter=metadata_filter)
elif retriever_type == "Kor filtering":
docs = retriever(p,data["question"])
else:
docs = retriever.similarity_search(query=data["question"],k=num_neighbors)
print("--DOCS--")
if not docs:
docs=[Document(page_content="I was unable to recover any information about the question!")]
print(docs)
# Get answer
answer = chain.run(input_documents=docs,question=data["question"])
predictions_list.append({"question": data["question"], "answer": data["answer"], "result": answer})
gt_dataset.append(data)
end_time = time.time()
elapsed_time = end_time - start_time
latencies_list.append(elapsed_time)
# Get doc text
retrieved_doc_text = ""
for i, doc in enumerate(docs):
retrieved_doc_text += "Doc %s: " % str(i + 1) + doc.page_content + " "
retrieved = {"question": data["question"], "answer": data["answer"], "result": retrieved_doc_text}
retrieved_docs.append(retrieved)
# Grade docs and answer
answers_grade = grade_model_answer(gt_dataset, predictions_list, grade_prompt)
retrieval_grade = grade_model_retrieval(gt_dataset, retrieved_docs, grade_prompt)
return answers_grade, retrieval_grade, latencies_list, predictions_list
# Auth
st.sidebar.image("img/diagnostic.jpg")
with st.sidebar.form("user_input"):
# Pinecone params
oai_api_key = st.text_input("OpenAI API Key:", type="password").strip()
pc_api_key = st.text_input("Pinecone API Key:", type="password").strip()
pc_region = st.text_input("Pinecone region:", type="password").strip()
pc_index = st.text_input("Pinecone index:", type="password").strip()
retriever_type = st.radio("Choose retriever",
("Pinecone",
"Pinecone w/ self-querying",
"Pinecone w/ metadata filtering",
"Kor filtering"),
index=0)
num_neighbors = st.select_slider("Choose # chunks to retrieve",
options=[3, 4, 5, 6, 7, 8])
embeddings = st.radio("Choose embeddings",
("HuggingFace",
"OpenAI"),
index=1)
model = st.radio("Choose model",
("gpt-3.5-turbo",
"gpt-4"),
index=0)
grade_prompt = st.radio("Grading style prompt",
("Fast",
"Descriptive",
"Descriptive w/ bias check",
"OpenAI grading prompt"),
index=3)
submitted = st.form_submit_button("Submit evaluation")
# App
st.header("VectorDB auto-evaluator")
st.info(
"`I am an evaluation tool for question-answering using an existing vectorDB (currently Pinecone is supported) and an eval set. "
"I will generate and grade an answer to each eval set question with the user-specific retrival setting, such as metadata filtering or self-querying retrieval."
" Experiments with different configurations are logged. For an example eval set, see eval_sets/lex-pod-eval.json.`")
with st.form(key='file_inputs'):
uploaded_eval_set = st.file_uploader("Please upload eval set (.json): ",
type=['json'],
accept_multiple_files=False)
submitted = st.form_submit_button("Submit files")
# Build an index from the supplied docs
if uploaded_eval_set and pc_api_key and pc_region and pc_index:
# Set API key
os.environ["OPENAI_API_KEY"] = oai_api_key
# Set embeddings (must match your Pinecone DB)
if embeddings == "OpenAI":
embedding = OpenAIEmbeddings()
elif embeddings == "HuggingFace":
embedding = HuggingFaceEmbeddings()
# Set Pinecone
pinecone.init(api_key=str(pc_api_key), environment=str(pc_region))
p = Pinecone.from_existing_index(index_name=str(pc_index), embedding=embedding)
# Eval set
eval_set = json.loads(uploaded_eval_set.read())
# Make LLM
llm = make_llm(model)
# Make retriver
retriever = make_retriever(retriever_type,embeddings,pc_api_key,pc_region,pc_index)
# Make chain
qa_chain = make_chain(llm)
# Grade model
graded_answers, graded_retrieval, latency, predictions = run_evaluation(qa_chain, retriever, eval_set, grade_prompt,
retriever_type, num_neighbors)
# Assemble outputs
d = pd.DataFrame(predictions)
d['answer score'] = [g['text'] for g in graded_answers]
d['docs score'] = [g['text'] for g in graded_retrieval]
d['latency'] = latency
# Summary statistics
mean_latency = d['latency'].mean()
correct_answer_count = len([text for text in d['answer score'] if "Incorrect" not in text])
correct_docs_count = len([text for text in d['docs score'] if "Incorrect" not in text])
percentage_answer = (correct_answer_count / len(graded_answers)) * 100
percentage_docs = (correct_docs_count / len(graded_retrieval)) * 100
st.subheader("Run Results")
st.info(
"`I will grade the chain based on: 1/ the relevance of the retrived documents relative to the question and 2/ "
"the summarized answer relative to the ground truth answer. You can see (and change) to prompts used for "
"grading in text_utils`")
st.dataframe(data=d, use_container_width=True)
# Accumulate results
st.subheader("Aggregate Results")
st.info(
"`Retrieval and answer scores are percentage of retrived documents deemed relevant by the LLM grader ("
"relative to the question) and percentage of summarized answers deemed relevant (relative to ground truth "
"answer), respectively. The size of point correponds to the latency (in seconds) of retrieval + answer "
"summarization (larger circle = slower).`")
new_row = pd.DataFrame({'model': [model],
'retriever': [retriever_type],
'embedding': [embeddings],
'num_neighbors': [num_neighbors],
'Latency': [mean_latency],
'Retrieval score': [percentage_docs],
'Answer score': [percentage_answer]})
summary = pd.concat([summary, new_row], ignore_index=True)
st.dataframe(data=summary, use_container_width=True)
st.session_state.existing_df = summary
# Dataframe for visualization
show = summary.reset_index().copy()
show.columns = ['expt number', 'model', 'retriever', 'embedding', 'num_neighbors', 'Latency', 'Retrieval score','Answer score']
show['expt number'] = show['expt number'].apply(lambda x: "Expt #: " + str(x + 1))
c = alt.Chart(show).mark_circle().encode(x='Retrieval score',
y='Answer score',
size=alt.Size('Latency'),
color='expt number',
tooltip=['expt number', 'Retrieval score', 'Latency', 'Answer score'])
st.altair_chart(c, use_container_width=True, theme="streamlit")
else:
st.warning('Please specify a Pinecone index and add an eval set.', icon="⚠") |