Spaces:
Runtime error
Runtime error
v0.8.4
Browse files- src/chromaIntf.py +4 -2
src/chromaIntf.py
CHANGED
|
@@ -75,7 +75,7 @@ class ChromaIntf():
|
|
| 75 |
print("retriver state",self.retriever.search_kwargs)
|
| 76 |
print("retriver state",self.retriever.search_type)
|
| 77 |
self.retriever.search_kwargs["k"]=count
|
| 78 |
-
retVal=await self.retriever.
|
| 79 |
value=[]
|
| 80 |
excludeMeta=True
|
| 81 |
try:
|
|
@@ -109,7 +109,9 @@ class ChromaIntf():
|
|
| 109 |
pass
|
| 110 |
if "source" not in metadata.keys():
|
| 111 |
metadata['source']="conversation"
|
| 112 |
-
|
|
|
|
|
|
|
| 113 |
metadata['Year']=metadata['timestamp'].year
|
| 114 |
metadata['Month']=metadata['timestamp'].month
|
| 115 |
metadata['Day']=int(metadata['timestamp'].strftime("%d"))
|
|
|
|
| 75 |
print("retriver state",self.retriever.search_kwargs)
|
| 76 |
print("retriver state",self.retriever.search_type)
|
| 77 |
self.retriever.search_kwargs["k"]=count
|
| 78 |
+
retVal=await self.retriever.aget_relevant_documents(query)
|
| 79 |
value=[]
|
| 80 |
excludeMeta=True
|
| 81 |
try:
|
|
|
|
| 109 |
pass
|
| 110 |
if "source" not in metadata.keys():
|
| 111 |
metadata['source']="conversation"
|
| 112 |
+
#TODO: If url is present in input or when the splitting need to be done, then we'll need to change how we
|
| 113 |
+
# formulate the ID and may be filename to store information
|
| 114 |
+
metadata['ID']=metadata['timestamp'].strftime("%Y-%m-%d %H-%M-%S")+"-"+metadata['source']
|
| 115 |
metadata['Year']=metadata['timestamp'].year
|
| 116 |
metadata['Month']=metadata['timestamp'].month
|
| 117 |
metadata['Day']=int(metadata['timestamp'].strftime("%d"))
|