Merge pull request #5 from athina-ai/dev

minor fix
athina-ai · Feb 23, 2024 · 9c3762c · 9c3762c
2 parents 8046777 + 52a73a7
commit 9c3762c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/evaluations/run_athina_evals.py b/evaluations/run_athina_evals.py
@@ -22,6 +22,8 @@ def load_data():
             "expected_response": "Y Combinator takes a 7% equity stake in companies in return for $125,000 on a post-money SAFE, and a 1.5% equity stake in companies participating in the YC Fellowship Program in exchange for a $20,000 investment.",
         }
     ]
+    for item in raw_data:
+        item['context'], item['response'] = app.generate_response(item['query'])
 
     # # or read from file
     # with open('evaluations/golden_dataset.jsonl', 'r') as file: