Update agent.py

agent.py

@@ -20,7 +20,7 @@ from sentence_transformers import SentenceTransformer
 from langchain.embeddings.base import Embeddings
 from typing import List
 import numpy as np
-
+import yaml
 
 import pandas as pd
 import uuid