From 9e26d96b3cdd4e6ac1f8a3ae66b94e20950e8fd7 Mon Sep 17 00:00:00 2001
From: Mohd Kaif <98801504+KaifAhmad1@users.noreply.github.com>
Date: Mon, 13 Apr 2026 17:46:50 +0530
Subject: [PATCH] Potential fix for pull request finding 'Wrong number of
 arguments in a call'

Co-authored-by: Copilot Autofix powered by AI <223894421+github-code-quality[bot]@users.noreply.github.com>
---
 mcp/tools/extraction.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/mcp/tools/extraction.py b/mcp/tools/extraction.py
index edda83bd..7ccd385b 100644
--- a/mcp/tools/extraction.py
+++ b/mcp/tools/extraction.py
@@ -54,8 +54,9 @@ def handle_extract_relations(args: dict) -> dict:
         return {"error": "text is required", "relations": [], "triplets": []}
     _clear_cache()
     try:
-        from semantica.semantic_extract import RelationExtractor, TripletExtractor
-        relations = RelationExtractor().extract(text) or []
+        from semantica.semantic_extract import NamedEntityRecognizer, RelationExtractor, TripletExtractor
+        entities = NamedEntityRecognizer().extract(text) or []
+        relations = RelationExtractor().extract(text, entities) or []
         triplets = TripletExtractor().extract(text) or []
         return {
             "relations": [