luis.erlacher/Archon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Add timeout handling to MCP server requests - Add 5-minute timeout, proper error handling, and improved logging (#27)

Browse Source 
Co-authored-by: Takshit Mathur <takshitmathur@gmail.com>
This commit is contained in:
takshit12 2025-03-09 17:39:14 +05:30 committed by GitHub
parent fc3c77e943
commit 30e586bfcb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
mcp/mcp_server.py
View File

@ -57,17 +57,25 @@ async def create_thread() -> str:
def _make_request(thread_id: str, user_input: str, config: dict) -> str:
"""Make synchronous request to graph service"""
response = requests.post(
f"{GRAPH_SERVICE_URL}/invoke",
json={
"message": user_input,
"thread_id": thread_id,
"is_first_message": not active_threads[thread_id],
"config": config
}
)
response.raise_for_status()
return response.json()
try:
response = requests.post(
f"{GRAPH_SERVICE_URL}/invoke",
json={
"message": user_input,
"thread_id": thread_id,
"is_first_message": not active_threads[thread_id],
"config": config
},
timeout=300 # 5 minute timeout for long-running operations
)
response.raise_for_status()
return response.json()
except requests.exceptions.Timeout:
write_to_log(f"Request timed out for thread {thread_id}")
raise TimeoutError("Request to graph service timed out. The operation took longer than expected.")
except requests.exceptions.RequestException as e:
write_to_log(f"Request failed for thread {thread_id}: {str(e)}")
raise
@mcp.tool()