[PYTHON/LANGCHAIN] VectorStoreRetrieverMemory 클래스 : 생성자에서 retriever 인자를 사용해 VectorStoreRetrieverMemory 객체 만들기

icodebroker LANGCHAIN 2025-01-13

■ VectorStoreRetrieverMemory 클래스의 생성자에서 retriever 인자를 사용해 VectorStoreRetrieverMemory 객체를 만드는 방법을 보여준다.

※ OPENAI_API_KEY 환경 변수 값은 .env 파일에 정의한다.

▶ 예제 코드 (PY)


import faiss

from dotenv                 import load_dotenv
from langchain_openai       import OpenAIEmbeddings
from langchain.docstore     import InMemoryDocstore
from langchain.vectorstores import FAISS
from langchain.memory       import VectorStoreRetrieverMemory

load_dotenv()

openAIEmbeddings = OpenAIEmbeddings()

indexFlatL2 = faiss.IndexFlatL2(1536)

faiss = FAISS(openAIEmbeddings, indexFlatL2, InMemoryDocstore({}), {})

vectorStoreRetriever = faiss.as_retriever(search_kwargs = {"k" : 1})

vectorStoreRetrieverMemory = VectorStoreRetrieverMemory(retriever = vectorStoreRetriever)

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

import faiss

from dotenv import load_dotenv

from langchain_openai import OpenAIEmbeddings

from langchain.docstore import InMemoryDocstore

from langchain.vectorstores import FAISS

from langchain.memory import VectorStoreRetrieverMemory

load_dotenv()

openAIEmbeddings = OpenAIEmbeddings()

indexFlatL2 = faiss.IndexFlatL2(1536)

faiss = FAISS(openAIEmbeddings, indexFlatL2, InMemoryDocstore({}), {})

vectorStoreRetriever = faiss.as_retriever(search_kwargs = {"k" : 1})

vectorStoreRetrieverMemory = VectorStoreRetrieverMemory(retriever = vectorStoreRetriever)

※ pip install python-dotenv langchain-community langchain_openai faiss-gpu 명령을 실행했다.

Post Views: 4

AI FAISS LANGCHAIN LLM MEMORY OPENAI PYTHON VECTOR STORE

icodebroker

[PYTHON/LANGCHAIN] VectorStoreRetrieverMemory 클래스 : 생성자에서 retriever 인자를 사용해 VectorStoreRetrieverMemory 객체 만들기

분류

보관함