hugging2021 commited on
Commit
801b168
·
verified ·
1 Parent(s): c623396

Update rag_system.py

Browse files
Files changed (1) hide show
  1. rag_system.py +4 -4
rag_system.py CHANGED
@@ -1,13 +1,13 @@
1
  import os
2
  import argparse
3
  import sys
4
- from langchain.chains import RetrievalQA
5
- from langchain.prompts import PromptTemplate
6
  from vector_store import get_embeddings, load_vector_store
7
  from llm_loader import load_llama_model
8
 
9
- def create_refine_prompts_with_pages(language="ko"):
10
- if language == "ko":
11
  question_prompt = PromptTemplate(
12
  input_variables=["context_str", "question"],
13
  template="""
 
1
  import os
2
  import argparse
3
  import sys
4
+ from langchain_community.chains import RetrievalQA
5
+ from langchain_communit.prompts import PromptTemplate
6
  from vector_store import get_embeddings, load_vector_store
7
  from llm_loader import load_llama_model
8
 
9
+ def create_refine_prompts_with_pages(language="de"):
10
+ if language == "de":
11
  question_prompt = PromptTemplate(
12
  input_variables=["context_str", "question"],
13
  template="""