3차시. RAG 기본 구조 깊이 있게 이해하기

7. 실제 코드 실행 흐름 미리보기

출처: 원본 PDF 03차시_RAG기본구조깊이있게이해하기.pdf 10~11쪽 · 쪽 그림 10쪽 11쪽

다음 차시부터 실제 코드를 작성하게 됩니다. 코드의 흐름을 미리 이해하면 훨씬 쉽게 배울 수 있습니다.

7-1) 준비 단계 코드 구조

Python

# 1. 문서 로딩
documents = load_documents("files/")

# 2. 텍스트 분할
chunks = split_text(documents, chunk_size=500)

# 3. 임베딩
embeddings = embed_chunks(chunks)

# 4. 벡터 DB에 저장
vector_db = save_to_vectordb(chunks, embeddings)

7-2) 실행 단계 코드 구조

출처: 원본 PDF 11쪽

Python

# 1. 사용자 질문 입력
query = "우리 제품의 색상은?"

# 2. 질문을 벡터로 변환
query_vector = embed_query(query)

# 3. 유사도 검색
top_chunks = vector_db.search(query_vector, top_k=3)

# 4. Context 생성 및 프롬프트 조합
context = combine_chunks(top_chunks)
prompt = create_prompt(context, query)

# 5. LLM 호출
answer = llm.generate(prompt)

# 6. 결과 반환
print(answer)

실행 결과 예시 (코드를 읽고 만든 예상 화면입니다. 실제 실행 결과, 특히 모델의 답은 다를 수 있습니다)

출력

우리 제품의 색상은 검은색과 흰색입니다.