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)
|