Skip to content

Commit f3eed44

Browse files
authored
two minor ragulate query fixes (#581)
1 parent dc1d8d2 commit f3eed44

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

libs/ragulate/ragstack_ragulate/pipelines/base_pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def get_ingredients(
3535
"""Return ingredients for the given method params."""
3636
ingredients = {}
3737
for method_param in method_params:
38-
if method_param in reserved_params or method_param in ["kwargs"]:
38+
if method_param in reserved_params or method_param in ["kwargs", "_"]:
3939
continue
4040
if method_param not in passed_ingredients:
4141
raise ValueError(

libs/ragulate/ragstack_ragulate/pipelines/query_pipeline.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
# ruff: noqa: T201
12
import random
23
import signal
34
import time
@@ -192,6 +193,8 @@ def query(self):
192193
pipeline = query_method(**self.ingredients)
193194
llm_provider = self.get_provider()
194195

196+
print(f"Using provider: {llm_provider} for evaluation.")
197+
195198
feedbacks = Feedbacks(llm_provider=llm_provider, pipeline=pipeline)
196199

197200
self.start_evaluation()
@@ -240,7 +243,7 @@ def query(self):
240243
err = f"Query: '{query}' caused exception, skipping."
241244
logger.exception(err)
242245
# TODO: figure out why the logger isn't working after tru-lens starts. For now use print(). # noqa: E501
243-
print(f"{err} Exception {e}") # noqa: T201
246+
print(f"{err} Exception {e}")
244247
finally:
245248
self.update_progress(query_change=1)
246249

0 commit comments

Comments
 (0)