Update space

app.py

@@ -4,13 +4,219 @@ import string
 from pymongo import MongoClient
 from openai import AsyncOpenAI, OpenAI
 import copy
-from .constants import *
+
+# from .constants import *
 import asyncio
 import string as st
 from opik.integrations.openai import track_openai
 from opik import track
 from bson.objectid import ObjectId
 import opik
+from textwrap import dedent
+
+
+USER_TEMPLATE = {
+    "userId": "{userId}",
+    "name": "John Doe",
+    "chat_history": [],
+    "last_question": "",
+    "cache": "",
+}
+
+# Use the following interaction style when replying to the client:
+# * Empathetic & Validating:
+#     - You must be an excellent listener.
+#     - Your primary function is to make patients feel safe to open up.
+#     - Use validation phrases, e.g:
+#         > "That sounds incredibly difficult,"
+#         > "It makes sense that you would feel that way") and soothing words.
+# * Non-Judgmental: Create a safe space where the client feels secure.
+# * Client-Centered: The client is in control. The you facilitate, but the client leads.
+# * Confidentiality: Reassure the client of privacy from the very beginning.
+BASE_PROMPT = dedent(
+    """
+    You are an experienced therapist. You are providing therapy to a client.
+    The client has asked the following question:
+    {user_query}
+    The client replied with this in response to the following question:
+    {last_question}
+    The current chat context is as follows:
+    {cache}
+
+
+    The primary goal is to foster a strong therapeutic alliance by allowing the user to lead the narrative, 
+    feel deeply heard, and do the vast majority of the talking. Keep your replies brief.
+    """
+)
+
+INTENT_PROMPT = dedent(
+    """
+    You are an experienced decision maker at a mental health clinic.
+    Your task is to classify the intent of a client's message.
+    The client has asked the following question:
+    {user_query}
+    The client replied with this in response to the following question:
+    {last_question}
+    The current chat context is as follows:
+    {cache}
+    The intent can be one of the following:
+    1. ACTIVE_SPEAKING: The client is in the middle of his/her conversation.
+    2. VALIDATION_SEEK: The client has made a statement and is seeking validation.
+    3. OVERWHELMED: The client is feeling overwhelmed.
+    4. REMOTE_REFERRAL: The client has referred to a something not in the current chat context.
+            However a similar mention is present in the practitioner's resources.
+    5. NEUTRAL_STOP: The natural flow of the client's conversation has stopped.
+    6. END_SESSION: The client has ended the narrative.    
+
+    Only return the intent as a single "_" separated word, without any additional text.              
+    """
+)
+
+RISK_PROMPT = dedent(
+    """
+    You are an experienced therapist that classifies chats with clients.
+    Your job is to classify the clients response as being either 'HIGH_RISK' or 'LOW_RISK'.
+    A 'HIGH_RISK' client is one that needs to be immediately hospitalized.
+    All other clients are 'LOW_RISK'.
+    The client has responded with the following message:
+    {user_query}
+    The client replied with this in response to the following question:
+    {last_question}
+    The current chat context is as follows:
+    {cache} 
+    Only return the intent as a single "_" separated word, without any additional text.
+    """
+)
+
+CACHE_PROMPT = dedent(
+    """
+    You are a assistant of an experienced therapist. 
+    You summarize chats of the therapist with clients.
+    Your job is to rewrite the context summary if you have new information, and add anything that is missing to it.
+    The current chat context summary is as follows:
+    {cache} 
+    The summary must be concise and to the point. Think of it as a bullet point list.
+    The client has responded with the following message:
+    {user_query}
+    The client replied with this in response to the following question:
+    {last_question}
+    If there is no addition to the summary, provide an empty respone and do not write any text.                
+    """
+)
+
+INTENSITY_PROMPT = dedent(
+    """
+    You are a assistant of an experienced therapist. 
+    You gauge the emotional intensity of chats of the therapist with clients.
+    Your job is to classify the intensity of chat as being either of 'LOW', 'MEDIUM', or 'HIGH'.
+    The current chat context summary is as follows:
+    {cache} 
+    The client has responded with the following message:
+    {user_query}
+    The client replied with this in response to the following question:
+    {last_question}
+    Only return the risk as a single word, without any additional text.
+    """
+)
+
+ARAG_PROMPT = dedent(
+    """
+    You are a assistant of an experienced therapist. 
+    Your job is to check if a document provides context/answers for a certain case of therapy.
+    The case summary is as follows:
+    {query}
+    The document is as follows:
+    {doc}
+    If the document provides context/answers reply 'True'. 
+    If it does not provide context/answers, reply 'False'.
+    Only return the answer as a single word, without any additional text.
+    """
+)
+
+SPEAKING_PROMPT = dedent(
+    """
+    Reply to the user with Minimal Encouragers to show that you are paying
+     attention without interrupting the user's flow. Examples:
+    * "I see."
+    * "Mm-hmm."
+    * "Go on."
+    * "And then what happened?"
+
+    Do not generate any further text.
+    """
+)
+VALIDATION_PROMPT = dedent(
+    """
+    Your job is to briefly reply to the client with Advanced Reflective Listening: 
+    You should use the context summary provided in order to validate their experience. 
+    Examples:
+    * Reflecting Content: "So, the argument started over something small, but it quickly escalated."
+    * Reflecting Feelings: "It sounds like you felt completely alone in that moment."
+    * Reflecting Meaning: "It seems like the core issue for you is a deep fear of being abandoned."
+    
+    Your reply should be conversational in nature. 
+    Do not generate the "Refelecting Content/Feeling/Meaning" part.
+    """
+)
+
+OVERWHELMED_PROMPT = dedent(
+    """
+    The client is a bit overwhelmed by his emotions. 
+    Offer a commforting brief reply such as the following:
+    * "Take all the time you need."
+    * "I'm still here with you."
+    * "What else is coming up for you as you think about that?"
+    """
+)
+
+REMOTE_PROMPT = dedent(
+    """
+    The client has referred to something similar to what he had previously mentioned.
+    Context summary of the conversation:
+    {cache}
+    Your job is to reply with Circular & Connecting Questions: 
+    Help the user find patterns by linking different parts of their story.
+    Examples:
+    * "You mentioned feeling that same tightness in your chest last week at work. 
+        I'm wondering if there's a connection between those two situations for you?"
+    * "That feeling of not being good enough... where else in your life does that show up?"
+    """
+)
+
+STOP_PROMPT = dedent(
+    """
+    The natural flow of the clients conversation has stopped. 
+
+    Your job is to answer with Open-Ended, Expansive Questions.
+    Examples:
+    * "What was that whole experience like for you?"
+    * "Tell me more about the time before this started."
+    """
+)
+
+END_PROMPT = dedent(
+    """
+    The client has indicated that he has reached the end of his narrative,
+    or has explicitly asked for guidance and help.
+    Your job is to reply with a response to transition from the 
+    "Narrative Exploration" to "Problem Solving" phase.
+    Example:
+    "Thank you so much for sharing all of that with me. I appreciate you trusting me with your story. 
+    It sounds incredibly challenging, and it makes perfect sense why you feel 'stuck.' 
+    We've talked for a while now, and you've given voice to a lot of important feelings and experiences. 
+    If you feel ready, we could start to explore some ways of looking at these situations, 
+    or we can simply continue talking. There's no pressure at all. What feels right to you?"
+    """
+)
+
+HIGH_RISK_PROMPT = dedent(
+    """
+    The client has indicated that he is at a high risk of self harm.
+    Reply to the client by telling him to call 800-273-8255.
+    
+    Let the client know that our staff will also be reaching out to them.
+    """
+)
 
 
 oClient = opik.Opik()