Create Chatbot Service - Complete Developer Documentation¶

async def create_chatbot(
    user_id: str = Form(...),
    database_type: str = Form(None)  # Optional: "cosmosdb" or "milvus"
)

curl -X POST http://localhost:8003/v2/create-chatbot \
  -F "user_id=User-123456" \
  -F "database_type=milvus"

{
  "message": "Chatbot created",
  "project_id": "User-123456_Project_1",
  "database_type": "milvus"
}

{
  "message": "Chatbot created",
  "project_id": "User-123456_Project_1"
}

GET /v2/get-chatbots?user_id=User-123456

{
  "user_id": "User-123456",
  "projects": [
    {
      "project_id": "User-123456_Project_1",
      "active": true
    },
    {
      "project_id": "User-123456_Project_2",
      "active": false
    }
  ]
}

db.projectid_creation.find(
  {
    user_id: "User-123456",
    $or: [{ isDeleted: { $exists: false } }, { isDeleted: false }],
  },
  {
    _id: 0,
    project_id: 1,
    active: 1,
  }
);

GET /v2/get-chatbot-status?user_id=User-123456&project_id=User-123456_Project_1

{
  "user_id": "User-123456",
  "project_id": "User-123456_Project_1",
  "active": true
}

async def create_chatbot(
    user_id: str = Form(...),
    project_id: str = Form(...),
    chatbot_name: str = Form(...)
)

{
  "user_id": "User-123456",
  "project_id": "User-123456_Project_1",
  "database_type": "milvus",
  "is_default": false
}

def generate_project_id(user_id, existing_count):
    return f"{user_id}_Project_{existing_count + 1}"

chatbot_count = projectid_collection.count_documents({"user_id": user_id})

// User has 2 existing chatbots
db.projectid_creation.count_documents({ user_id: "User-123456" });
// Returns: 2

// Generate 3rd project ID
generate_project_id("User-123456", 2); // Returns: "User-123456_Project_3"

def get_user_chatbot_limit(user_id: str) -> int:
    """Get the chatbot limit for a user from features_per_user collection"""
    try:
        user_features = features_per_user_collection.find_one({"user_id": user_id})
        if user_features and "no_of_chatbots" in user_features:
            limit = int(user_features["no_of_chatbots"])
            logger.debug(f"Chatbot limit for user {user_id}: {limit}")
            return limit
        else:
            logger.warning(f"No chatbot limit found for user {user_id}, using default: 1")
            return 1
    except (ValueError, TypeError) as e:
        logger.warning(f"Error converting chatbot limit for user {user_id}: {e}")
        return 1
    except Exception as e:
        logger.error(f"Error fetching chatbot limit for user {user_id}: {e}")
        return 1

db.features_per_user.findOne({ user_id: "User-123456" });

{
    "_id": ObjectId("..."),
    "user_id": "User-123456",
    "subscription_id": "sub_009",  // Free plan
    "no_of_chatbots": 1,           // LIMIT
    "no_of_chat_sessions": 50,
    // ... other features
}

# This should be added to create_chatbot() around line 265
chatbot_count = projectid_collection.count_documents({"user_id": user_id})
user_limit = get_user_chatbot_limit(user_id)

if chatbot_count >= user_limit:
    # User exceeded quota
    send_quota_exceeded_email(user_id, chatbot_count, user_limit)
    raise HTTPException(
        status_code=403,
        detail=f"Chatbot limit exceeded. You have {chatbot_count}/{user_limit} chatbots."
    )
elif chatbot_count == user_limit - 1:
    # User is creating their last allowed chatbot
    send_quota_warning_email(user_id, chatbot_count + 1, user_limit)

EMAIL_ENDPOINT = "https://mailing-sevice.india.communication.azure.com/"
EMAIL_ACCESS_KEY = "CgdEWi6fBCJv4..."  # Hardcoded!
SENDER_EMAIL = "DoNotReply@machineagents.ai"

