pvanand commited on
Commit
843a697
·
verified ·
1 Parent(s): d8ec593

Update rag_routerv2.py

Browse files
Files changed (1) hide show
  1. rag_routerv2.py +3 -3
rag_routerv2.py CHANGED
@@ -195,10 +195,10 @@ async def startup():
195
  )
196
  # load digiyatra csv and create node for each row using csv.reader
197
  with open("combined_digi_yatra.csv", "r") as file:
198
- reader = csv.reader(file)
199
  nodes = []
200
- for row in reader:
201
- node = TextNode(text=row, id_=str(uuid.uuid4()))
202
  nodes.append(node)
203
 
204
  index = VectorStoreIndex(nodes, vector_store=vector_store)
 
195
  )
196
  # load digiyatra csv and create node for each row using csv.reader
197
  with open("combined_digi_yatra.csv", "r") as file:
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)
203
 
204
  index = VectorStoreIndex(nodes, vector_store=vector_store)