Fix imports

motion/operations/equijoin.py

@@ -8,13 +8,12 @@
 from typing import Any, Dict, List, Tuple
 
 from jinja2 import Template
-from litellm import completion_cost
+from litellm import completion_cost, embedding
 from sklearn.metrics.pairwise import cosine_similarity
 
 from motion.operations.base import BaseOperation
 from motion.operations.utils import (
     call_llm,
-    embedding,
     parse_llm_response,
     rich_as_completed,
     validate_output,