refactor: new TODO's added

gradio_app.py

@@ -18,7 +18,7 @@ def load_db_configs():
 
 # Async-compatible wrapper
 async def run_agent(request):
-    # configs = load_db_configs()  # Optional if needed
+    # configs = load_db_configs()
     final_answer, last_tool_answer, = await pg_mcp_exec(request)
     return final_answer, last_tool_answer
 
@@ -31,7 +31,7 @@ demo = gr.Interface(
     ),
     outputs=gr.Textbox(label="SQL Query / Result"),
     title="PostgreSQL Query Agent",
-    description="Ask your database in natural language and get results using the smolagent executor."
+    description="Ask your database in natural language and get results."
 )
 
 if __name__ == "__main__":

main.py

@@ -7,6 +7,7 @@ import logging
 
 #logger = logging.getLogger(__name__)
 # TODO add config
+# TODO add README
 def load_db_configs():
     """Load database configurations from databases.yaml"""
     configs_path = Path("configs.yaml")
@@ -23,7 +24,7 @@ def load_db_configs():
 async def main():
     #configs = load_db_configs()
 
-    request = "Show me the table of join posts and users tables."
+    request = "send the table"
     await pg_mcp_exec(request)
 
 

postgre_mcp_client.py

@@ -1,4 +1,6 @@
 import os.path
+from typing import Tuple, Any
+
 from mcp import ClientSession, StdioServerParameters
 from mcp.client.stdio import stdio_client
 from langchain_mcp_adapters.tools import load_mcp_tools
@@ -10,7 +12,7 @@ import logging
 
 logger = logging.getLogger(__name__)
 
-async def pg_mcp_exec(request: str) -> str:
+async def pg_mcp_exec(request: str) -> tuple[Any, Any]:
     """
     Execute the full PostgreSQL MCP pipeline: load summary, connect session,
     load memory and tools, build prompt, run agent, update memory.
@@ -27,7 +29,7 @@ async def pg_mcp_exec(request: str) -> str:
     server_params = get_server_params()
 
     # TODO: give key from env
-    llm = init_chat_model(model="gemini-2.0-flash-lite", model_provider="google_genai",
+    llm = init_chat_model(model="gemini-1.5-flash", model_provider="google_genai",
                           api_key="AIzaSyAuxYmci0DVU5l5L_YcxLlxHzR5MLn70js")
 
     async with stdio_client(server_params) as (read, write):

postgre_smolagent_clinet.py

@@ -45,6 +45,7 @@ async def pg_mcp_smolagent_exec(request: str) -> str:
                 prompt = await build_prompt(session, intent, request, tools, past_data)
                 agent_response = agent.run(task=prompt, stream=False)
 
+                # TODO: add a smolagent output parser
 
                 #parsed_steps, _ = parse_mcp_output(agent_response)
                 #memory.update_from_parsed(parsed_steps, request)

streamlit_app.py

@@ -26,7 +26,7 @@ def run_agent(message):
 
 # Streamlit UI
 st.title("PostgreSQL Query Agent")
-st.write("Ask your database in natural language and get results using the smolagent executor.")
+st.write("Ask your database in natural language and get results.")
 
 user_input = st.text_input("Natural Language Request", placeholder="e.g., Show me the table of join posts and users tables.")