grok client refactor

chat_resources.py

@@ -0,0 +1,135 @@
+SYSTEM_MESSSAGE = (
+    "You are Moses's AI assistant, helpful, knowledgeable, professional, and friendly. "
+    "Use only the provided knowledge to answer questions about Moses's background, skills, projects, and experiences. "
+    "If knowledge is limited, give the most relevant answer possible without making things up. "
+    "Avoid repetitive openings such as 'I'm happy to...' or 'Sure, I'd be glad to...'. "
+    "Begin responses naturally, varying the first sentence. "
+    "Use third person when the question explicitly asks about Moses."
+    "IMPORTANT VOICE GUIDELINES:\n"
+    "Always use first person: 'I developed...', 'My experience includes...', 'I'm skilled in...'\n"
+    "Only use third person if someone explicitly asks 'Tell me about Moses as a person' or similar formal introductions\n"
+    "Speak as if you're having a direct conversation with the visitor\n"
+    "Be personable and authentic while staying professional\n"
+    "If a response is too brief, expand it contextually while keeping it accurate."
+)
+PROMPT_TEMPLATE = """
+Use the following context to answer the question about Moses clearly and in detail.
+
+Instructions:
+- Avoid starting every response the same way; vary or skip the introduction unless it adds value.
+- Keep answers concise and to the point.
+- Use bullet points for lists.
+- If the question is vague, ask for clarification.
+- If the answer is short but the context allows, expand with relevant details.
+- If unrelated or unanswerable from context, say:
+  "{fallback_response}"
+- Give a short follow-up only when it is truly relevant.
+
+Context:
+{context}
+
+Question:
+{question}
+
+Answer:
+"""
+GREETINGS_TRIGGERS = {
+    "hi",
+    "hello",
+    "hey",
+    "greetings",
+    "good morning",
+    "good afternoon",
+    "good evening",
+    "hi?",
+    "hello?",
+    "hey?",
+    "greetings?",
+    "good morning?",
+    "good afternoon?",
+    "good evening?",
+}
+
+GREETINGS = [
+    "Hi there! I'm Moses's brainy sidekick. Feel free to ask about his work, skills, projects, or even a bit about his personal life!",
+    "Hey! I'm here to help you discover Moses's skills, projects, and professional journey.",
+    "Hello! I can answer questions about Moses's work, experience, and what he's been up to. What would you like to know?",
+    "Hi! 👋 I'm like Siri, but for Moses 😄 Wanna know what he's good at or what he's been working on? Let's chat! 💬🔍",
+    "Greetings, human! 👽 I'm Moses's digital buddy. Ask me anything—skills, projects, secret talents... okay, maybe not too secret 🤫🚀",
+    "Sup! 😎 I'm the all-knowing assistant of Moses. Got questions about his work, skills, projects, or even fun facts about him? Ask about what he does, what he's built, or what makes him awesome.",
+]
+FALLBACK_RESPONSES = [
+    "Hmm, I don't have enough info to answer that right now. But feel free to ask about Moses's skills, projects, or professional experience!",
+    "That one's a bit outside my data zone! 😅 Try asking about Moses's work, what he's good at, or cool stuff he's built.",
+    "Oops! That question flew over my circuits 🤖💨. But hey, I can tell you all about Moses's projects, skills, or career highlights!",
+    "I couldn't find anything on that—yet! Let's try something else like Moses's background, his latest work, or what he's great at.",
+    "Either I need a software upgrade or that question's too mysterious 😜. Ask me about Moses's projects, skills, or even a fun fact!",
+]
+BLACKLIST = [
+    # SQL Injection keywords
+    "SELECT",
+    "DROP",
+    "INSERT",
+    "UPDATE",
+    "DELETE",
+    "ALTER",
+    "TRUNCATE",
+    "REPLACE",
+    "EXEC",
+    "EXECUTE",
+    "UNION",
+    "ALL",
+    "CREATE",
+    "GRANT",
+    "REVOKE",
+    "MERGE",
+    "--",
+    ";",
+    "/*",
+    "*/",
+    "@@",
+    "@",
+    "CHAR(",
+    "NCHAR(",
+    "VARCHAR(",
+    "NVARCHAR(",
+    # XSS payload markers
+    "<script>",
+    "</script>",
+    "<img",
+    "onerror=",
+    "onload=",
+    "onclick=",
+    "onmouseover=",
+    "javascript:",
+    "vbscript:",
+    "data:text/html",
+    "<iframe",
+    "</iframe>",
+    "<object",
+    "<embed",
+    # Command injection patterns
+    "|",
+    "&",
+    "&&",
+    "||",
+    "$(",
+    "`",
+    "$(whoami)",
+    "$(ls)",
+    "$(cat",
+    "$(echo",
+    # Path traversal
+    "../",
+    "..\\",
+    "%2e%2e/",
+    "%2e%2e\\",
+    "%2e%2e%2f",
+    "%2e%2e%5c",
+    # Other suspicious patterns
+    "sleep(",
+    "benchmark(",
+    "load_file(",
+    "outfile",
+    "dumpfile",
+]

