mirror of
http://88.130.71.182:3000/BlitTech/contexta_be.git
synced 2026-06-12 23:23:21 +00:00
417 lines
13 KiB
Python
417 lines
13 KiB
Python
from pydantic import BaseModel, EmailStr, Field
|
|
from typing import Optional, List, Dict, Any
|
|
from datetime import datetime
|
|
from enum import Enum
|
|
import uuid
|
|
|
|
|
|
# ─── Enums ────────────────────────────────────────────────────────────────────
|
|
|
|
class PlanType(str, Enum):
|
|
free = "free"
|
|
starter = "starter"
|
|
business = "business"
|
|
agency = "agency"
|
|
enterprise = "enterprise"
|
|
|
|
|
|
class SubscriptionStatus(str, Enum):
|
|
active = "active"
|
|
canceled = "canceled"
|
|
past_due = "past_due"
|
|
unpaid = "unpaid"
|
|
trialing = "trialing"
|
|
|
|
|
|
class ChatbotVisibility(str, Enum):
|
|
preview = "preview"
|
|
published = "published"
|
|
|
|
|
|
class DocumentStatus(str, Enum):
|
|
pending = "pending"
|
|
processing = "processing"
|
|
completed = "completed"
|
|
failed = "failed"
|
|
|
|
|
|
class MessageRole(str, Enum):
|
|
user = "user"
|
|
assistant = "assistant"
|
|
system = "system"
|
|
|
|
|
|
# ─── Auth Models ──────────────────────────────────────────────────────────────
|
|
|
|
class UserSignup(BaseModel):
|
|
email: EmailStr
|
|
password: str = Field(min_length=8)
|
|
company_name: str = Field(min_length=2, max_length=100)
|
|
|
|
|
|
class UserLogin(BaseModel):
|
|
email: EmailStr
|
|
password: str
|
|
|
|
|
|
class UserResponse(BaseModel):
|
|
id: str
|
|
email: str
|
|
company_name: Optional[str] = None
|
|
plan: str = "free"
|
|
created_at: Optional[datetime] = None
|
|
|
|
|
|
class TokenResponse(BaseModel):
|
|
access_token: str
|
|
token_type: str = "bearer"
|
|
user: UserResponse
|
|
|
|
|
|
# ─── Company Models ────────────────────────────────────────────────────────────
|
|
|
|
class CompanyCreate(BaseModel):
|
|
name: str = Field(min_length=2, max_length=100)
|
|
website: Optional[str] = None
|
|
industry: Optional[str] = None
|
|
|
|
|
|
class CompanyUpdate(BaseModel):
|
|
name: Optional[str] = None
|
|
website: Optional[str] = None
|
|
industry: Optional[str] = None
|
|
logo_url: Optional[str] = None
|
|
|
|
|
|
class CompanyResponse(BaseModel):
|
|
id: str
|
|
owner_id: str
|
|
name: str
|
|
website: Optional[str] = None
|
|
industry: Optional[str] = None
|
|
logo_url: Optional[str] = None
|
|
created_at: Optional[datetime] = None
|
|
|
|
|
|
# ─── Chatbot Models ────────────────────────────────────────────────────────────
|
|
|
|
class ChatbotCreate(BaseModel):
|
|
name: str = Field(min_length=2, max_length=100)
|
|
description: Optional[str] = None
|
|
system_prompt: Optional[str] = None
|
|
model: str = "accounts/fireworks/models/kimi-k2-instruct-0905"
|
|
temperature: float = Field(default=0.7, ge=0.0, le=2.0)
|
|
max_tokens: int = Field(default=1000, ge=100, le=8000)
|
|
primary_color: str = "#6366f1"
|
|
welcome_message: str = "Hello! How can I help you today?"
|
|
logo_url: Optional[str] = None
|
|
category: Optional[str] = None
|
|
industry: Optional[str] = None
|
|
languages: List[str] = ["en"]
|
|
show_branding: bool = True
|
|
lead_capture_enabled: bool = False
|
|
lead_capture_fields: List[str] = ["email"]
|
|
lead_capture_trigger: str = "after_first_message"
|
|
handoff_enabled: bool = False
|
|
handoff_message: str = "I'll connect you with our team. Please wait."
|
|
handoff_email: Optional[str] = None
|
|
handoff_keywords: List[str] = ["human", "agent", "speak to someone", "talk to a person", "real person"]
|
|
|
|
|
|
class ChatbotUpdate(BaseModel):
|
|
name: Optional[str] = None
|
|
description: Optional[str] = None
|
|
system_prompt: Optional[str] = None
|
|
model: Optional[str] = None
|
|
temperature: Optional[float] = None
|
|
max_tokens: Optional[int] = None
|
|
primary_color: Optional[str] = None
|
|
welcome_message: Optional[str] = None
|
|
logo_url: Optional[str] = None
|
|
category: Optional[str] = None
|
|
industry: Optional[str] = None
|
|
languages: Optional[List[str]] = None
|
|
show_branding: Optional[bool] = None
|
|
lead_capture_enabled: Optional[bool] = None
|
|
lead_capture_fields: Optional[List[str]] = None
|
|
lead_capture_trigger: Optional[str] = None
|
|
handoff_enabled: Optional[bool] = None
|
|
handoff_message: Optional[str] = None
|
|
handoff_email: Optional[str] = None
|
|
handoff_keywords: Optional[List[str]] = None
|
|
|
|
|
|
class ChatbotResponse(BaseModel):
|
|
id: str
|
|
company_id: str
|
|
name: str
|
|
description: Optional[str] = None
|
|
system_prompt: Optional[str] = None
|
|
model: str
|
|
temperature: float
|
|
max_tokens: int
|
|
primary_color: str
|
|
welcome_message: str
|
|
logo_url: Optional[str] = None
|
|
category: Optional[str] = None
|
|
industry: Optional[str] = None
|
|
languages: List[str]
|
|
visibility: str
|
|
is_published: bool
|
|
qdrant_collection_name: Optional[str] = None
|
|
document_count: int = 0
|
|
conversation_count: int = 0
|
|
average_rating: Optional[float] = None
|
|
created_at: Optional[datetime] = None
|
|
published_at: Optional[datetime] = None
|
|
show_branding: bool = True
|
|
lead_capture_enabled: bool = False
|
|
lead_capture_fields: List[str] = ["email"]
|
|
lead_capture_trigger: str = "after_first_message"
|
|
handoff_enabled: bool = False
|
|
handoff_message: str = "I'll connect you with our team. Please wait."
|
|
handoff_email: Optional[str] = None
|
|
handoff_keywords: List[str] = ["human", "agent", "speak to someone", "talk to a person", "real person"]
|
|
|
|
|
|
class ChatbotPublicResponse(BaseModel):
|
|
"""For marketplace display"""
|
|
id: str
|
|
name: str
|
|
description: Optional[str] = None
|
|
category: Optional[str] = None
|
|
industry: Optional[str] = None
|
|
languages: List[str]
|
|
primary_color: str
|
|
welcome_message: str
|
|
logo_url: Optional[str] = None
|
|
average_rating: Optional[float] = None
|
|
total_conversations: int = 0
|
|
company_name: Optional[str] = None
|
|
company_logo: Optional[str] = None
|
|
created_at: Optional[datetime] = None
|
|
published_at: Optional[datetime] = None
|
|
|
|
|
|
# ─── Document Models ───────────────────────────────────────────────────────────
|
|
|
|
class DocumentResponse(BaseModel):
|
|
id: str
|
|
chatbot_id: str
|
|
file_name: str
|
|
file_type: str
|
|
file_size: int
|
|
chunk_count: int
|
|
status: str
|
|
error_message: Optional[str] = None
|
|
created_at: Optional[datetime] = None
|
|
|
|
|
|
# ─── Chat Models ───────────────────────────────────────────────────────────────
|
|
|
|
class ChatMessage(BaseModel):
|
|
message: str = Field(min_length=1, max_length=4000)
|
|
session_id: Optional[str] = None
|
|
language: str = "en"
|
|
|
|
|
|
class SourceDocument(BaseModel):
|
|
document_name: str
|
|
chunk_text: str
|
|
score: float
|
|
page_number: Optional[int] = None
|
|
|
|
|
|
class ChatResponse(BaseModel):
|
|
response: str
|
|
session_id: str
|
|
sources: List[SourceDocument] = []
|
|
model_used: str
|
|
tokens_used: int = 0
|
|
needs_lead_capture: bool = False
|
|
handoff: bool = False
|
|
|
|
|
|
class MessageResponse(BaseModel):
|
|
id: str
|
|
role: str
|
|
content: str
|
|
sources: Optional[List[Dict]] = None
|
|
created_at: Optional[datetime] = None
|
|
|
|
|
|
class ConversationResponse(BaseModel):
|
|
id: str
|
|
chatbot_id: str
|
|
session_id: Optional[str] = None
|
|
language: str
|
|
message_count: int
|
|
created_at: Optional[datetime] = None
|
|
messages: List[MessageResponse] = []
|
|
|
|
|
|
# ─── Subscription Models ───────────────────────────────────────────────────────
|
|
|
|
class SubscriptionResponse(BaseModel):
|
|
id: str
|
|
user_id: str
|
|
plan: str
|
|
status: str
|
|
stripe_customer_id: Optional[str] = None
|
|
current_period_start: Optional[datetime] = None
|
|
current_period_end: Optional[datetime] = None
|
|
chatbots_published: int = 0
|
|
conversations_used: int = 0
|
|
created_at: Optional[datetime] = None
|
|
|
|
|
|
class CheckoutSessionCreate(BaseModel):
|
|
plan: str # starter, business, or agency
|
|
success_url: str
|
|
cancel_url: str
|
|
|
|
|
|
class CheckoutSessionResponse(BaseModel):
|
|
checkout_url: str
|
|
session_id: str
|
|
|
|
|
|
# ─── Analytics Models ──────────────────────────────────────────────────────────
|
|
|
|
class ChatbotAnalytics(BaseModel):
|
|
chatbot_id: str
|
|
total_conversations: int = 0
|
|
unique_users: int = 0
|
|
average_conversation_length: float = 0.0
|
|
total_messages: int = 0
|
|
average_rating: float = 0.0
|
|
top_queries: List[str] = []
|
|
conversations_last_7_days: List[Dict] = []
|
|
conversations_last_30_days: int = 0
|
|
|
|
|
|
# ─── Marketplace Models ────────────────────────────────────────────────────────
|
|
|
|
class MarketplaceFilter(BaseModel):
|
|
category: Optional[str] = None
|
|
industry: Optional[str] = None
|
|
language: Optional[str] = None
|
|
search: Optional[str] = None
|
|
page: int = 1
|
|
limit: int = 20
|
|
|
|
|
|
class MarketplaceResponse(BaseModel):
|
|
chatbots: List[ChatbotPublicResponse]
|
|
total: int
|
|
page: int
|
|
limit: int
|
|
has_more: bool
|
|
|
|
|
|
class RatingCreate(BaseModel):
|
|
rating: int = Field(ge=1, le=5)
|
|
feedback: Optional[str] = None
|
|
|
|
|
|
# ─── Code Export Models ────────────────────────────────────────────────────────
|
|
|
|
class CodeExportRequest(BaseModel):
|
|
chatbot_id: str
|
|
include_frontend: bool = True
|
|
|
|
|
|
# ─── Generic Response Models ───────────────────────────────────────────────────
|
|
|
|
class MessageDetail(BaseModel):
|
|
detail: str
|
|
|
|
|
|
class SuccessResponse(BaseModel):
|
|
success: bool
|
|
message: str
|
|
|
|
|
|
class ErrorResponse(BaseModel):
|
|
error: str
|
|
detail: Optional[str] = None
|
|
|
|
|
|
# ─── Lead Models ───────────────────────────────────────────────────────────────
|
|
|
|
class LeadCreate(BaseModel):
|
|
email: Optional[str] = None
|
|
name: Optional[str] = None
|
|
phone: Optional[str] = None
|
|
company: Optional[str] = None
|
|
conversation_id: Optional[str] = None
|
|
|
|
|
|
class LeadResponse(BaseModel):
|
|
id: str
|
|
chatbot_id: str
|
|
conversation_id: Optional[str] = None
|
|
email: Optional[str] = None
|
|
name: Optional[str] = None
|
|
phone: Optional[str] = None
|
|
company: Optional[str] = None
|
|
created_at: Optional[datetime] = None
|
|
|
|
|
|
# ─── URL Source Models ─────────────────────────────────────────────────────────
|
|
|
|
class UrlSourceCreate(BaseModel):
|
|
url: str
|
|
|
|
|
|
class UrlSourceResponse(BaseModel):
|
|
id: str
|
|
chatbot_id: str
|
|
url: str
|
|
status: str
|
|
page_title: Optional[str] = None
|
|
chunk_count: int = 0
|
|
error_message: Optional[str] = None
|
|
created_at: Optional[datetime] = None
|
|
|
|
|
|
# ─── Feedback Models ───────────────────────────────────────────────────────────
|
|
|
|
class FeedbackCreate(BaseModel):
|
|
message_id: str
|
|
feedback: str # 'positive' or 'negative'
|
|
|
|
|
|
# ─── Inbox Models ─────────────────────────────────────────────────────────────
|
|
|
|
class InboxConversation(BaseModel):
|
|
id: str
|
|
chatbot_id: str
|
|
chatbot_name: str
|
|
session_id: Optional[str] = None
|
|
language: str
|
|
message_count: int
|
|
first_message: Optional[str] = None
|
|
created_at: Optional[datetime] = None
|
|
|
|
|
|
class InboxMessage(BaseModel):
|
|
id: str
|
|
role: str
|
|
content: str
|
|
sources: Optional[List[Dict]] = None
|
|
confidence_score: Optional[float] = None
|
|
is_handoff: bool = False
|
|
created_at: Optional[datetime] = None
|
|
|
|
|
|
# ─── Channel Models ────────────────────────────────────────────────────────────
|
|
|
|
class ChannelConnectionResponse(BaseModel):
|
|
id: str
|
|
channel: str
|
|
bot_username: Optional[str] = None
|
|
wa_keyword: Optional[str] = None
|
|
wa_link: Optional[str] = None
|
|
is_active: bool
|
|
created_at: Optional[datetime] = None |