Spaces:
Runtime error
Runtime error
Update ChromaAgenticRAG.py
Browse files- ChromaAgenticRAG.py +8 -3
ChromaAgenticRAG.py
CHANGED
|
@@ -15,7 +15,7 @@ from langchain.tools.retriever import create_retriever_tool
|
|
| 15 |
import json
|
| 16 |
from langchain.vectorstores import Chroma
|
| 17 |
from langchain.embeddings import HuggingFaceEmbeddings
|
| 18 |
-
|
| 19 |
|
| 20 |
load_dotenv()
|
| 21 |
|
|
@@ -145,8 +145,13 @@ sys_msg = SystemMessage(content=system_prompt)
|
|
| 145 |
|
| 146 |
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
|
| 147 |
|
| 148 |
-
|
| 149 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 150 |
|
| 151 |
documents = []
|
| 152 |
for sample in json_QA:
|
|
|
|
| 15 |
import json
|
| 16 |
from langchain.vectorstores import Chroma
|
| 17 |
from langchain.embeddings import HuggingFaceEmbeddings
|
| 18 |
+
from langchain.schema import Document
|
| 19 |
|
| 20 |
load_dotenv()
|
| 21 |
|
|
|
|
| 145 |
|
| 146 |
embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
|
| 147 |
|
| 148 |
+
with open('metadata.jsonl', 'r') as jsonl_file:
|
| 149 |
+
json_list = list(jsonl_file)
|
| 150 |
+
|
| 151 |
+
json_QA = []
|
| 152 |
+
for json_str in json_list:
|
| 153 |
+
json_data = json.loads(json_str)
|
| 154 |
+
json_QA.append(json_data)
|
| 155 |
|
| 156 |
documents = []
|
| 157 |
for sample in json_QA:
|