Yoon-gu Hwang commited on
Commit
812e2f4
·
1 Parent(s): 015e384

tool을 gradio mcp에 추가

Browse files
Files changed (2) hide show
  1. app.py +14 -0
  2. main.py +1 -1
app.py CHANGED
@@ -70,6 +70,16 @@ def echo(message, history):
70
  def vote(data: gr.LikeData):
71
  print(data.liked)
72
 
 
 
 
 
 
 
 
 
 
 
73
  with gr.Blocks() as chat_ui:
74
  chatbot = gr.Chatbot(placeholder="<strong>Your Personal Echo-Man</strong><br>Ask Me Anything")
75
  chatbot.like(vote, None, None)
@@ -81,12 +91,16 @@ demo = gr.TabbedInterface(
81
  gr.Interface(prime_factors, gr.Textbox("1001"), gr.Textbox()),
82
  gr.Interface(image_orientation, gr.Image(type="pil"), gr.Textbox()),
83
  gr.Interface(sepia, gr.Image(), gr.Image()),
 
 
84
  ],
85
  [
86
  "Chat UI",
87
  "Prime Factors",
88
  "Image Orientation Checker",
89
  "Sepia Filter",
 
 
90
  ]
91
  )
92
 
 
70
  def vote(data: gr.LikeData):
71
  print(data.liked)
72
 
73
+
74
+ def add(a: int, b: int) -> int:
75
+ """Add two numbers"""
76
+ return a + b
77
+
78
+ def multiply(a: int, b: int) -> int:
79
+ """Multiply two numbers"""
80
+ return a * b
81
+
82
+
83
  with gr.Blocks() as chat_ui:
84
  chatbot = gr.Chatbot(placeholder="<strong>Your Personal Echo-Man</strong><br>Ask Me Anything")
85
  chatbot.like(vote, None, None)
 
91
  gr.Interface(prime_factors, gr.Textbox("1001"), gr.Textbox()),
92
  gr.Interface(image_orientation, gr.Image(type="pil"), gr.Textbox()),
93
  gr.Interface(sepia, gr.Image(), gr.Image()),
94
+ gr.Interface(add, [gr.Number(), gr.Number()], gr.Number()),
95
+ gr.Interface(multiply, [gr.Number(), gr.Number()], gr.Number())
96
  ],
97
  [
98
  "Chat UI",
99
  "Prime Factors",
100
  "Image Orientation Checker",
101
  "Sepia Filter",
102
+ "Add",
103
+ "multiply"
104
  ]
105
  )
106
 
main.py CHANGED
@@ -1,10 +1,10 @@
1
  import asyncio
2
  from mcp import ClientSession
3
  from mcp.client.sse import sse_client
4
-
5
  from langchain_mcp_adapters.tools import load_mcp_tools
6
  from langgraph.prebuilt import create_react_agent
7
  from pprint import pprint
 
8
  async def main():
9
  sse_url = "http://127.0.0.1:7860/gradio_api/mcp/sse" # 실제 SSE 서버 URL로 변경
10
 
 
1
  import asyncio
2
  from mcp import ClientSession
3
  from mcp.client.sse import sse_client
 
4
  from langchain_mcp_adapters.tools import load_mcp_tools
5
  from langgraph.prebuilt import create_react_agent
6
  from pprint import pprint
7
+
8
  async def main():
9
  sse_url = "http://127.0.0.1:7860/gradio_api/mcp/sse" # 실제 SSE 서버 URL로 변경
10