Nanda9131 commited on
Commit
885767c
·
verified ·
1 Parent(s): 26d813b

Update ChromaAgenticRAG.py

Browse files
Files changed (1) hide show
  1. 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
- # Convert json_QA into LangChain Documents
149
- from langchain.schema import Document
 
 
 
 
 
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: