Skip to content

Commit 2c8ba8c

Browse files
committed
feat (retrievers): Implements #12 Pre-select checkpoint in retriever's parameters
1 parent 6d9aeb2 commit 2c8ba8c

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

src/applications/qa/index.js

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -274,7 +274,10 @@ function QuestionAnswering({ application, showSettings }) {
274274

275275
{showSettings ? (
276276
<div className="application__settings">
277-
<Retrievers disableParent={setDisabled}></Retrievers>
277+
<Retrievers
278+
disableParent={setDisabled}
279+
defaultCheckpoint={selectedCollection?.checkpoint}
280+
></Retrievers>
278281
<Collections
279282
retriever={retrievers.selectedRetriever}
280283
selectedCollection={selectedCollection}

src/components/retrievers/index.js

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ import { addNotification } from "../notifications/notificationsSlice";
4141

4242
import "./styles.scss";
4343

44-
function Retrievers({ disableParent }) {
44+
function Retrievers({ disableParent, defaultCheckpoint }) {
4545
// Redux connectivity
4646
const retrievers = useSelector((state) => state.retrievers);
4747
const dispatch = useDispatch();
@@ -313,7 +313,7 @@ function Retrievers({ disableParent }) {
313313
<Select
314314
id={"parameter-" + parameter.parameter_id}
315315
labelText={parameter.name}
316-
value={parameter.value || "placeholder-item"}
316+
value={parameter.value || defaultCheckpoint || "placeholder-item"}
317317
onChange={(event) => {
318318
dispatch(
319319
updateParameterValue({
@@ -374,6 +374,7 @@ function Retrievers({ disableParent }) {
374374

375375
Retrievers.propTypes = {
376376
disableParent: PropTypes.func,
377+
defaultCheckpoint: PropTypes.string,
377378
};
378379

379380
export default Retrievers;

0 commit comments

Comments
 (0)