Update tools.py

tools.py

@@ -147,8 +147,12 @@ def build_graph(csv_file_path: str = "embeddings.csv"):
     llm_with_tools = chat_model.bind_tools(tools=tools)
 
     from sentence_transformers import SentenceTransformer
+    
+    model = SentenceTransformer("all-mpnet-base-v2")  # 768-dim
+    
+    query_embedding = model.encode(query_text, convert_to_tensor=True)
+    doc_embedding = model.encode(doc_text, convert_to_tensor=True)
 
-    embedding_model = SentenceTransformer("all-MiniLM-L6-v2")
     
     def get_query_embedding(query: str) -> np.ndarray:
         return embedding_model.encode(query)