Spaces:
Sleeping
Sleeping
Update rag_routerv2.py
Browse files- rag_routerv2.py +3 -2
rag_routerv2.py
CHANGED
@@ -186,6 +186,7 @@ async def startup():
|
|
186 |
print("RAG Router started")
|
187 |
from llama_index.core.schema import TextNode
|
188 |
table_name = "digiyatra"
|
|
|
189 |
vector_store = LanceDBVectorStore(
|
190 |
|
191 |
uri="./lancedb/dev",
|
@@ -194,9 +195,9 @@ async def startup():
|
|
194 |
query_type="hybrid"
|
195 |
)
|
196 |
# load digiyatra csv and create node for each row using csv.reader
|
197 |
-
with open(
|
|
|
198 |
data = list(reader)
|
199 |
-
nodes = []
|
200 |
for row in data[1:]:
|
201 |
node = TextNode(text=str(row), id_=str(uuid.uuid4()))
|
202 |
nodes.append(node)
|
|
|
186 |
print("RAG Router started")
|
187 |
from llama_index.core.schema import TextNode
|
188 |
table_name = "digiyatra"
|
189 |
+
nodes = []
|
190 |
vector_store = LanceDBVectorStore(
|
191 |
|
192 |
uri="./lancedb/dev",
|
|
|
195 |
query_type="hybrid"
|
196 |
)
|
197 |
# load digiyatra csv and create node for each row using csv.reader
|
198 |
+
with open('combined_digi_yatra.csv', newline='') as f:
|
199 |
+
reader = csv.reader(f)
|
200 |
data = list(reader)
|
|
|
201 |
for row in data[1:]:
|
202 |
node = TextNode(text=str(row), id_=str(uuid.uuid4()))
|
203 |
nodes.append(node)
|