Skip to content

Commit 979200f

Browse files
committed
Merge remote-tracking branch 'origin/iris/chore/unify-chat-pipelines' into iris/chore/unify-chat-pipelines
2 parents 569cf9c + 9160caa commit 979200f

1 file changed

Lines changed: 10 additions & 3 deletions

File tree

iris/src/iris/pipeline/autonomous_tutor_pipeline.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,10 @@ def get_tools(
127127
query_text = self._generate_retrieval_query_text(discussion)
128128

129129
if allow_lecture_tool:
130-
self.lecture_retriever = LectureRetrieval(state.db.client)
130+
self.lecture_retriever = LectureRetrieval(
131+
state.db.client,
132+
local=state.dto.settings is not None and state.dto.settings.is_local(),
133+
)
131134
tool_list.append(
132135
create_tool_lecture_content_retrieval(
133136
self.lecture_retriever,
@@ -141,7 +144,10 @@ def get_tools(
141144
)
142145

143146
if allow_faq_tool:
144-
self.faq_retriever = FaqRetrieval(state.db.client)
147+
self.faq_retriever = FaqRetrieval(
148+
state.db.client,
149+
local=state.dto.settings is not None and state.dto.settings.is_local(),
150+
)
145151
tool_list.append(
146152
create_tool_faq_content_retrieval(
147153
self.faq_retriever,
@@ -301,7 +307,8 @@ def __call__(
301307
"""Run the autonomous tutor pipeline."""
302308
try:
303309
logger.info("Running autonomous tutor pipeline...")
304-
super().__call__(dto, variant, callback)
310+
local = dto.settings is not None and dto.settings.is_local()
311+
super().__call__(dto, variant, callback, local=local)
305312
except Exception as e:
306313
logger.error(
307314
"An error occurred while running the autonomous tutor pipeline",

0 commit comments

Comments
 (0)