email_client = EmailClient(endpoint=EMAIL_ENDPOINT, credential=EMAIL_ACCESS_KEY)

Dear User,

You have just created your 3th chatbot out of your allowed limit of 3 chatbots.

This means you have reached your maximum chatbot creation limit according to your current subscription plan. If you need to create more chatbots, please consider upgrading your subscription plan.

For any assistance, please contact our support team.

Thank you for using our service!

Best regards,
The Machine Agents Team

message = {
    "senderAddress": SENDER_EMAIL,
    "recipients": {"to": [{"address": user_email}]},
    "content": {
        "subject": "Chatbot Quota Limit Notification",
        "plainText": f"""
Dear User,

You have just created your {current_count}th chatbot out of your allowed limit of {limit} chatbots.

This means you have reached your maximum chatbot creation limit according to your current subscription plan. If you need to create more chatbots, please consider upgrading your subscription plan.

For any assistance, please contact our support team.

Thank you for using our service!

Best regards,
The Machine Agents Team
        """
    }
}

email_client.begin_send(message)

Dear User,

Your attempt to create a new chatbot has been blocked because you have already reached your maximum limit.

Current Status:
- Chatbots Created: 1
- Maximum Allowed: 1
- Remaining: 0

To create more chatbots, please upgrade your subscription plan. Our premium plans offer higher chatbot limits and additional features.

For assistance with upgrading your plan or any questions, please contact our support team.

Thank you for using our service!

Best regards,
The Machine Agents Team

def check_selection_state_3D(user_id: str, project_id: str):
    selection = selection_collection.find_one({"user_id": user_id, "project_id": project_id})

    if not selection or selection.get("chatbot_type") != "3D-chatbot":
        return {"state": "start", "message": "No chatbot collection found."}

    elif "chatbot_type" in selection and "sitemap_urls" not in selection:
        return {"state": "chatbot_type_selected"}

    elif "chatbot_type" in selection and "sitemap_urls" in selection and "chatbot_purpose" not in selection:
        return {"state": "sitemap_urls_fetched"}

    elif ... and "selection_avatar" not in selection:
        return {"state": "chatbot_purpose_selected"}

    elif ... and "selection_voice" not in selection:
        return {"state": "avatar_selected"}

    elif ... and "selection_voice" in selection:
        return {"state": "END"}  # Final step completed

    else:
        return {"state": "start"}

# Check if last project is incomplete
if chatbot_count > 0:
    last_project_id = generate_project_id(user_id, chatbot_count)
    existing_project = projectid_collection.find_one({"user_id": user_id, "project_id": last_project_id})

    if existing_project:
        # Get chatbot type
        selection = selection_collection.find_one({"user_id": user_id, "project_id": last_project_id})

        if selection and "chatbot_type" in selection:
            chatbot_type = selection["chatbot_type"]

            # Call appropriate state check
            if chatbot_type == "text-chatbot":
                state_response = check_selection_state_text(user_id, project_id)
            elif chatbot_type == "voice-chatbot":
                state_response = check_selection_state_voice(user_id, project_id)
            elif chatbot_type == "3D-chatbot":
                state_response = check_selection_state_3D(user_id, project_id)

            # If not complete, return existing project
            if state_response["state"] != "END":
                return JSONResponse(
                    content={"message": "Chatbot created", "project_id": last_project_id},
                    status_code=201
                )

{
    "_id": ObjectId("..."),
    "user_id": "User-123456",
    "project_id": "User-123456_Project_1",
    "state": "IN_PROGRESS",  // or "COMPLETED"
    "created_at": ISODate("2025-01-15T10:00:00Z"),
    "active": true,
    "database_type": "milvus",  // or "cosmosdb"
    "isDeleted": false  // Soft delete flag
}

