cleanup

libs/langchain/langchain/chains/rl_chain/__init__.py

@@ -15,6 +15,7 @@ from langchain.chains.rl_chain.base import (
 from langchain.chains.rl_chain.pick_best_chain import (
     PickBest,
     PickBestEvent,
+    PickBestFeatureEmbedder,
     PickBestSelected,
 )
 
@@ -37,6 +38,7 @@ __all__ = [
     "PickBest",
     "PickBestEvent",
     "PickBestSelected",
+    "PickBestFeatureEmbedder",
     "Embed",
     "BasedOn",
     "ToSelectFrom",

libs/langchain/langchain/chains/rl_chain/pick_best_chain.py

@@ -60,9 +60,7 @@ class PickBestFeatureEmbedder(base.Embedder[PickBestEvent]):
 
         if model is None:
             from sentence_transformers import SentenceTransformer
-
             model = SentenceTransformer("all-mpnet-base-v2")
-            # model = SentenceTransformer("all-MiniLM-L6-v2")
 
         self.model = model
         self.auto_embed = auto_embed