Javier-Jimenez99 commited on
Commit
4f4d8ef
·
1 Parent(s): 878c0f0

Actualizar la conexión SSE a localhost y modificar la función de chat para usar 'tab_id' en lugar de 'correlation_id'

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -10,7 +10,8 @@ import gradio as gr
10
  import re
11
 
12
  async def initialize_agent():
13
- connection = SSEConnection(url="https://javier-jimenez99-owlbear-mcp-server.hf.space/gradio_api/mcp/sse", transport="sse")
 
14
  tools = await load_mcp_tools(session=None,connection=connection)
15
  llm = ChatOllama(model="qwen3:8b")
16
  agent = create_react_agent(llm, tools)
@@ -30,10 +31,10 @@ def initialize_agent_sync():
30
  agent = initialize_agent_sync()
31
 
32
  # Function for HTTP API
33
- async def chat(message: str, correlation_id: str = None):
34
  """Function to handle direct HTTP requests"""
35
- if correlation_id:
36
- message = f"{message} \n\n This is your correlation ID: {correlation_id}"
37
 
38
  print(f"Received message: {message}")
39
 
@@ -50,7 +51,7 @@ async def chat(message: str, correlation_id: str = None):
50
  # Create API interface
51
  demo = gr.Interface(
52
  fn=chat,
53
- inputs="text",
54
  outputs="text",
55
  )
56
 
 
10
  import re
11
 
12
  async def initialize_agent():
13
+ #connection = SSEConnection(url="https://javier-jimenez99-owlbear-mcp-server.hf.space/gradio_api/mcp/sse", transport="sse")
14
+ connection = SSEConnection(url="http://localhost:7861/gradio_api/mcp/sse", transport="sse")
15
  tools = await load_mcp_tools(session=None,connection=connection)
16
  llm = ChatOllama(model="qwen3:8b")
17
  agent = create_react_agent(llm, tools)
 
31
  agent = initialize_agent_sync()
32
 
33
  # Function for HTTP API
34
+ async def chat(message: str, tab_id: str):
35
  """Function to handle direct HTTP requests"""
36
+ if tab_id:
37
+ message = f"{message}\nThis is your correlation ID: {tab_id}"
38
 
39
  print(f"Received message: {message}")
40
 
 
51
  # Create API interface
52
  demo = gr.Interface(
53
  fn=chat,
54
+ inputs=[gr.Textbox(label="message"), gr.Textbox(label="tab_id")],
55
  outputs="text",
56
  )
57