Spaces:
Running
Running
MVPilgrim
commited on
Commit
·
0c173aa
1
Parent(s):
c1cc1e0
debug
Browse files
app.py
CHANGED
|
@@ -406,6 +406,21 @@ try:
|
|
| 406 |
#collection = client.collections.get("Chunks")
|
| 407 |
logger.info("#### getRagData() exited.")
|
| 408 |
return ragData
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 409 |
|
| 410 |
|
| 411 |
# Display UI
|
|
@@ -493,9 +508,9 @@ try:
|
|
| 493 |
return fullPrompt
|
| 494 |
|
| 495 |
|
| 496 |
-
def
|
| 497 |
logger = st.session_state.logger
|
| 498 |
-
logger.info("###
|
| 499 |
st.session_state.sysTAtext = st.session_state.sysTA
|
| 500 |
logger.info(f"sysTAtext: {st.session_state.sysTAtext}")
|
| 501 |
|
|
@@ -509,12 +524,19 @@ try:
|
|
| 509 |
st.session_state.rspTA = rsp
|
| 510 |
logger.info(f"rspTAtext: {st.session_state.rspTA}")
|
| 511 |
|
| 512 |
-
logger.info("###
|
| 513 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 514 |
|
| 515 |
with st.sidebar:
|
| 516 |
-
st.selectRag
|
| 517 |
-
st.
|
|
|
|
| 518 |
|
| 519 |
logger.info("#### semsearch.py end of code.")
|
| 520 |
except Exception as e:
|
|
|
|
| 406 |
#collection = client.collections.get("Chunks")
|
| 407 |
logger.info("#### getRagData() exited.")
|
| 408 |
return ragData
|
| 409 |
+
|
| 410 |
+
|
| 411 |
+
def getAllRagData():
|
| 412 |
+
logger.info("#### getAllRagData() entered.")
|
| 413 |
+
|
| 414 |
+
chunksCollection = client.collections.get("Chunks")
|
| 415 |
+
response = chunksCollection.query.fetch_objects()
|
| 416 |
+
|
| 417 |
+
wstrObjs = str(response.objects)
|
| 418 |
+
logger.info(f"### response.objects: {wstrObjs}")
|
| 419 |
+
for o in response.objects:
|
| 420 |
+
wstr = o.properties
|
| 421 |
+
logger.info(f"### o.properties: {wstr}")
|
| 422 |
+
|
| 423 |
+
return wstrObjs
|
| 424 |
|
| 425 |
|
| 426 |
# Display UI
|
|
|
|
| 508 |
return fullPrompt
|
| 509 |
|
| 510 |
|
| 511 |
+
def on_runLLMButton_Clicked():
|
| 512 |
logger = st.session_state.logger
|
| 513 |
+
logger.info("### on_runLLMButton_Clicked entered.")
|
| 514 |
st.session_state.sysTAtext = st.session_state.sysTA
|
| 515 |
logger.info(f"sysTAtext: {st.session_state.sysTAtext}")
|
| 516 |
|
|
|
|
| 524 |
st.session_state.rspTA = rsp
|
| 525 |
logger.info(f"rspTAtext: {st.session_state.rspTA}")
|
| 526 |
|
| 527 |
+
logger.info("### on_runLLMButton_Clicked exited.")
|
| 528 |
|
| 529 |
+
def on_getAllRagDataButton_Clicked():
|
| 530 |
+
logger = st.session_state.logger
|
| 531 |
+
logger.info("### on_getAllRagButton_Clicked entered.")
|
| 532 |
+
st.session_state.ragpTA = getAllRagData()
|
| 533 |
+
|
| 534 |
+
logger.info("### on_getAllRagButton_Clicked entered.")
|
| 535 |
|
| 536 |
with st.sidebar:
|
| 537 |
+
st.selectRag = st.checkbox("Enable Query With RAG",value=False,key="selectRag",help=None,on_change=None,args=None,kwargs=None,disabled=False,label_visibility="visible")
|
| 538 |
+
st.runLLMButton = st.button("Run LLM Query",key=None,help=None,on_click=on_runLLMButton_Clicked,args=None,kwargs=None,type="secondary",disabled=False,use_container_width=False)
|
| 539 |
+
st.getAllRagDataButton = st.button("Get All Rag Data",key=None,help=None,on_click=on_getAllRagDataButton_Clicked,args=None,kwargs=None,type="secondary",disabled=False,use_container_width=False)
|
| 540 |
|
| 541 |
logger.info("#### semsearch.py end of code.")
|
| 542 |
except Exception as e:
|