[PYTHON/LANGCHAIN] VectorStoreRetriever 클래스 : as_tool 메소드를 사용해 Tool 객체 만들기

■ VectorStoreRetriever 클래스의 as_tool 메소드를 사용해 Tool 객체를 만드는 방법을 보여준다.

▶ main.py


from langchain_core.documents    import Document
from langchain_openai            import OpenAIEmbeddings
from langchain_core.vectorstores import InMemoryVectorStore

documentList = [
    Document(page_content = "Dogs are great companions, known for their loyalty and friendliness."),
    Document(page_content = "Cats are independent pets that often enjoy their own space."         )
]

openAIEmbeddings = OpenAIEmbeddings()

inMemoryVectorStore = InMemoryVectorStore.from_documents(
    documentList,
    embedding = openAIEmbeddings
)

vectorStoreRetriever = inMemoryVectorStore.as_retriever(
    search_type   = "similarity",
    search_kwargs = {"k" : 1}
)

tool = vectorStoreRetriever.as_tool(
    name        = "pet_info_retriever",
    description = "Get information about pets.",
)

from langchain_core.documents import Document

from langchain_openai import OpenAIEmbeddings

from langchain_core.vectorstores import InMemoryVectorStore

documentList = [

Document(page_content = "Dogs are great companions, known for their loyalty and friendliness."),

Document(page_content = "Cats are independent pets that often enjoy their own space." )

]

openAIEmbeddings = OpenAIEmbeddings()

inMemoryVectorStore = InMemoryVectorStore.from_documents(

documentList,

embedding = openAIEmbeddings

)

vectorStoreRetriever = inMemoryVectorStore.as_retriever(

search_type = "similarity",

search_kwargs = {"k" : 1}

)

tool = vectorStoreRetriever.as_tool(

name = "pet_info_retriever",

description = "Get information about pets.",

)

※ pip install langchain-openai 명령을 실행했다.