Update mcp/orchestrator.py

mcp/orchestrator.py

@@ -16,8 +16,7 @@ from mcp.disgenet import disease_to_genes
 from mcp.gemini import gemini_summarize, gemini_qa
 from mcp.openai_utils import ai_summarize, ai_qa
 
-
-def _safe_call(
+async def _safe_call(
     func: Any,
     *args,
     default: Any = None,
@@ -27,13 +26,12 @@ def _safe_call(
     Safely call an async function, returning a default on HTTP or other failures.
     """
     try:
-        return await func(*args, **kwargs)  # type: ignore
+        return await func(*args, **kwargs)
     except httpx.HTTPStatusError:
         return default
     except Exception:
         return default
 
-
 async def _gather_tasks(tasks: List[asyncio.Task]) -> List[Any]:
     """
     Await a list of asyncio.Tasks and return their results in order.
@@ -60,7 +58,6 @@ def _flatten_unique(items: List[Union[List[Any], Any]]) -> List[Any]:
                 flat.append(elem)
     return flat
 
-
 async def orchestrate_search(
     query: str,
     llm: Literal['openai', 'gemini'] = 'openai',
@@ -129,7 +126,6 @@ async def orchestrate_search(
         'llm_used': llm_used,
     }
 
-
 async def answer_ai_question(
     question: str,
     context: str = "",