Update tool_handler.py

tool_handler.py

@@ -2,11 +2,7 @@ import sys
 import os
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), 'pyspur/backend/')))
 
-from embedding import (
-    get_embedding,
-    discharge_collection_persistent,
-    trials_collection_persistent,
-)
+from embedding import discharge_collection, trials_collection, get_embedding
 
 from groq import Client
 from serpapi import GoogleSearch