ruff format

02-household-queries/retrieval.py

@@ -7,6 +7,7 @@ def create_retriever(vectordb):
  retrieve_k = int(os.environ.get("RETRIEVE_K", "1"))
  return vectordb.as_retriever(search_kwargs={"k": retrieve_k})
 
+
 def retrieval_call(llm, vectordb, question):
  # Create the retrieval chain
  template = """
@@ -39,4 +40,3 @@ def retrieval_call(llm, vectordb, question):
  print(d)
  print()
  return response
-

02-household-queries/run.py

@@ -77,6 +77,7 @@ def load_training_json():
  # print(json.dumps(json_data, indent=2))
  return json_data
 
+
 def evaluate_retrieval():
  qa = load_training_json()
  results = []
@@ -89,13 +90,18 @@ def evaluate_retrieval():
  # print(f" Desired CARDS : {guru_cards}")
 
  retrieval = retriever.invoke(question)
- results.append({
- "question": question,
- "guru_cards": guru_cards,
- "retrieved_cards": [doc.metadata['source'] for doc in retrieval]
- })
+ results.append(
+ {
+ "id": qa_dict["id"],
+ "question": question,
+ "guru_cards": guru_cards,
+ "retrieved_cards": [doc.metadata["source"] for doc in retrieval],
+ }
+ )
  print(retriever)
- print("EVALUATION RESULTS:\n", "\n".join([json.dumps(r, indent=2) for r in results]))
+ print(
+ "EVALUATION RESULTS:\n", "\n".join([json.dumps(r, indent=2) for r in results])
+ )
 
 
 print("""