custom model in search

2024-11-06 13:22:23 -03:00 · 2024-11-06 13:22:23 -03:00 · 5aab6bdb69
commit 5aab6bdb69
parent 3ea4e41a78
1 changed files with 3 additions and 2 deletions
--- a/pages/3_🔍_Ask_and_Search.py
+++ b/pages/3_🔍_Ask_and_Search.py
@ -37,8 +37,9 @@ with ask_tab:
        messages = [question]
        rag_results = rag_graph.invoke(
            dict(
-                messages=messages
-            ),  # config=dict(configurable=dict(model_id=model.id))
+                messages=messages,
+            ),
+            config=dict(configurable=dict(model_id=model.id)),
        )
        st.markdown(convert_source_references(rag_results["messages"][-1].content))
        with st.expander("Details (for debugging)"):