[feature] langgraph tools (websearch, vectorstore search) 생성

app/langgraph/tools.py

@@ -0,0 +1,51 @@
+import getpass
+import os
+import sys
+
+from langchain_community.tools import DuckDuckGoSearchRun
+from langchain_core.tools import tool
+from app.vectorstore.opensearch_hybrid import OpenSearchHybridSearch
+
+# tool에 필요한 class 초기화
+duckduckgo_search = DuckDuckGoSearchRun()
+
+opensearch_password = getpass.getpass("Enter your Opensearch password: ")
+opensearch = OpenSearchHybridSearch(user="admin", pw=opensearch_password)
+
+@tool(parse_docstring=True)
+def web_search(query: str):
+    """A tool to use when websearch is needed. Use this tool when there isn't enough information in OpenSearch vector store Database
+
+    Args:
+        query : a query to websearch
+
+    Returns:
+        list[str]: websearch result in list of strings
+    """
+    res = duckduckgo_search.invoke({"query": query})
+    return [res]
+
+
+@tool(parse_docstring=True)
+def vectorstore_search(query: str, search_type: str = "hybrid"):
+    """search OpenSearch vector databsee for documents that are relevant to a given query
+
+    Args:
+        query: a query to search vector database
+        search_type (str, optional): one of three similarity search method: hybrid, BM25, and Cosine similarity search Defaults to "hybrid".
+
+    Returns:
+        list[pd.DataFrmae]: list of langchain document
+    """
+    if search_type == "hybrid":
+        result = opensearch.hybrid_search(query)
+    elif search_type == "bm25":
+        result = opensearch.bm25_search(query)
+    elif search_type == "cosine":
+        result = opensearch.cosine_similarity_search(query)
+
+    result["text"].tolist()
+
+    return result
+
+

pyproject.toml

@@ -4,6 +4,7 @@ version = "0.0.0"
 description = "ML/DL 관련 지식과 최신 정보관련 대화할 수 있는 챗봇"
 authors = ["tmdqja75 <[email protected]>"]
 readme = "README.md"
+packages = [{ include = "app" }]
 
 [tool.poetry.dependencies]
 python = "^3.10"
@@ -24,6 +25,7 @@ langchain-huggingface = "^0.1.0"
 faiss-cpu = "^1.9.0"
 langchain-ollama = "^0.2.0"
 tavily-python = "^0.5.0"
+duckduckgo-search = "^6.3.7"
 
 
 [tool.poetry.group.dev.dependencies]