denizkaya2022 commited on
Commit
f1cee88
·
verified ·
1 Parent(s): ea971fb

Update src/rag_pipeline.py

Browse files
Files changed (1) hide show
  1. src/rag_pipeline.py +1 -1
src/rag_pipeline.py CHANGED
@@ -29,7 +29,7 @@ RERANK_MODEL = "BAAI/bge-reranker-base"
29
  EMBEDDINGS_FILE = os.path.join(BASE_DIR, "data", "embeddings.npy")
30
  CHUNKS_FILE = os.path.join(BASE_DIR, "data", "all_chunks.json") # child chunks
31
  PARENTS_FILE = os.path.join(BASE_DIR, "data", "parent_chunks.json") # parent chunks
32
- LLM_MODEL = "openai/gpt-oss-120b"
33
 
34
  BM25_TOP_N = 20
35
  SEMANTIC_TOP_N = 20
 
29
  EMBEDDINGS_FILE = os.path.join(BASE_DIR, "data", "embeddings.npy")
30
  CHUNKS_FILE = os.path.join(BASE_DIR, "data", "all_chunks.json") # child chunks
31
  PARENTS_FILE = os.path.join(BASE_DIR, "data", "parent_chunks.json") # parent chunks
32
+ LLM_MODEL = "gemini-2.5-flash-lite"
33
 
34
  BM25_TOP_N = 20
35
  SEMANTIC_TOP_N = 20