init commit of amirkia_test branch

.env.sample

@@ -1 +1,2 @@
-API_KEY = ""
+API_KEY=
+MCP_SERVER_PATH=/home/andre/andre/postgre_mcp_server.py

gradio_app.py

@@ -24,8 +24,8 @@ async def run_agent(request, history):
     # final_answer, last_tool_answer, = await pg_mcp_exec(request)
     # return final_answer, last_tool_answer
 
-    result = await lc_mcp_exec(request, history)
-    return result
+    response, message = await lc_mcp_exec(request, history)
+    return response
 
 # Gradio UI
 demo = gr.Interface(

langchain_mcp_client.py

@@ -94,9 +94,10 @@ def load_table_summary(path: str) -> str:
 
 def get_server_params() -> StdioServerParameters:
     # TODO: give server params from config
+    load_dotenv()
     return StdioServerParameters(
         command="python",
-        args=[r"C:\Users\yukcus\Desktop\query_mcp_server\postgre_mcp_server.py"],
+        args=[os.environ["MCP_SERVER_PATH"]],
     )
 
 async def load_and_enrich_tools(session: ClientSession):

postgre_smolagent_clinet.py

@@ -8,6 +8,8 @@ from conversation_memory import ConversationMemory
 from utils import parse_mcp_output, classify_intent
 import logging
 from smolagents import LiteLLMModel, ToolCollection, CodeAgent
+from dotenv import load_dotenv
+
 
 logger = logging.getLogger(__name__)
 
@@ -63,9 +65,10 @@ def load_table_summary(path: str) -> str:
 
 def get_server_params() -> StdioServerParameters:
     # TODO: give server params from config
+    load_dotenv()
     return StdioServerParameters(
         command="python",
-        args=[r"/home/amirkia/Desktop/query_mcp_server/postgre_mcp_server.py"],
+        args=[os.environ["MCP_SERVER_PATH"]],
     )
 
 async def load_or_create_memory() -> ConversationMemory: