fix lint issues (#37)

.pylintrc

@@ -15,5 +15,6 @@ disable =
     too-many-return-statements, 
     import-outside-toplevel, 
     eval-used,
-    too-few-public-methods
+    too-few-public-methods,
+    unused-argument
 

vectara_agentic/_callback.py

@@ -1,3 +1,7 @@
+"""
+Module to handle agent callbacks
+"""
+
 import inspect
 from typing import Any, Dict, Optional, List, Callable
 
@@ -55,6 +59,18 @@ def on_event_end(
         event_id: str = "",
         **kwargs: Any,
     ) -> None:
+        """
+        Handle the end of an event
+
+        Args:
+            event_type: the type of event
+            payload: the event payload
+            event_id: the event ID
+            kwargs: additional keyword arguments
+
+        Returns:
+            None
+        """
         if self.fn is not None and payload is not None:
             if inspect.iscoroutinefunction(self.fn):
                 raise ValueError("Synchronous callback handler cannot use async callback function")
@@ -70,6 +86,19 @@ async def aon_event_start(
         parent_id: str = "",
         **kwargs: Any,
     ) -> str:
+        """
+        Handle the start of an event
+
+        Args:
+            event_type: the type of event
+            payload: the event payload
+            event_id: the event ID
+            parent_id: the parent event ID
+            kwargs: additional keyword arguments
+
+        Returns:
+            event_id: the event ID
+        """
         if self.fn is not None and payload is not None:
             await self._ahandle_event(event_type, payload)
         return event_id
@@ -81,6 +110,9 @@ async def aon_event_end(
         event_id: str = "",
         **kwargs: Any,
     ) -> None:
+        """
+        Handle the end of an event (async)
+        """
         if self.fn is not None and payload is not None:
             await self._ahandle_event(event_type, payload)
 

vectara_agentic/db_tools.py

@@ -34,7 +34,10 @@ def __call__(self, query: str) -> Any:
         count_rows = self.load_data_fn(count_query)
         num_rows = int(count_rows[0].text)
         if num_rows > self.max_rows:
-            return [f"The query is expected to return more than {self.max_rows} rows. Please refine your query to make it return less rows."]
+            return [
+                f"The query is expected to return more than {self.max_rows} rows. "
+                "Please refine your query to make it return less rows."
+            ]
 
         res = self.load_data_fn(query)
         return [d.text for d in res]

vectara_agentic/tools.py

@@ -478,8 +478,8 @@ def database_tools(
             password (str, optional): The database password. Defaults to "Password".
             dbname (str, optional): The database name. Defaults to "postgres".
                You must specify either the sql_database object or the scheme, host, port, user, password, and dbname.
-            max_rows (int, optional): if specified, instructs the load_data tool to never return more than max_rows rows.
-               Defaults to 500.
+            max_rows (int, optional): if specified, instructs the load_data tool to never return more than max_rows
+               rows. Defaults to 500.
 
         Returns:
             List[VectaraTool]: A list of VectaraTool objects.