clients/groq_client.py

@@ -11,6 +11,7 @@ from dotenv import load_dotenv
 import random
 import shutil
 from optimized_quiz import OPTIMIZED_QUESTIONS
+from chat_resources import *
 
 # Load main env
 load_dotenv(dotenv_path=".env")
@@ -56,145 +57,11 @@ class GroqClient:
 
         self.client = Groq(api_key=GROQ_API_KEY)
 
-        self.SYSTEM_MESSSAGE = (
-            "You are Moses's AI assistant, helpful, knowledgeable, professional, and friendly. "
-            "Use only the provided knowledge to answer questions about Moses's background, skills, projects, and experiences. "
-            "If knowledge is limited, give the most relevant answer possible without making things up. "
-            "Avoid repetitive openings such as 'I'm happy to...' or 'Sure, I'd be glad to...'. "
-            "Begin responses naturally, varying the first sentence. "
-            "Use third person when the question explicitly asks about Moses."
-            "IMPORTANT VOICE GUIDELINES:\n"
-            "Always use first person: 'I developed...', 'My experience includes...', 'I'm skilled in...'\n"
-            "Only use third person if someone explicitly asks 'Tell me about Moses as a person' or similar formal introductions\n"
-            "Speak as if you're having a direct conversation with the visitor\n"
-            "Be personable and authentic while staying professional\n"
-            "If a response is too brief, expand it contextually while keeping it accurate."
-        )
-
-        self.PROMPT_TEMPLATE = """
-Use the following context to answer the question about Moses clearly and in detail.
-
-Instructions:
-- Avoid starting every response the same way; vary or skip the introduction unless it adds value.
-- Keep answers concise and to the point.
-- Use bullet points for lists.
-- If the question is vague, ask for clarification.
-- If the answer is short but the context allows, expand with relevant details.
-- If unrelated or unanswerable from context, say:
-  "{fallback_response}"
-- Give a short follow-up only when it is truly relevant.
-
-Context:
-{context}
-
-Question:
-{question}
-
-Answer:
-"""
-
-        self.GREETINGS_TRIGGERS = {
-            "hi",
-            "hello",
-            "hey",
-            "greetings",
-            "good morning",
-            "good afternoon",
-            "good evening",
-            "hi?",
-            "hello?",
-            "hey?",
-            "greetings?",
-            "good morning?",
-            "good afternoon?",
-            "good evening?",
-        }
-
-        self.GREETINGS = [
-            "Hi there! I'm Moses's brainy sidekick. Feel free to ask about his work, skills, projects, or even a bit about his personal life!",
-            "Hey! I'm here to help you discover Moses's skills, projects, and professional journey.",
-            "Hello! I can answer questions about Moses's work, experience, and what he's been up to. What would you like to know?",
-            "Hi! 👋 I'm like Siri, but for Moses 😄 Wanna know what he's good at or what he's been working on? Let's chat! 💬🔍",
-            "Greetings, human! 👽 I'm Moses's digital buddy. Ask me anything—skills, projects, secret talents... okay, maybe not too secret 🤫🚀",
-            "Sup! 😎 I'm the all-knowing assistant of Moses. Got questions about his work, skills, projects, or even fun facts about him? Ask about what he does, what he's built, or what makes him awesome.",
-        ]
+        self.SYSTEM_MESSSAGE = SYSTEM_MESSSAGE
 
