Spaces:
Runtime error
Runtime error
v0.1.3
Browse files- src/chromaIntf.py +8 -0
- src/main.py +2 -2
src/chromaIntf.py
CHANGED
@@ -83,9 +83,13 @@ class ChromaIntf():
|
|
83 |
self.retriever.search_kwargs[key]=kwargs[key]
|
84 |
except:
|
85 |
print("setting search args failed")
|
|
|
86 |
retVal=await self.retriever.aget_relevant_documents(query)
|
87 |
value=[]
|
88 |
excludeMeta=True
|
|
|
|
|
|
|
89 |
try:
|
90 |
for item in retVal:
|
91 |
if excludeMeta:
|
@@ -96,8 +100,10 @@ class ChromaIntf():
|
|
96 |
if key != "ID":
|
97 |
v+=key+":"+str(item.metadata[key])+" "
|
98 |
value.append(v)
|
|
|
99 |
self.db_interface.add_to_cache(input=query,value=value)
|
100 |
except:
|
|
|
101 |
for item in retVal:
|
102 |
if excludeMeta:
|
103 |
v=item.page_content+" \n"
|
@@ -107,7 +113,9 @@ class ChromaIntf():
|
|
107 |
if key != "ID":
|
108 |
v+=key+":"+str(item.metadata[key])+" "
|
109 |
value.append(v)
|
|
|
110 |
self.db_interface.add_to_cache(input=query,value=value)
|
|
|
111 |
return retVal
|
112 |
|
113 |
|
|
|
83 |
self.retriever.search_kwargs[key]=kwargs[key]
|
84 |
except:
|
85 |
print("setting search args failed")
|
86 |
+
print("reaching step2")
|
87 |
retVal=await self.retriever.aget_relevant_documents(query)
|
88 |
value=[]
|
89 |
excludeMeta=True
|
90 |
+
print("reaching step3")
|
91 |
+
print(",".join(retVal))
|
92 |
+
print("reaching step4")
|
93 |
try:
|
94 |
for item in retVal:
|
95 |
if excludeMeta:
|
|
|
100 |
if key != "ID":
|
101 |
v+=key+":"+str(item.metadata[key])+" "
|
102 |
value.append(v)
|
103 |
+
print("reaching step5")
|
104 |
self.db_interface.add_to_cache(input=query,value=value)
|
105 |
except:
|
106 |
+
print("reaching step6")
|
107 |
for item in retVal:
|
108 |
if excludeMeta:
|
109 |
v=item.page_content+" \n"
|
|
|
113 |
if key != "ID":
|
114 |
v+=key+":"+str(item.metadata[key])+" "
|
115 |
value.append(v)
|
116 |
+
print("reaching step7")
|
117 |
self.db_interface.add_to_cache(input=query,value=value)
|
118 |
+
print("reaching step8")
|
119 |
return retVal
|
120 |
|
121 |
|
src/main.py
CHANGED
@@ -86,11 +86,11 @@ async def get_matching_docs_sync(inStr: str, kwargs: Dict [Any, Any]) -> Any:
|
|
86 |
TODO: Add parameter to return the source information as well
|
87 |
"""
|
88 |
logger.info("================================================")
|
89 |
-
logger.info("Received new request at /
|
90 |
retVal=await chromaIntf.getRelevantDocs(inStr,kwargs)
|
91 |
#background_tasks.add_task(chromaIntf.getRelevantDocs,inStr,kwargs)
|
92 |
#retVal="added to queue"
|
93 |
-
logger.info("Returning:"+retVal)
|
94 |
logger.info("================================================")
|
95 |
return retVal
|
96 |
|
|
|
86 |
TODO: Add parameter to return the source information as well
|
87 |
"""
|
88 |
logger.info("================================================")
|
89 |
+
logger.info("Received new request at /getMatchingDocsSync:"+inStr)
|
90 |
retVal=await chromaIntf.getRelevantDocs(inStr,kwargs)
|
91 |
#background_tasks.add_task(chromaIntf.getRelevantDocs,inStr,kwargs)
|
92 |
#retVal="added to queue"
|
93 |
+
logger.info("Returning:"+",,".join(retVal))
|
94 |
logger.info("================================================")
|
95 |
return retVal
|
96 |
|