File size: 1,035 Bytes
c3c6e00 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
import gradio as gr
from mcp import StdioServerParameters
from smolagents import MCPClient, CodeAgent
def respond(prompt, history=[]):
# Launch MCP subprocess
params = StdioServerParameters(command="python", args=["mcp_server.py"])
with MCPClient(params) as tools:
agent = CodeAgent(tools=tools, model="gpt-4") # Replace with your local or API model
result = agent.run(prompt)
history.append((prompt, result))
return history, history
with gr.Blocks() as demo:
chat_history = gr.State([])
chatbot = gr.Chatbot(label="Enterprise SQL Agent")
textbox = gr.Textbox(show_label=False, placeholder="Ask questions like 'List inactive customers in Northeast'")
textbox.submit(respond, [textbox, chat_history], [chatbot, chat_history])
gr.Markdown(\"\"\"\n
### Example Prompts:
- Who are my Northeast customers who haven’t ordered in 6 months?
- List customers sorted by last order date.
- Find clients from the West with recent orders.
\"\"\")
demo.launch()
|