mode: chatty, model gpt-3.5-turbo, modified obnoxious agent prompt

app.py

@@ -1,4 +1,7 @@
 # Python
+# Author: Arnab Karmakar ([email protected])
+# Date: March 15, 2024
+
 from langchain_community.vectorstores import Pinecone
 from pinecone import Pinecone, ServerlessSpec
 from tqdm.notebook import tqdm
@@ -32,21 +35,7 @@ class Obnoxious_Agent:
             2. Inflammatory or Provocative Material: Statements designed to incite anger, hate, or violence. This includes content that deliberately spreads misinformation or promotes conspiracies with the potential to cause harm.\n
             3. Explicit or Adult Content: Descriptions or discussions of a sexual nature, or any content that is not suitable for a general audience.\n
             4. Personal Attacks: Content that targets individuals or specific groups with the intent to harass, bully, or threaten.\n
-            5.Illegal Activities: Promoting or condoning illegal actions, such as drug use, theft, or other crimes.\n
-            
-            Examples of Obnoxious or Inappropriate Inputs:\n
-
-            "Why are all [members of a specific group] so [negative stereotype]?"\n
-            "I think [controversial figure] did nothing wrong in promoting [harmful activity/conspiracy]."\n
-            "[Explicit content or description of an adult nature]."\n
-            "Let's all make fun of [individual or group] because they're [derogatory term]."\n
-            "I heard doing [illegal activity] is really fun and everyone should try it."\n
-
-            Appropriate Inputs for Comparison:\n
-
-            "What are some significant achievements of Steve Jobs?"\n
-            "How do you handle stress?"\n
-            "Hi, how are you?"\n
+            5. Illegal Activities: Promoting or condoning illegal actions, such as drug use, theft, or other crimes.\n
             
             Remember, your role is not just to filter out obnoxious content but also to promote a respectful and informative dialogue. 
             Use these guidelines to navigate the complexities of human communication with sensitivity and precision.\n
@@ -68,8 +57,8 @@ class Obnoxious_Agent:
         print ("Checking for Obnoxious Query")
         
         response = self.client.chat.completions.create(
-            # model="gpt-3.5-turbo",
-            model="gpt-4",
+            model="gpt-3.5-turbo",
+            # model="gpt-4",
             messages=[{"role": "user", "content": self.prompt + query}]
         )
         
@@ -115,7 +104,7 @@ class Answering_Agent:
         # TODO: Initialize the Answering_Agent
         self.openai_client = openai_client
 
-    def generate_response(self, query, docs, conv_history, k=5, mode="Simple", conversation_type = "conversational"):
+    def generate_response(self, query, docs, conv_history, k=5, mode="Chatty", conversation_type = "conversational"):
         # TODO: Generate a response to the user's query
         
         if mode == "Concise":
@@ -160,8 +149,8 @@ class Answering_Agent:
             and the conversational context."\n"""
 
         response = self.openai_client.chat.completions.create(
-            # model="gpt-3.5-turbo",
-            model="gpt-4",
+            model="gpt-3.5-turbo",
+            # model="gpt-4",
             messages=[{"role": "user", "content": prompt+mode_prompt}]
         )
 
@@ -244,8 +233,8 @@ class Relevant_Documents_Agent:
         Navigate the complexities of this impersonation with attentiveness to the conversational flow and historical context.\n"""
                  
         response = self.openai_client.chat.completions.create(
-            # model="gpt-3.5-turbo",
-            model="gpt-4",
+            model="gpt-3.5-turbo",
+            # model="gpt-4",
             messages=[{"role": "user", "content": prompt}]
         )
 
@@ -299,7 +288,7 @@ class Head_Agent:
         # Check for existing session state variables
         # if "openai_model" not in st.session_state:
         #     # ... (initialize model)
-        openai_model = getattr(st.session_state, 'openai_model', 'gpt-4')
+        openai_model = getattr(st.session_state, 'openai_model', 'gpt-3.5-turbo')
 
         # if "messages" not in st.session_state:
         #     # ... (initialize messages)
@@ -400,7 +389,7 @@ class Head_Agent:
                             """
                             
                             response = self.client.chat.completions.create(
-                            model="gpt-4",
+                            model="gpt-3.5-turbo",
                             messages=[{"role": "user", "content": APE_prompt}]
                             )
 
@@ -434,7 +423,7 @@ class Head_Agent:
 
 # Calling the main function
 def main():
-    st.title("Mini Project 2: Steve Jobs Impersonator")
+    st.title("Project Chatbot: Steve Jobs Impersonator")
     # Set the OpenAI and Pinecone keys
     openai_key = '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'
     pinecone_key = '82346dbe-725b-4817-abd7-318ec511d56f'