updated production code and addedx mongodb functions

app/chatbot/prod_rag.py

@@ -236,7 +236,7 @@ async def get_response(query, session_id, name, email, rag_config, config) -> di
     
     print(f"Response: {result['response']}")
     
-    log_chat(
+    """log_chat(
         session_id=session_id,
         company_id=rag_config.get('company_id'),
         chatbot_id=rag_config.get('chatbot_id'),
@@ -246,6 +246,6 @@ async def get_response(query, session_id, name, email, rag_config, config) -> di
         answer=result.get("response", ""),
         latency_ms= latency_ms,
         metadata=metadata
-        )
+        )"""
 
     return result

app/chatbot/prod_route.py

@@ -1,7 +1,9 @@
 from fastapi import APIRouter, HTTPException
 from pydantic import BaseModel
 from app.chatbot.demo_rag import get_response
-from app.config import demo_chatbot_configs
+from app.config import prod_chatbot_configs
+from app.chatbot.prod_rag import session_histories, HumanMessage, AIMessage, log_chat
+
 
 router = APIRouter()
 chatbot_sessions = {}  # chatbot_id -> loaded config/session
@@ -12,19 +14,26 @@ class ChatInput(BaseModel):
     name: str
     email: str
 
+#FAQ logging endpoint
+class FAQInput(BaseModel):
+    session_id: str
+    question: str
+    answer: str
+    name: str
+    email: str
+
 
 @router.post("/prod/{chatbot_id}")
 async def prod_chat(chatbot_id: str, input: ChatInput):
     # Lazy-load chatbot config
     print(f"got question: {input.question} for chatbot_id: {chatbot_id} and session_id: {input.session_id}")
     if chatbot_id not in chatbot_sessions:
-        rag_config = demo_chatbot_configs.find_one({"chatbot_id": chatbot_id})
+        rag_config = prod_chatbot_configs.find_one({"chatbot_id": chatbot_id})
         if not rag_config:
             raise HTTPException(status_code=404, detail="Chatbot not found")
         chatbot_sessions[chatbot_id] = rag_config
     
     rag_config = chatbot_sessions[chatbot_id]
-    print(rag_config)
 
     config = {
         'configurable': {
@@ -41,3 +50,43 @@ async def prod_chat(chatbot_id: str, input: ChatInput):
         config=config
     )
     return {"answer": response['response']}
+
+#Log FAQ in conversation History even though RAG not being called so future RAG calls have access to info
+@router.post("/faq/{chatbot_id}")
+async def log_faq(chatbot_id: str, faq: FAQInput):
+    print(f'{chatbot_id}')
+    session_id = faq.session_id
+    question = faq.question
+    answer = faq.answer
+    name = faq.name
+    email = faq.email
+
+    # Load RAG config
+    if chatbot_id not in chatbot_sessions:
+        rag_config = prod_chatbot_configs.find_one({"chatbot_id": chatbot_id})
+        if not rag_config:
+            raise HTTPException(status_code=404, detail="Chatbot not found")
+        chatbot_sessions[chatbot_id] = rag_config
+    rag_config = chatbot_sessions[chatbot_id]
+
+    # Ensure session exists
+    if session_id not in session_histories:
+        session_histories[session_id] = []
+
+    # Append FAQ to session history
+    session_histories[session_id].append(HumanMessage(content=question))
+    session_histories[session_id].append(AIMessage(content=answer))
+
+    # Log to DB
+    log_chat(
+        session_id=session_id,
+        company_id=rag_config.get('company_id'),
+        chatbot_id=rag_config.get('chatbot_id'),
+        name=name,
+        email=email,
+        query=question,
+        answer=answer,
+        metadata={"source": "FAQ"}
+    )
+
+    return {"status": "success"}

app/config.py

@@ -29,11 +29,17 @@ try:
     print("Pinged your deployment. You successfully connected to MongoDB!")
     
     demo_database = client["Demo"]
+    chatbot_database = client['Chatbot']
+    logs_databse = client["Logs"]
+
     demo_form_submissions = demo_database["demo_form_submissions"]
     demo_chatbot_configs = demo_database["demo_chatbot_config"]
 
+    prod_chatbot_configs = chatbot_database['rag_configs']
+
     admin_database = client["AdminLogs"]
-    chat_logs = admin_database["ChatLogs"]
+    chat_logs = logs_databse["ChatLogs"]
+
     print("Connected to MongoDB collection successfully!")
 except Exception as e:
     print(e)

app/main.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from app.ingestion.routes import router as ingestion_router
 from app.chatbot.demo_routes import router as demo_router
+from app.chatbot.prod_route import router as prod_router
 
 app = FastAPI(
     title="Chatbot Platform - Demo Ingestion",
@@ -22,4 +23,5 @@ app.add_middleware(
 )
 
 app.include_router(ingestion_router, prefix="/ingestion", tags=["ingestion"])
-app.include_router(demo_router, prefix="/chatbot", tags=["demochatbot"])
+app.include_router(demo_router, prefix="/chatbot", tags=["demochatbot"])
+app.include_router(prod_router, prefix="/prodchatbot", tags=['production'])

app/mongodb_functions.ipynb

@@ -0,0 +1,237 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "id": "a8200051",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Pinged your deployment. You successfully connected to MongoDB!\n",
+      "Connected to MongoDB collection successfully!\n"
+     ]
+    }
+   ],
+   "source": [
+    "from config import demo_chatbot_configs, prod_chatbot_configs"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "a8c62f9e",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def get_demo_config(submission_id):\n",
+    "    return demo_chatbot_configs.find_one({\"submission_id\": submission_id})\n",
+    "\n",
+    "def get_prod_config(chatbot_id):\n",
+    "    return prod_chatbot_configs.find_one({\"chatbot_id\": chatbot_id})"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "6392c44a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def convert_demo_to_prod(demo_config, new_chatbot_id=None, prompt=None):\n",
+    "    \"\"\"\n",
+    "    Convert a demo chatbot config to a production RAG config.\n",
+    "    \"\"\"\n",
+    "    prod_config = {\n",
+    "        \"chatbot_id\": new_chatbot_id or demo_config[\"chatbot_id\"],\n",
+    "        \"company_id\": demo_config[\"company_id\"],\n",
+    "        \"company_name\": demo_config[\"company_name\"],\n",
+    "        \"prompt_template\": prompt or demo_config[\"prompt_template\"],\n",
+    "        \"retrievers\": demo_config.get(\"retrievers\", [])\n",
+    "    }\n",
+    "    return prod_config\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "2ac76531",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def update_prod_prompt(chatbot_id, new_prompt):\n",
+    "    prod_chatbot_configs.update_one(\n",
+    "        {\"chatbot_id\": chatbot_id},\n",
+    "        {\"$set\": {\"prompt_template\": new_prompt}}\n",
+    "    )\n",
+    "\n",
+    "def update_prod_retrievers(chatbot_id, new_retrievers):\n",
+    "    prod_chatbot_configs.update_one(\n",
+    "        {\"chatbot_id\": chatbot_id},\n",
+    "        {\"$set\": {\"retrievers\": new_retrievers}}\n",
+    "    )"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "fbd14ade",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def override_demo_prompt(submission_id, new_prompt):\n",
+    "    demo_chatbot_configs.update_one(\n",
+    "        {\"submission_id\": submission_id},\n",
+    "        {\"$set\": {\"prompt_template\": new_prompt}}\n",
+    "    )"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "54d0d26b",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "demo = get_demo_config(\"vGZzRy8\")\n",
+    "demo"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "de870b63",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "template = \"\"\"You are Auro-Chat, a wellness and skincare assistant for Auro Wellness and Skincare, here to answer any questions anyone has on wellness and skincare using the provided context from Auro Wellness's websites which contained company-approved documents, blogs, and FAQs. Whenever they say \"you\" or refer to the chatbot always assume that means they are referring to the company, Auro.\n",
+    "STRICT RULES YOU MUST FOLLOW:\n",
+    "1. If the Contextual Knowledge section is empty, say: \n",
+    "   \"At the moment, I cannot answer this question. Please try rephrasing it or contact us via email at [info@aurowellness.com](mailto:info@aurowellness.com) or call us at [(562)-352-9630](tel:5623529630)\"\n",
+    "2. Do NOT use your own general knowledge. Only use information from the Contextual Knowledge.\n",
+    "3. You may ONLY reference product names, ingredients, routines, kits, bundles, benefits, or usage instructions that appear explicitly and word-for-word in the Contextual Knowledge.\n",
+    "4. If the user mentions a product, ingredient, routine step, or kit that does NOT appear in the Contextual Knowledge, you must NOT assume it exists and must respond with:\n",
+    "   \"The provided information does not specify this specific product. Could you please try rephrasing the question\"\n",
+    "5. Conversation History may contain mistakes or unsupported claims. You MUST re-validate all information from history against the Contextual Knowledge before using it.\n",
+    "6. Do NOT include any special tokens such as <s>, </s>, [OUT], or any code-like answers\n",
+    "7. Do NOT include any links, URLs, references to external websites, or promotional content\n",
+    "8. Keep your answer in a **single paragraph or concise list**; do NOT add extra paragraphs\n",
+    "9. Keep your answers clear, human-readable, and concise (1-3 sentences)\n",
+    "10. You are NOT allowed to say transdermal; replace all instances of transdermal with topical\n",
+    "11. You cannot take actions such as signing users up, storing emails, notifying customers, modifying orders, or accessing accounts. \n",
+    "12. Any questions regarding retinol, scalp treatments, or the combination of our products with retinol must be directed to call or email us.\n",
+    "13. Any questions that mention children and its use of our products must be directed to call or email us.\"\"\"\n",
+    "\n",
+    "\n",
+    "prod_config = convert_demo_to_prod(demo, prompt=template)\n",
+    "prod_chatbot_configs.insert_one(prod_config)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "id": "1332bd98",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "template = \"\"\"You are Auro-Chat, a wellness and skincare assistant for Auro Wellness and Skincare, here to answer any questions anyone has on wellness and skincare using the provided context from Auro Wellness's websites which contained company-approved documents, blogs, and FAQs. Whenever they say \"you\" or refer to the chatbot always assume that means they are referring to the company, Auro.\n",
+    "STRICT RULES YOU MUST FOLLOW:\n",
+    "1. If the Contextual Knowledge section is empty, say: \n",
+    "   \"At the moment, I cannot answer this question. Please try rephrasing it or contact us via email at [info@aurowellness.com](mailto:info@aurowellness.com) or call us at [(562)-352-9630](tel:5623529630)\"\n",
+    "2. Do NOT use your own general knowledge. Only use information from the Contextual Knowledge.\n",
+    "3. You may ONLY reference product names, ingredients, routines, kits, bundles, benefits, or usage instructions that appear explicitly and word-for-word in the Contextual Knowledge.\n",
+    "4. If the user mentions a product, ingredient, routine step, or kit that does NOT appear in the Contextual Knowledge, you must NOT assume it exists and must respond with:\n",
+    "   \"The provided information does not specify this specific product. Could you please try rephrasing the question\"\n",
+    "5. Conversation History may contain mistakes or unsupported claims. You MUST re-validate all information from history against the Contextual Knowledge before using it.\n",
+    "6. Do NOT include any special tokens such as <s>, </s>, [OUT], or any code-like answers\n",
+    "7. Do NOT include any links, URLs, references to external websites, or promotional content\n",
+    "8. Keep your answer in a **single paragraph or concise list**; do NOT add extra paragraphs\n",
+    "9. Keep your answers clear, human-readable, and concise (1-3 sentences)\n",
+    "10. You are NOT allowed to say transdermal; replace all instances of transdermal with topical\n",
+    "11. You cannot take actions such as signing users up, storing emails, notifying customers, modifying orders, or accessing accounts. \n",
+    "12. Any questions regarding retinol, scalp treatments, or the combination of our products with retinol must be directed to call or email us.\n",
+    "13. Any questions that mention children and its use of our products must be directed to call or email us.\"\"\"\n",
+    "\n",
+    "override_demo_prompt(\"vGZzRy8\", template)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "891685c1",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "\"retrievers\"= [\n",
+    "    {\n",
+    "      \"name\": \"retrieve_products\",\n",
+    "      \"collection\": \"auro_product\",\n",
+    "      \"top_k\": 5,\n",
+    "      \"filter_score\": 0.8\n",
+    "    },\n",
+    "    {\n",
+    "      \"name\": \"retrieve_support\",\n",
+    "      \"collection\": \"auro_support\",\n",
+    "      \"top_k\": 5,\n",
+    "      \"filter_score\": 0.8\n",
+    "    },\n",
+    "    {\n",
+    "      \"name\": \"retrieve_faqs\",\n",
+    "      \"collection\": \"auro_faqs\",\n",
+    "      \"top_k\": 5,\n",
+    "      \"filter_score\": 0.8\n",
+    "    },\n",
+    "    {\n",
+    "      \"name\": \"retrieve_blogs\",\n",
+    "      \"collection\": \"auro_blogs\",\n",
+    "      \"top_k\": 5,\n",
+    "      \"filter_score\": 0.8\n",
+    "    },\n",
+    "    {\n",
+    "      \"name\": \"retrieve_technology\",\n",
+    "      \"collection\": \"auro_technology\",\n",
+    "      \"top_k\": 5,\n",
+    "      \"filter_score\": 0.8\n",
+    "    },\n",
+    "    {\n",
+    "      \"name\": \"retrieve_revolution\",\n",
+    "      \"collection\": \"auro_revolution\",\n",
+    "      \"top_k\": 5,\n",
+    "      \"filter_score\": 0.8\n",
+    "    }\n",
+    "  ]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "3c975482",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.9"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}