fix: summarization removed, it caused a lot of latency

app.py

@@ -6,7 +6,6 @@ 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:
@@ -38,12 +37,12 @@ class GradioAgent:
         rag_agent = get_book_retriver_agent()
 
         tools = [
-            get_retrieve_book_context_rag_agent_tool(rag_agent),
-            get_summarization_tool()
+            get_retrieve_book_context_rag_agent_tool(rag_agent)
         ]
 
         prompt = ("You are a helpful and usefull coordinator agent, you have access to a collection of tools and"
-                  " agents to help you with reliable data to your query's.")
+                  " agents to help you with reliable data to your query's. "
+                  "One of your main objectives is to generate user friendly answers based on the information you have.")
 
         return create_agent(
             tools=tools,

tools/RetriveBooksDataTool.py

@@ -9,7 +9,6 @@ def get_retrieve_book_context_tool(vector_store):
     )
     def retrieve_book_context(query: str) -> str:
         """Search the books database for records matching the query.
-
            Args:
                query: Search terms to look for
         """

tools/TextSummarizerTool.py

@@ -1,26 +0,0 @@
-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 giving text.
-
-           Args:
-               text: text to be summarized
-        """
-        sumirized_text = hf_summarizer.invoke(
-            text,
-            max_length=200,
-            min_length=50,
-            do_sample=False
-        )
-        return sumirized_text
-
-    return summarize_tool

tools/__init__.py

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