VirtualOasis commited on
Commit
4b633dd
·
verified ·
1 Parent(s): 4c0ce46

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -0
app.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from smolagents import (
2
+ load_tool,
3
+ CodeAgent,
4
+ LiteLLMModel,
5
+ GradioUI,
6
+ MCPClient
7
+ )
8
+ from mcp import StdioServerParameters
9
+ import os
10
+ from dotenv import load_dotenv
11
+ load_dotenv()
12
+
13
+ kgb_server_parameters = StdioServerParameters(
14
+ command="npx",
15
+ args=[
16
+ "mcp-remote",
17
+ "https://agents-mcp-hackathon-kgb-mcp.hf.space/gradio_api/mcp/sse",
18
+ "--transport",
19
+ "sse-only"],
20
+ )
21
+
22
+ # Initialize the model
23
+ model = LiteLLMModel(
24
+ model_id="gemini/gemini-2.0-flash-exp",
25
+ api_key=os.getenv("GEMINI_API_TOKEN")
26
+ )
27
+
28
+ # Load tools from all MCP servers using MCPClient
29
+ server_parameters = [kgb_server_parameters]
30
+
31
+ mcp = MCPClient(server_parameters)
32
+ with mcp:
33
+ all_tools = mcp.get_tools()
34
+ agent = CodeAgent(tools=all_tools, model=model, add_base_tools=True)
35
+
36
+ GradioUI(agent).launch()