anubhav77 commited on
Commit
bb22bf8
·
1 Parent(s): 34de5cc
Files changed (1) hide show
  1. 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.get_relevant_documents(query)
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
- metadata['ID']=metadata['timestamp'].strftime("%Y-%m-%d %H:%M:%S::%f")+"-conversation"
 
 
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"))