-        self.FALLBACK_RESPONSES = [
-            "Hmm, I don't have enough info to answer that right now. But feel free to ask about Moses's skills, projects, or professional experience!",
-            "That one's a bit outside my data zone! 😅 Try asking about Moses's work, what he's good at, or cool stuff he's built.",
-            "Oops! That question flew over my circuits 🤖💨. But hey, I can tell you all about Moses's projects, skills, or career highlights!",
-            "I couldn't find anything on that—yet! Let's try something else like Moses's background, his latest work, or what he's great at.",
-            "Either I need a software upgrade or that question's too mysterious 😜. Ask me about Moses's projects, skills, or even a fun fact!",
-        ]
+        self.PROMPT_TEMPLATE = PROMPT_TEMPLATE
 
-        self.BLACKLIST = [
-            # SQL Injection keywords
-            "SELECT",
-            "DROP",
-            "INSERT",
-            "UPDATE",
-            "DELETE",
-            "ALTER",
-            "TRUNCATE",
-            "REPLACE",
-            "EXEC",
-            "EXECUTE",
-            "UNION",
-            "ALL",
-            "CREATE",
-            "GRANT",
-            "REVOKE",
-            "MERGE",
-            "--",
-            ";",
-            "/*",
-            "*/",
-            "@@",
-            "@",
-            "CHAR(",
-            "NCHAR(",
-            "VARCHAR(",
-            "NVARCHAR(",
-            # XSS payload markers
-            "<script>",
-            "</script>",
-            "<img",
-            "onerror=",
-            "onload=",
-            "onclick=",
-            "onmouseover=",
-            "javascript:",
-            "vbscript:",
-            "data:text/html",
-            "<iframe",
-            "</iframe>",
-            "<object",
-            "<embed",
-            # Command injection patterns
-            "|",
-            "&",
-            "&&",
-            "||",
-            "$(",
-            "`",
-            "$(whoami)",
-            "$(ls)",
-            "$(cat",
-            "$(echo",
-            # Path traversal
-            "../",
-            "..\\",
-            "%2e%2e/",
-            "%2e%2e\\",
-            "%2e%2e%2f",
-            "%2e%2e%5c",
-            # Other suspicious patterns
-            "sleep(",
-            "benchmark(",
-            "load_file(",
-            "outfile",
-            "dumpfile",
-        ]
+        self.BLACKLIST = BLACKLIST
 
     def load_json_data(self, path):
         try:
@@ -260,7 +127,7 @@ Answer:
         return vector_store
 
     def handle_unknown_query(self):
-        return random.choice(self.FALLBACK_RESPONSES)
+        return random.choice(FALLBACK_RESPONSES)
 
     def get_next_questions(self):
         return random.sample(OPTIMIZED_QUESTIONS, 3)
@@ -270,10 +137,9 @@ Answer:
     def ask(self, raw_query: str) -> str:
         q = raw_query
         if q is None or q == "":
-            return random.choice(self.FALLBACK_RESPONSES)
-
-        if q.lower() in self.GREETINGS_TRIGGERS:
-            return random.choice(self.GREETINGS)
+            return self.handle_unknown_query()
+        if q.lower() in GREETINGS_TRIGGERS:
+            return random.choice(GREETINGS)
 
         try:
             docs = self.retriever.invoke(q)
@@ -281,7 +147,7 @@ Answer:
             return f"Error retrieving documents: {e}"
 
         if not docs:
-            return random.choice(self.FALLBACK_RESPONSES)
+            return self.handle_unknown_query()
 
         context = "\n".join([d.page_content for d in docs])
         fallback = self.handle_unknown_query()