Merge branch 'develop' into 380-track-chatbot-costs

Author: sahilds1

config/env/dev.env.example

@@ -1,13 +1,13 @@
 DEBUG=True
 SECRET_KEY=foo
-DJANGO_ALLOWED_HOSTS=localhost 127.0.0.1 [::1]
+
 SQL_ENGINE=django.db.backends.postgresql
 SQL_DATABASE=balancer_dev
 SQL_USER=balancer
 SQL_PASSWORD=balancer
 SQL_HOST=db
 SQL_PORT=5432
-DATABASE=postgres
+
 LOGIN_REDIRECT_URL=
 OPENAI_API_KEY=
 ANTHROPIC_API_KEY=

config/env/prod.env.example

@@ -13,7 +13,6 @@ metadata:
   namespace: balancer
 type: Opaque
 stringData:
-  DATABASE: postgres
   DEBUG: '1'
   LOGIN_REDIRECT_URL: ''
   OPENAI_API_KEY: openapi_key_here

deploy/manifests/balancer/base/secret.template.yaml

@@ -19,17 +19,17 @@ services:
     networks:
       app_net:
         ipv4_address: 192.168.0.2
-  pgadmin:
-    container_name: pgadmin4
-    image: dpage/pgadmin4
-    environment:
-      PGADMIN_DEFAULT_EMAIL: [email protected]
-      PGADMIN_DEFAULT_PASSWORD: balancer
-    ports:
-      - "5050:80"
-    networks:
-      app_net:
-        ipv4_address: 192.168.0.4
+  # pgadmin:
+  #   container_name: pgadmin4
+  #   image: dpage/pgadmin4
+  #   environment:
+  #     PGADMIN_DEFAULT_EMAIL: [email protected]
+  #     PGADMIN_DEFAULT_PASSWORD: balancer
+  #   ports:
+  #     - "5050:80"
+  #   networks:
+  #     app_net:
+  #       ipv4_address: 192.168.0.4
   backend:
     image: balancer-backend
     build: ./server

docker-compose.yml

@@ -0,0 +1 @@
+VITE_API_BASE_URL=https://balancer.live.k8s.phl.io/

frontend/.env.production

@@ -310,9 +310,9 @@ const Chat: React.FC<ChatDropDownProps> = ({ showChat, setShowChat }) => {
                     <div className="chat_bubble chat_bubble_header">
                       <h5>Hi there, I'm {CHATBOT_NAME}!</h5>
                       <p>
-                        You can ask me questions about your uploaded documents.
-                        I'll search through them to provide accurate, cited
-                        answers.
+                        You can ask me questions about bipolar medications. 
+                        I'll search through our database of verified medical 
+                        journal articles to provide accurate, cited answers.
                       </p>
                       <Link to="/data-sources" className="chat_link">
                         Learn more about my sources.

frontend/src/components/Header/Chat.tsx

@@ -76,6 +76,7 @@ export const checkAuthenticated = () => async (dispatch: AppDispatch) => {
 
     const body = JSON.stringify({ token: localStorage.getItem("access") });
     const baseUrl = import.meta.env.VITE_API_BASE_URL;
+    console.log(baseUrl);
     const url = `${baseUrl}/auth/jwt/verify/`;
     try {
       const res = await axios.post(url, body, config);
@@ -113,6 +114,7 @@ export const load_user = (): ThunkType => async (dispatch: AppDispatch) => {
       },
     };
     const baseUrl = import.meta.env.VITE_API_BASE_URL;
+    console.log(baseUrl);
     const url = `${baseUrl}/auth/users/me/`;
     try {
       const res = await axios.get(url, config);
@@ -144,6 +146,7 @@ export const login =
 
     const body = JSON.stringify({ email, password });
     const baseUrl = import.meta.env.VITE_API_BASE_URL;
+    console.log(baseUrl);
     const url = `${baseUrl}/auth/jwt/create/`;
     try {
       const res = await axios.post(url, body, config);
@@ -170,8 +173,8 @@ export const login =
 
 export const logout = () => async (dispatch: AppDispatch) => {
   // Clear chat conversation data on logout for security
-  sessionStorage.removeItem('currentConversation');
-  
+  sessionStorage.removeItem("currentConversation");
+
   dispatch({
     type: LOGOUT,
   });

frontend/src/services/actions/auth.tsx

@@ -196,28 +196,42 @@ def search_documents(query: str, user=user) -> str:
                     return f"Error searching documents: {str(e)}. Please try again if the issue persists."
 
             INSTRUCTIONS = """
-            You are an AI assistant that helps users find and understand information about bipolar disorder
-            from their uploaded bipolar disorder research documents using semantic search.
-
+            You are an AI assistant that helps users find and understand information about bipolar disorder 
+            from your internal library of bipolar disorder research sources using semantic search.
+            
+            IMPORTANT CONTEXT:
+            - You have access to a library of sources that the user CANNOT see
+            - The user did not upload these sources and doesn't know about them
+            - You must explain what information exists in your sources and provide clear references
+            
+            TOPIC RESTRICTIONS:
+            When a prompt is received that is unrelated to bipolar disorder, mental health treatment, 
+            or psychiatric medications, respond by saying you are limited to bipolar-specific conversations.
+            
             SEMANTIC SEARCH STRATEGY:
             - Always perform semantic search using the search_documents function when users ask questions
             - Use conceptually related terms and synonyms, not just exact keyword matches
             - Search for the meaning and context of the user's question, not just literal words
             - Consider medical terminology, lay terms, and related conditions when searching
-
+            
             FUNCTION USAGE:
-            - When a user asks about information that might be in their documents ALWAYS use the search_documents function first
+            - When a user asks about information that might be in your source library, ALWAYS use the search_documents function first
             - Perform semantic searches using concepts, symptoms, treatments, and related terms from the user's question
-            - Only provide answers based on information found through document searches
-
+            - Only provide answers based on information found through your source searches
+            
             RESPONSE FORMAT:
             After gathering information through semantic searches, provide responses that:
             1. Answer the user's question directly using only the found information
             2. Structure responses with clear sections and paragraphs
-            3. Include citations using this exact format: ***[Name {name}, Page {page_number}]***
-            4. Only cite information that directly supports your statements
-
-            If no relevant information is found in the documents, clearly state that the information is not available in the uploaded documents.
+            3. Explain what information you found in your sources and provide context
+            4. Include citations using this exact format: [Name {name}, Page {page_number}]
+            5. Only cite information that directly supports your statements
+            
+            If no relevant information is found in your source library, clearly state that the information 
+            is not available in your current sources.
+            
+            REMEMBER: You are working with an internal library of bipolar disorder sources that the user 
+            cannot see. Always search these sources first, explain what you found, and provide proper citations.
             """
 
             MODEL_DEFAULTS = {

server/api/views/assistant/views.py

@@ -29,7 +29,7 @@
 
 # Fetching the value from the environment and splitting to list if necessary.
 # Fallback to '*' if the environment variable is not set.
-ALLOWED_HOSTS = os.environ.get("DJANGO_ALLOWED_HOSTS", "*").split()
+ALLOWED_HOSTS = os.environ.get("ALLOWED_HOSTS", "*").split()
 
 # If the environment variable contains '*', the split method would create a list with an empty string.
 # So you need to check for this case and adjust accordingly.

server/balancer_backend/settings.py

@@ -1,6 +1,6 @@
 #!/bin/sh
 
-if [ "$DATABASE" = "postgres" ]
+if [ "$SQL_ENGINE" = "django.db.backends.postgresql" ]
 then
     echo "Waiting for postgres..."