Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(wren-ai-service): change sql through reasoning #1273

Merged
merged 19 commits into from
Feb 12, 2025
8 changes: 3 additions & 5 deletions deployment/kustomizations/base/cm.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -137,11 +137,6 @@ data:
- name: sql_expansion
llm: litellm_llm.gpt-4o-mini-2024-07-18
engine: wren_ui
- name: sql_explanation
llm: litellm_llm.gpt-4o-mini-2024-07-18
- name: sql_regeneration
llm: litellm_llm.gpt-4o-mini-2024-07-18
engine: wren_ui
- name: semantics_description
llm: litellm_llm.gpt-4o-mini-2024-07-18
- name: relationship_recommendation
Expand Down Expand Up @@ -181,6 +176,9 @@ data:
llm: litellm_llm.gpt-4o-mini-2024-07-18
- name: sql_generation_reasoning
llm: litellm_llm.gpt-4o-mini-2024-07-18
- name: sql_regeneration
llm: litellm_llm.gpt-4o-mini-2024-07-18
engine: wren_ui

---
settings:
Expand Down
8 changes: 3 additions & 5 deletions docker/config.example.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -89,11 +89,6 @@ pipes:
- name: sql_expansion
llm: litellm_llm.gpt-4o-mini-2024-07-18
engine: wren_ui
- name: sql_explanation
llm: litellm_llm.gpt-4o-mini-2024-07-18
- name: sql_regeneration
llm: litellm_llm.gpt-4o-mini-2024-07-18
engine: wren_ui
- name: semantics_description
llm: litellm_llm.gpt-4o-mini-2024-07-18
- name: relationship_recommendation
Expand Down Expand Up @@ -133,6 +128,9 @@ pipes:
llm: litellm_llm.gpt-4o-mini-2024-07-18
- name: sql_generation_reasoning
llm: litellm_llm.gpt-4o-mini-2024-07-18
- name: sql_regeneration
llm: litellm_llm.gpt-4o-mini-2024-07-18
engine: wren_ui

---
settings:
Expand Down
18 changes: 2 additions & 16 deletions wren-ai-service/demo/app.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,24 +52,14 @@
st.session_state["preview_sql"] = None
if "query_history" not in st.session_state:
st.session_state["query_history"] = None
if "sql_explanation_question" not in st.session_state:
st.session_state["sql_explanation_question"] = None
if "sql_explanation_steps_with_analysis" not in st.session_state:
st.session_state["sql_explanation_steps_with_analysis"] = None
if "sql_analysis_results" not in st.session_state:
st.session_state["sql_analysis_results"] = None
if "sql_explanation_results" not in st.session_state:
st.session_state["sql_explanation_results"] = None
if "sql_user_corrections_by_step" not in st.session_state:
st.session_state["sql_user_corrections_by_step"] = []
if "sql_regeneration_results" not in st.session_state:
st.session_state["sql_regeneration_results"] = None
if "language" not in st.session_state:
st.session_state["language"] = "English"
if "timezone" not in st.session_state:
st.session_state["timezone"] = "UTC"
if "chosen_tab_id" not in st.session_state:
st.session_state["chosen_tab_id"] = "1"
if "sql_generation_reasoning" not in st.session_state:
st.session_state["sql_generation_reasoning"] = None


def onchange_demo_dataset():
Expand Down Expand Up @@ -240,10 +230,6 @@ def onchange_timezone():
ask_details_result = ask_details()
if ask_details_response := ask_details_result.get("response"):
st.session_state["asks_details_result"] = ask_details_response
st.session_state["sql_explanation_question"] = None
st.session_state["sql_explanation_steps_with_analysis"] = None
st.session_state["sql_analysis_results"] = None
st.session_state["sql_explanation_results"] = None
else:
st.error(
f'An error occurred while processing the query: {ask_details_result.get("error")}',
Expand Down
Loading
Loading