Merge pull request #53 from Sayvai-io/tools-langchain-0.1

Tools langchain 0.1

pyproject.toml

@@ -22,7 +22,7 @@ google = "^3.0.0"
 google-cloud-speech = "^2.23.0"
 cloud-sql-python-connector = "^1.5.0"
 pg8000 = "^1.30.4"
-pandas = "^2.1.4"
+pandas = "^2.2.0"
 openpyxl = "^3.1.2"
 gspread = "5.11.3"
 pydub = "^0.25.1"

src/sayvai_tools/tools/__init__.py

@@ -8,8 +8,7 @@
 from sayvai_tools.tools.date import GetDate
 from sayvai_tools.tools.display_events import DisplayEvents
 from sayvai_tools.tools.forms import FormTool
-from sayvai_tools.tools.pdfreader import (ReadPagesTool, ReadPageTool,
-                                          ReadPdfTool)
+from sayvai_tools.tools.pdfreader import ReadPagesTool, ReadPageTool, ReadPDFTool
 from sayvai_tools.tools.retrive_details import RetrieveEmail, RetrievePhone
 from sayvai_tools.tools.send_mail import SendMail
 from sayvai_tools.tools.spreadsheets import Sheets, SQLSheet
@@ -37,7 +36,7 @@
     "PGVectorDB",
     "PineconeDB",
     "ReadPagesTool",
-    "ReadPdfTool",
+    "ReadPDFTool",
     "ReadPageTool",
 ]
 
@@ -51,6 +50,7 @@ def get_tool(tool_name: str) -> str:
         return tool_name
     else:
         return "Tool not found"
-
+
+
 def get_langchain_tools() -> List[str]:
     return langchain_tools

src/sayvai_tools/tools/pdfreader/__init__.py

@@ -1,5 +1,4 @@
 # init file from pdf reader tool
-from sayvai_tools.tools.pdfreader.tool import (ReadPagesTool, ReadPageTool,
-                                               ReadPdfTool)
+from sayvai_tools.tools.pdfreader.tool import ReadPagesTool, ReadPageTool, ReadPDFTool
 
-__all__ = ["ReadPagesTool", "ReadPdfTool", "ReadPageTool"]
+__all__ = ["ReadPagesTool", "ReadPDFTool", "ReadPageTool"]

src/sayvai_tools/utils/__init__.py

@@ -5,6 +5,7 @@
 from sayvai_tools.utils.google.mail import EmailSender
 from sayvai_tools.utils.google.sheets import GSheets
 from sayvai_tools.utils.microsoft.excel import Excel
+
 # from sayvai_tools.utils.voice.stt import STT
 from sayvai_tools.utils.voice.tts import ElevenlabsAudioStreaming
 

src/sayvai_tools/utils/database/dbchain.py

@@ -8,13 +8,12 @@
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
-from langchain.chains.sql_database.prompt import (DECIDER_PROMPT, PROMPT,
-                                                  SQL_PROMPTS)
+from langchain.chains.sql_database.prompt import DECIDER_PROMPT, PROMPT, SQL_PROMPTS
 from langchain.prompts.prompt import PromptTemplate
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.sql_database.prompt import QUERY_CHECKER
-from langchain_experimental.pydantic_v1 import Extra, Field, root_validator
+from langchain.pydantic_v1 import  Extra, Field, root_validator
 
 from sayvai_tools.utils.database.dbbase import SQLDatabase
 

src/sayvai_tools/utils/database/sheetschain.py

@@ -8,13 +8,12 @@
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
-from langchain.chains.sql_database.prompt import (DECIDER_PROMPT, PROMPT,
-                                                  SQL_PROMPTS)
+from langchain.chains.sql_database.prompt import DECIDER_PROMPT, PROMPT, SQL_PROMPTS
 from langchain.prompts.prompt import PromptTemplate
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.sql_database.prompt import QUERY_CHECKER
-from langchain_experimental.pydantic_v1 import Extra, Field, root_validator
+from langchain.pydantic_v1 import Extra, Field, root_validator
 
 from sayvai_tools.utils.database.dbsheetsbase import SQLDatabaseSheetsBase
 

src/sayvai_tools/utils/exception.py

@@ -15,3 +15,11 @@ class SayvaiToolsWarning(Warning):
     def __init__(self, message: str):
         self.message = message
         super().__init__(self.message)
+
+
+class SayvaiToolsDeprecatedWarning(SayvaiToolsWarning):
+    """Base class for deprecated warnings in this module."""
+
+    def __init__(self, message: str):
+        self.message = message
+        super().__init__(self.message)

src/sayvai_tools/utils/pdfreader/__init__.py

@@ -1,5 +1,4 @@
 # init file
-from sayvai_tools.utils.pdfreader.readpdf import (read_page, read_pages,
-                                                  read_pdf)
+from sayvai_tools.utils.pdfreader.readpdf import read_page, read_pages, read_pdf
 
 __all__ = ["read_page", "read_pdf", "read_pages"]

src/sayvai_tools/utils/system/properties/__init__.py

@@ -1,6 +1,8 @@
 """init file for properties utils"""
 
 from sayvai_tools.utils.system.properties.properties import (
-    SystemProperties, get_all_properties)
+    SystemProperties,
+    get_all_properties,
+)
 
 __properties__ = properties