extract url

.env.sample

@@ -1 +1,2 @@
-OPENAI_API_KEY=
+OPENAI_API_KEY=
+MCP_SERVER=

src/agent.py

@@ -2,6 +2,7 @@ from langgraph.prebuilt import create_react_agent
 from pydantic import BaseModel, Field
 from dotenv import load_dotenv
 from langchain_mcp_adapters.client import MultiServerMCPClient
+import os
 
 class AgentOutput(BaseModel):
     final_response: str = Field(description="The final response to the user.", default="")
@@ -61,12 +62,11 @@ Remember: Stay focused on audio-related assistance and use your specialized tool
 class AudioAgent:
     def __init__(
         self,
-        model_name: str = "gpt-4.1",
-        server_url: str = "https://agents-mcp-hackathon-audioeditor.hf.space/gradio_api/mcp/sse",
+        model_name: str = "gpt-4.1-mini",
     ):
         load_dotenv()
         self.model_name = model_name
-        self.server_url = server_url
+        self.server_url = os.getenv("MCP_SERVER")
         self.graph = None
 
         self._client = MultiServerMCPClient({
@@ -79,7 +79,7 @@ class AudioAgent:
         tools = await self._client.get_tools()
 
         agent = create_react_agent(
-            model="gpt-4.1",
+            model=self.model_name,
             tools=tools,
             prompt=system_prompt,
             response_format=AgentOutput,
@@ -97,7 +97,7 @@ class AudioAgent:
         """
 
         res = await self.agent.ainvoke(
-            {"messages": [{"role": "user", "content": input_context}]}
+            {"messages": [{"role": "user", "content": input_context}]},
         )
 
         return res["structured_response"]