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
|
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 =
|
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:
|