Merge pull request #15 from almutareb/add-answer-qa-chain

.gitignore

@@ -164,3 +164,8 @@ cython_debug/
 *.sqlite3
 *.bin
 *.pickle
+
+
+# Databases
+
+*.db

app_gui.py

@@ -1,10 +1,13 @@
 # Import Gradio for UI, along with other necessary libraries
 import gradio as gr
 from fastapi import FastAPI
-from rag_app.agents.react_agent import agent_executor
+from rag_app.agents.react_agent import agent_executor, llm
+from rag_app.chains import user_response_sentiment_prompt
 # need to import the qa!
 
 app = FastAPI()
+user_sentiment_chain = user_response_sentiment_prompt | llm
+
 
 if __name__ == "__main__":
 
@@ -32,6 +35,10 @@ if __name__ == "__main__":
     def infer(question, history):
         # Use the question and history to query the RAG model
         #result = qa({"query": question, "history": history, "question": question})
+        try:
+            data = user_sentiment_chain.invoke({"user_reponse":question})
+        except Exception as e:
+            raise e
         try:
             result = agent_executor.invoke(
                 {
@@ -39,6 +46,8 @@ if __name__ == "__main__":
                     "chat_history": history
                 }
             )
+            
+           
             return result
         except Exception:
             raise gr.Error("Model is Overloaded, Please retry later!")

rag_app/chains/__init__.py

@@ -0,0 +1 @@
+from rag_app.chains.user_response_sentiment_chain import user_response_sentiment_prompt

rag_app/chains/user_response_sentiment_chain.py

@@ -0,0 +1,42 @@
+from langchain_core.prompts import PromptTemplate
+
+
+user_response_sentiment_template = """ 
+You will be given a user response to an agent.
+=================
+{user_reponse} 
+====================
+You must determine if the user has has their questions answered. 
+If the user seems satisfied respond saying "1" or "0" ONLY.
+
+Examples:
+================
+
+Example 1
+USER: Great Work! 
+YOUR RESPONSE: 1
+=================
+
+USER: I still need help! 
+YOUR RESPONSE: 0
+Example 2
+================================
+
+USER: I don't understand what you mean 
+YOUR RESPONSE: 0
+Example 3
+================================
+
+USER: That makes sense! 
+YOUR RESPONSE: 1
+Example 4
+================================
+
+  
+
+"""
+
+user_response_sentiment_prompt = PromptTemplate.from_template(user_response_sentiment_template)
+
+
+# llm_chain = prompt | llms