ragV98 commited on
Commit
a092d54
·
1 Parent(s): d75c09b

retrival to retriver

Browse files
components/generators/daily_feed.py CHANGED
@@ -8,7 +8,7 @@ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
8
  import redis
9
  from typing import List, Dict
10
  from llama_index.core import VectorStoreIndex
11
- from llama_index.core.query_engine import RetrievalQueryEngine
12
  from components.indexers.news_indexer import load_news_index
13
 
14
  # Load environment variables
@@ -66,7 +66,7 @@ def summarize_topic(docs: List[str], topic: str) -> List[Dict]:
66
  # Main generation pipeline
67
  def generate_and_cache_daily_feed():
68
  index: VectorStoreIndex = load_news_index()
69
- query_engine = RetrievalQueryEngine.from_args(index)
70
 
71
  final_feed = []
72
  for topic in TOPICS:
 
8
  import redis
9
  from typing import List, Dict
10
  from llama_index.core import VectorStoreIndex
11
+ from llama_index.core.query_engine import RetrieverQueryEngine
12
  from components.indexers.news_indexer import load_news_index
13
 
14
  # Load environment variables
 
66
  # Main generation pipeline
67
  def generate_and_cache_daily_feed():
68
  index: VectorStoreIndex = load_news_index()
69
+ query_engine = RetrieverQueryEngine.from_args(index)
70
 
71
  final_feed = []
72
  for topic in TOPICS: