feat: text summarization tool for supervisor agent

app.py

@@ -6,6 +6,7 @@ from langgraph.checkpoint.memory import InMemorySaver
 from dotenv import load_dotenv
 from agents import get_book_retriver_agent
 from tools import get_retrieve_book_context_rag_agent_tool
+from tools import get_summarization_tool
 
 
 class GradioAgent:
@@ -37,7 +38,8 @@ class GradioAgent:
         rag_agent = get_book_retriver_agent()
 
         tools = [
-            get_retrieve_book_context_rag_agent_tool(rag_agent)
+            get_retrieve_book_context_rag_agent_tool(rag_agent),
+            get_summarization_tool()
         ]
 
         prompt = ("You are a helpful and usefull coordinator agent, you have access to a collection of tools and"

tools/TextSummarizerTool.py

@@ -0,0 +1,26 @@
+from langchain_huggingface import HuggingFacePipeline
+from transformers import pipeline
+from langchain.tools import tool
+
+
+def get_summarization_tool():
+
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+    hf_summarizer = HuggingFacePipeline(pipeline=summarizer)
+
+    @tool
+    def summarize_tool(text: str) -> str:
+        """ Summarize a given text only if the user explicitly asks for a summary or if summarization is clearly required.
+
+           Args:
+               text: text to be summarized
+        """
+        sumirized_text = hf_summarizer.invoke(
+            text,
+            max_length=350,
+            min_length=50,
+            do_sample=False
+        )
+        return sumirized_text
+
+    return summarize_tool

tools/__init__.py

@@ -1,2 +1,3 @@
 from .RetriveBooksDataTool import get_retrieve_book_context_tool
 from .BookRetriverRagAgent import get_retrieve_book_context_rag_agent_tool
+from .TextSummarizerTool import get_summarization_tool