VirtualOasis commited on
Commit
575de29
·
verified ·
1 Parent(s): 01ec81e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -19,6 +19,15 @@ kgb_server_parameters = StdioServerParameters(
19
  "sse-only"],
20
  )
21
 
 
 
 
 
 
 
 
 
 
22
  # Initialize the model
23
  model = LiteLLMModel(
24
  model_id="gemini/gemini-2.0-flash-exp",
@@ -26,7 +35,7 @@ model = LiteLLMModel(
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:
 
19
  "sse-only"],
20
  )
21
 
22
+ T2I_server_parameters = StdioServerParameters(
23
+ command="npx",
24
+ args=[
25
+ "mcp-remote",
26
+ "https://agents-mcp-hackathon-t2i.hf.space/gradio_api/mcp/sse",
27
+ "--transport",
28
+ "sse-only"],
29
+ )
30
+
31
  # Initialize the model
32
  model = LiteLLMModel(
33
  model_id="gemini/gemini-2.0-flash-exp",
 
35
  )
36
 
37
  # Load tools from all MCP servers using MCPClient
38
+ server_parameters = [kgb_server_parameters,T2I_server_parameters]
39
 
40
  mcp = MCPClient(server_parameters)
41
  with mcp: