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

Update rag_routerv2.py

Browse files
Files changed (1) hide show
  1. 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("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)
 
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)