refactor: Change general description of tools and visibility in book retrieval agent

agents/BookRetriverAgent.py

@@ -12,7 +12,7 @@ class BookRetriverAgent:
 
     def __init__(self):
         self.vector_store = self.__initiate_vector_store()
-        self.add_books()
+        self._add_books()
         self.agent = self.__create_agent()
 
     def __create_agent(self):
@@ -39,7 +39,7 @@ class BookRetriverAgent:
         embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
         return InMemoryVectorStore(embeddings)
 
-    def add_books(self):
+    def _add_books(self):
         text_splitter = RecursiveCharacterTextSplitter(
             chunk_size=1000,
             chunk_overlap=200,

agents/__init__.py

@@ -1 +1 @@
-from .BookRetriverAgent import *
+from .BookRetriverAgent import BookRetriverAgent

app.py

@@ -36,8 +36,12 @@ class GradioAgent:
 
         rag_agent = BookRetriverAgent().agent
 
+        tools = [
+            get_retrieve_book_context_rag_agent_tool(rag_agent)
+        ]
+
         return create_agent(
-            tools=[get_retrieve_book_context_rag_agent_tool(rag_agent)],
+            tools=tools,
             model=llm,
             checkpointer=InMemorySaver(),
             system_prompt="You are a helpful and usefull assistant."

tools/BookRetriverRagAgent.py

@@ -4,8 +4,8 @@ from langchain.tools import tool
 def get_retrieve_book_context_rag_agent_tool(agent):
 
     @tool(
-        "Computer science book retriever",
-        description="Used to retrieve computer science related books context and answer the request."
+        "Computer science knowledge retriever",
+        description="Used to retrieve computer science related context and answer the request."
     )
     def retrieve_book_context_rag_agent(request: str):
         result = agent.invoke({

tools/RetriveBooksDataTool.py

@@ -3,10 +3,12 @@ from langchain.tools import tool
 
 def get_retrieve_book_context_tool(vector_store):
 
-    @tool
+    @tool(
+        "Books knowledge retriever",
+        description="Search in books for important and truthful context for a given query."
+    )
     def retrieve_book_context(query: str):
-        """Search for important book context for a given query."""
-        retrieved_docs = vector_store.similarity_search(query, k=4)
+        retrieved_docs = vector_store.si(query, k=4)
         serialized = "\n\n".join(
             (f"Source: {doc.metadata}\nContent: {doc.page_content}")
             for doc in retrieved_docs