Docs

pyproject.toml

@@ -30,10 +30,10 @@ google-api-python-client = "^2.116.0"
 pymupdf = "^1.23.21"
 google-auth-oauthlib = "^1.2.0"
 google-auth-httplib2 = "^0.2.0"
-langchain = "^0.1.11"
+langchain = "^0.1.14"
 langchain-core = "^0.1.25"
-langchain-community = "^0.0.27"
-langchain-openai = "^0.0.8"
+langchain-community = "^0.0.31"
+langchain-openai = "^0.1.1"
 pydantic = "2.6.1"
 langchain-pinecone = "^0.0.3"
 

src/sayvai_tools/tools/__init__.py

@@ -18,6 +18,7 @@
 from sayvai_tools.tools.loader import load_tools
 from sayvai_tools.tools.pdfreader import (ReadPagesTool, ReadPageTool,
                                           ReadPDFTool)
+from sayvai_tools.tools.pincone import PineconeTool
 from sayvai_tools.tools.retrive_details import RetrieveEmail, RetrievePhone
 from sayvai_tools.tools.send_mail import SendMail
 from sayvai_tools.tools.sql_database import Database
@@ -53,6 +54,7 @@
     "InsertCommentTool",
     "ListCommentRepliesTool",
     "ReplyToCommentTool",
+    "PineconeTool",
     "load_tools",
 ]
 

src/sayvai_tools/tools/google_calendar/display_events.py

@@ -65,8 +65,10 @@ def _run(
         **kwargs,
     ) -> List[dict]:
         try:
-            events = self._display_events(calendar_id=kwargs.get("calendar_id", "primary"),
-                                          max_results=kwargs.get("max_results", 10))
+            events = self._display_events(
+                calendar_id=kwargs.get("calendar_id", "primary"),
+                max_results=kwargs.get("max_results", 10),
+            )
             return events
         except Exception as e:
             raise Exception(f"An error occurred: {e}")

src/sayvai_tools/tools/loader.py

@@ -18,6 +18,7 @@
                                               get_sheets_credentials)
 from sayvai_tools.tools.pdfreader import (ReadPagesTool, ReadPageTool,
                                           ReadPDFTool)
+from sayvai_tools.tools.pincone import PineconeTool
 from sayvai_tools.tools.retrive_details import RetrieveEmail, RetrievePhone
 from sayvai_tools.tools.send_mail import SendMail
 from sayvai_tools.tools.sql_database import Database
@@ -49,6 +50,7 @@
     "RetrieveEmail": RetrieveEmail,
     "RetrievePhone": RetrievePhone,
     "Database": Database,
+    "PineconeTool": PineconeTool,
 }
 
 

src/sayvai_tools/tools/pinecone/__init__.py

@@ -0,0 +1,5 @@
+"""Pinecone tool for searching from vector database"""
+
+from sayvai_tools.tools.pincone.tool import PineconeTool
+
+__all__ = ["PineconeTool"]

src/sayvai_tools/tools/pinecone/tool.py

@@ -0,0 +1,47 @@
+# Pinceone as a Tool
+import os
+from typing import Optional
+
+from langchain_core.callbacks import (AsyncCallbackManagerForToolRun,
+                                      CallbackManagerForToolRun)
+from langchain_core.pydantic_v1 import BaseModel, Field, root_validator
+from langchain_core.tools import BaseTool
+from langchain_openai import OpenAIEmbeddings
+from langchain_pinecone import PineconeVectorStore
+
+
+class BasePineconeTool(BaseModel):
+    """Base tool for interacting with a pinecone vector database."""
+
+    class _QueryPinconeBaseToolInput(BaseModel):
+        query: str = Field(..., description="A natural language question")
+
+
+class PineconeTool(BasePineconeTool, BaseTool):
+    """Tool that queries the Pinecone API."""
+
+    name: str = "pinecone-search"
+    description: str = (
+        "pinecone-search searches from vector database"
+        "This tool is will search answer for user's query from Pincone"
+        "Input should be a search query."
+    )
+
+    vectorstore: PineconeVectorStore = Field(
+        default_factory=lambda: PineconeVectorStore.from_existing_index(
+            index_name=os.environ["PINECONE_INDEX"], embedding=OpenAIEmbeddings()
+        ),
+        description="The Pinecone vector store",
+    )
+
+    @classmethod
+    def create(cls, **kwargs) -> "PineconeTool":
+        return cls()
+
+    def _run(
+        self,
+        query: str,
+        run_manager: Optional[CallbackManagerForToolRun] = None,
+    ) -> str:
+        """Use the tool."""
+        return self.vectorstore.similarity_search(query=query)[0].page_content

src/sayvai_tools/tools/sql_database/tool.py

@@ -36,7 +36,9 @@
 #     )
 
 
-@deprecated(message="Use langchain tool instead, Database will be removed from sayvai-tools 0.0.5")
+@deprecated(
+    message="Use langchain tool instead, Database will be removed from sayvai-tools 0.0.5"
+)
 class Database:
     """Tool that queries vector database."""