projectid_collection.insert_one({
    "user_id": user_id,
    "project_id": project_id,
    "state": "IN_PROGRESS",
    "created_at": datetime.now(),
    "active": True,
    "database_type": database_type
})

{
    "_id": ObjectId("..."),
    "user_id": "User-123456",
    "project_id": "User-123456_Project_1",

    // Database selection
    "database_type": "milvus",
    "database_selection_timestamp": ISODate("2025-01-15T10:00:05Z"),

    // Configuration steps (populated over time)
    "chatbot_type": "3D-chatbot",  // or "text-chatbot", "voice-chatbot"
    "chatbot_purpose": "Customer Support",
    "selection_avatar": "Avatar_Emma",
    "selection_voice": "en-US-JennyNeural",
    "selection_model": "openai-35",
    "sitemap_urls": ["https://example.com/page1", "https://example.com/page2"],

    // Optional
    "chatbot_name": "Support Bot",
    "domain": "https://example.com"
}

selection_collection.update_one(
    {"user_id": user_id, "project_id": project_id},
    {"$set": {
        "database_type": database_type,
        "database_selection_timestamp": datetime.utcnow()
    }},
    upsert=True  # Create if doesn't exist
)

user_features = features_per_user_collection.find_one({"user_id": user_id})
limit = int(user_features["no_of_chatbots"])

{
    "user_id": "User-123456",
    "subscription_id": "sub_009",
    "no_of_chatbots": 1,  // ← THIS IS THE LIMIT
    "no_of_chat_sessions": 50,
    // ... other features
}

user = users_collection.find_one({"user_id": user_id})
user_email = user["email"]

DEFAULT_DATABASE=milvus  # or "cosmosdb"

default_database = os.getenv("DEFAULT_DATABASE", "cosmosdb").lower()

# In endpoint
if database_type is None:
    database_type = default_database  # Use env default
else:
    database_type = database_type.lower()
    if database_type not in ["cosmosdb", "milvus"]:
        raise HTTPException(
            status_code=400,
            detail="Invalid database_type. Must be 'cosmosdb' or 'milvus'"
        )

{
    "project_id": "User-123456_Project_1",
    "database_type": "milvus"
}

{
    "project_id": "User-123456_Project_1",
    "database_type": "milvus",
    "database_selection_timestamp": ISODate("...")
}

EMAIL_ACCESS_KEY = "CgdEWi6fBCJv4..."  # Hardcoded!

EMAIL_ENDPOINT = os.getenv("AZURE_EMAIL_ENDPOINT")
EMAIL_ACCESS_KEY = os.getenv("AZURE_EMAIL_ACCESS_KEY")

from slowapi import Limiter

@app.post("/v2/create-chatbot")
@limiter.limit("10/minute")  # 10 chatbots per minute
async def create_chatbot(...):
    ...

FROM python:3.9-slim

WORKDIR /app

COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

COPY src/ .

EXPOSE 8003

CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8003"]

create-chatbot-service:
  build: ./create-chatbot-service
  container_name: create-chatbot-service
  ports:
    - "8003:8003"
  networks:
    - app-network
  labels:
    - "logging=loki"
    - "app=machine-agent-app"
  environment:
    - MONGO_URI=${MONGO_URI}
    - MONGO_DB_NAME=Machine_agent_demo
    - DEFAULT_DATABASE=milvus

    # Azure Email (should be added)
    - AZURE_EMAIL_ENDPOINT=${AZURE_EMAIL_ENDPOINT}
    - AZURE_EMAIL_ACCESS_KEY=${AZURE_EMAIL_ACCESS_KEY}

    # DataDog
    - DD_SERVICE=create-chatbot-service
    - DD_ENV=production
  restart: always

fastapi>=0.95.0
uvicorn[standard]>=0.22.0
pymongo>=4.3.3
python-multipart>=0.0.6
python-dotenv>=1.0.0
azure-communication-email>=1.0.0
ddtrace>=1.19.0