debug

app.py

@@ -377,17 +377,22 @@ try:
         # Initial the the sentence transformer and encode the query prompt.
         logger.debug(f"#### Encode text query prompt to create vectors. {promptText}")
         model = SentenceTransformer('/app/multi-qa-MiniLM-L6-cos-v1')
-
         vector = model.encode(promptText)
-        wrks = str(vector)
-        logger.debug(f"### vector: {wrks}")
-        vectorList = []
+        
+        logLevel = logger.getEffectiveLevel()
+        if logLevel >= logging.DEBUG:
+            wrks = str(vector)
+            logger.debug(f"### vector: {wrks}")
 
-        logger.debug("#### Print vectors.")
+        
+        vectorList = []
         for vec in vector:
             vectorList.append(vec)
-        wrks = str(vectorList)
-        logger.debug(f"vectorList: {wrks}")
+        
+        if logLevel >= logging.DEBUG:
+            logger.debug("#### Print vectors.")
+            wrks = str(vectorList)
+            logger.debug(f"vectorList: {wrks}")
 
         # Fetch chunks and print chunks.
         logger.debug("#### Retrieve semchunks from db using vectors from prompt.")
@@ -397,8 +402,10 @@ try:
             distance=0.7,
             limit=3
         )
-        wrks = str(semChunks)
-        logger.debug(f"### semChunks[0]: {wrks}")
+        
+        if logLevel >= logging.DEBUG:
+            wrks = str(semChunks)
+            logger.debug(f"### semChunks[0]: {wrks}")
 
         # Print chunks, corresponding document and document title.
         ragData = ""
@@ -425,12 +432,11 @@ try:
         
         chunksCollection = client.collections.get("Chunks")
         response = chunksCollection.query.fetch_objects()
-
         wstrObjs = str(response.objects)
         logger.debug(f"### response.objects: {wstrObjs}")
         for o in response.objects:
             wstr = o.properties
-            logger.info(f"### o.properties: {wstr}") 
+            logger.debug(f"### o.properties: {wstr}") 
         logger.info("#### getAllRagData() exited.")            
         return  wstrObjs
 
@@ -495,7 +501,7 @@ try:
            #stop=stop,
         )
         result = modelOutput["choices"][0]["message"]["content"]
-        result = str(modelOutput)
+        #result = str(modelOutput)
         logger.debug(f"### llmResult: {result}")
         logger.info("### runLLM exited.")
         return result
@@ -555,7 +561,7 @@ try:
             logger.info("### prettyPrint exited.")
             return prettyText
         except Exception as e:
-            print(f"### prettyPrint() e: {e}")
+            logger.error(f"### prettyPrint() e: {e}")
             return None
 
 
@@ -569,14 +575,16 @@ try:
         logger.debug(f"sysTAtext: {st.session_state.sysTAtext}")
         
         #st.session_state.userpTAtext = st.session_state.userpTA
-        wrkList = setPrompt(st.session_state.userpTA,st.selectRag)
+        wrklist = setPrompt(st.session_state.userpTA,st.selectRag)
         #wrkList = [ wrkList ]
-        st.session_state.userpTA = prettyPrint(wrkList)
+        #st.session_state.userpTA = prettyPrint(wrkList)
+        st.session_state.userpTA = wrklist[1]["content"]
         logger.debug(f"userpTAtext: {st.session_state.userpTA}")
         
         #st.session_state.rspTAtext = runLLM(st.session_state.userpTAtext)
-        rsp = runLLM(wrkList)
-        st.session_state.rspTA = prettyPrint(rsp)
+        rsp = runLLM(wrklist)
+        #st.session_state.rspTA = prettyPrint(rsp)
+        st.session_state.rspTA = rsp
         logger.debug(f"rspTAtext: {st.session_state.rspTA}")
         
         logger.info("### on_runLLMButton_Clicked exited.")