dhisana 0.0.1.dev17__tar.gz → 0.0.1.dev18__tar.gz

{dhisana-0.0.1.dev17 → dhisana-0.0.1.dev18}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: dhisana
-Version: 0.0.1.dev17
+Version: 0.0.1.dev18
 Summary: A Python SDK for Dhisana AI Platform
 Home-page: https://github.com/dhisana-ai/dhisana-python-sdk
 Author: Admin
@@ -27,3 +27,13 @@ Requires-Dist: simple_salesforce
 Requires-Dist: backoff
 Requires-Dist: html2text
 Requires-Dist: hubspot-api-client
+Requires-Dist: tldextract
+Requires-Dist: pyperclip
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: home-page
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

{dhisana-0.0.1.dev17 → dhisana-0.0.1.dev18}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='dhisana',  
-    version='0.0.1-dev17',
+    version='0.0.1-dev18',
     description='A Python SDK for Dhisana AI Platform',
     author='Admin',
     author_email='contact@dhisana.ai',
@@ -26,7 +26,9 @@ setup(
         'simple_salesforce',
         'backoff',
         'html2text',
-        'hubspot-api-client'
+        'hubspot-api-client',
+        'tldextract',
+        'pyperclip'
     ],
     entry_points={
         'console_scripts': [

dhisana-0.0.1.dev18/src/dhisana/schemas/common.py

@@ -0,0 +1,370 @@
+from uuid import UUID
+from pydantic import BaseModel, Field, HttpUrl
+from typing import List, Optional, Dict, Any
+from enum import Enum
+
+##
+# PERMISSION & ROLE
+##
+
+class PermissionBase(BaseModel):
+    name: str
+    description: Optional[str] = None
+
+class PermissionCreate(PermissionBase):
+    pass
+
+class Permission(PermissionBase):
+    id: int
+
+    class Config:
+        from_attributes = True
+
+class RoleBase(BaseModel):
+    name: str
+    description: Optional[str] = None
+
+class RoleCreate(RoleBase):
+    permission_ids: List[int] = []
+
+class Role(RoleBase):
+    id: int
+    permissions: List[Permission] = []
+
+    class Config:
+        from_attributes = True
+
+
+##
+# ORGANIZATION
+##
+
+class OrganizationBase(BaseModel):
+    name: str
+    description: Optional[str] = None
+
+class OrganizationCreate(OrganizationBase):
+    pass
+
+class Organization(OrganizationBase):
+    id: UUID
+    # changed from datetime -> milliseconds since epoch
+    created_at: Optional[int] = None
+
+    class Config:
+        from_attributes = True
+
+
+##
+# USER
+##
+
+class UserBase(BaseModel):
+    id: str
+    username: Optional[str]
+    email: Optional[str]
+    organization_id: Optional[UUID]
+    created_at: Optional[int] = None
+
+class UserCreate(UserBase):
+    pass
+
+class User(UserBase):
+    roles: List[Role] = []
+
+    class Config:
+        from_attributes = True
+
+
+##
+# AGENT + STATUSES
+##
+
+class AgentStatus(str, Enum):
+    PENDING = "PENDING"
+    APPROVED = "APPROVED"
+    REJECTED = "REJECTED"
+    PUBLISHED = "PUBLISHED"
+
+class AgentStatusUpdate(BaseModel):
+    status: AgentStatus
+
+class AgentInstanceStatus(str, Enum):
+    ACTIVE = "ACTIVE"
+    INACTIVE = "INACTIVE"
+    ERROR = "ERROR"
+
+
+##
+# UI / CONFIG FIELDS
+##
+
+class SelectOption(BaseModel):
+    label: str
+    value: Any
+
+class ValidationRules(BaseModel):
+    required: Optional[bool] = None
+    min: Optional[int] = None
+    max: Optional[int] = None
+    pattern: Optional[str] = None
+
+class ConfigField(BaseModel):
+    name: str
+    label: str
+    type: str
+    sensitive: Optional[bool] = False
+    default: Optional[Any] = Field(None, alias="default")
+    description: Optional[str] = None
+    validations: Optional[ValidationRules] = None
+    options: Optional[List[SelectOption]] = None
+    fields: Optional[List["ConfigField"]] = None  # for nested structures
+
+    class Config:
+        populate_by_name = True
+
+ConfigField.model_rebuild()
+
+class ConfigGroup(BaseModel):
+    groupName: str
+    fields: List[ConfigField]
+
+class BackendDefinition(BaseModel):
+    docker_image: Optional[str] = None
+    environment_variables: Optional[Dict[str, str]] = None
+    container_id: Optional[str] = None
+
+class LayoutType(str, Enum):
+    SINGLE_COLUMN = "single-column"
+    TWO_COLUMN = "two-column"
+    DASHBOARD = "dashboard"
+    CHAT_INTERFACE = "chat-interface"
+    SMART_LIST_BUILDER = "smart-list-builder"
+
+class ComponentType(str, Enum):
+    CHAT_WINDOW = "chat-window"
+    HEADER = "header"
+    FOOTER = "footer"
+    SIDEBAR = "sidebar"
+    MAIN_CONTENT = "main-content"
+    DATA_TABLE = "data-table"
+    CHART = "chart"
+    FORM = "form"
+    BUTTON = "button"
+    TEXT = "text"
+    IMAGE = "image"
+    VIDEO = "video"
+    CUSTOM_COMPONENT = "custom-component"
+    TABS = "tabs"
+    TAB = "tab"
+    FORM_ITEM = "form-item"
+    INPUT = "input"
+    TEXTAREA = "textarea"
+    UPLOAD = "upload"
+
+class ComponentPosition(BaseModel):
+    row: Optional[int] = None
+    column: Optional[int] = None
+    area: Optional[str] = None
+
+class RenderComponent(BaseModel):
+    type: ComponentType
+    position: Optional[ComponentPosition] = None
+    properties: Optional[Dict[str, Any]] = None
+    children: Optional[List["RenderComponent"]] = None
+    injector: Optional[Any] = None
+
+class ActionDefinition(BaseModel):
+    type: str
+    method: str
+    url: str
+    data: Optional[Any] = None
+    state: Optional[str] = None
+    onSuccess: Optional[str] = None
+
+class RenderDefinition(BaseModel):
+    layout: LayoutType
+    components: List[RenderComponent]
+    actions: Optional[Dict[str, ActionDefinition]] = None
+    initialActions: Optional[List[str]] = None
+
+class ConfigValue(BaseModel):
+    name: str
+    value: Any
+    fields: Optional[List["ConfigValue"]] = None
+
+    class Config:
+        populate_by_name = True
+
+ConfigValue.model_rebuild()
+
+class SchemaDefinition(BaseModel):
+    name: str
+    version: str
+    fields: List[ConfigField]
+    migrations: Optional[List[Dict[str, Any]]] = None
+
+##
+# AGENT
+##
+
+class Agent(BaseModel):
+    id: Optional[UUID] = None
+    name: str
+    description: str
+    image_url: str
+    expertise: List[str]
+    version: str
+    schema_definition: Optional[List[SchemaDefinition]] = None
+    config_definition: List[ConfigGroup]
+    agent_config_definition: Optional[List[ConfigGroup]] = None
+    agent_configuration: Optional[List[ConfigValue]] = None
+    render_definition: RenderDefinition
+    backend: BackendDefinition
+    repository_url: Optional[str] = None
+    status: Optional[AgentStatus] = AgentStatus.PENDING
+
+    organization_id: Optional[UUID] = None
+    created_by: Optional[str] = None
+
+    # changed from datetime -> milliseconds since epoch
+    created_at: Optional[int] = None
+
+    updated_by: Optional[str] = None
+    updated_at: Optional[int] = None
+
+    class Config:
+        from_attributes = True
+        populate_by_name = True
+
+##
+# AGENT INSTANCE
+##
+
+class AgentInstance(BaseModel):
+    id: Optional[UUID] = None
+    published_agent_id: UUID
+    name: str
+    description: str
+    image_url: str
+    expertise:  Optional[List[str]] = None
+    version: str
+    schema_definition: Optional[List[SchemaDefinition]] = None
+    configuration: List[ConfigValue]
+    render_definition: RenderDefinition = None
+    backend: BackendDefinition = None
+    status: AgentInstanceStatus = AgentInstanceStatus.INACTIVE
+    status_message: Optional[str] = None
+    service_url: Optional[str] = None
+    port: Optional[int] = None
+
+    organization_id: Optional[UUID] = None
+    created_by: Optional[str] = None
+    # changed from datetime -> milliseconds
+    created_at: Optional[int] = None
+    updated_by: Optional[str] = None
+    updated_at: Optional[int] = None
+
+    class Config:
+        from_attributes = True
+        populate_by_name = True
+
+
+##
+# AGENT INSTANCE DATA
+##
+
+class AgentInstanceData(BaseModel):
+    id: Optional[UUID] = None
+    type: str
+    version: str
+    agent_instance_id: UUID
+    organization_id: UUID
+    data: Dict[str, Any]
+    # changed from datetime -> milliseconds
+    created_at: Optional[int] = None
+    updated_at: Optional[int] = None
+
+    class Config:
+        populate_by_name = True
+
+
+##
+# SOURCE
+##
+
+class SourceType(str, Enum):
+    FILE = "file"
+    GOOGLE_DOC = "google_doc"
+    WEBSITE = "website"
+    YOUTUBE = "youtube"
+    TEXT = "text"
+
+class SourceBase(BaseModel):
+    source_type: SourceType
+    title: Optional[str] = None
+    source_metadata: Optional[dict] = None
+    blob_url: Optional[HttpUrl] = None
+    content: Optional[str] = None
+
+class SourceCreate(SourceBase):
+    pass
+
+class SourceUpdate(SourceBase):
+    pass
+
+class Source(SourceBase):
+    id: UUID
+    agent_instance_id: UUID
+    # changed from datetime -> milliseconds
+    created_at: Optional[int] = None
+
+    class Config:
+        populate_by_name = True
+
+##
+# INTEGRATION
+##
+
+class IntegrationStatus(str, Enum):
+    CONNECTED = "connected"
+    DISCONNECTED = "disconnected"
+    COMING_SOON = "coming soon"
+
+class IntegrationBase(BaseModel):
+    name: str
+    label: str
+    icon: Optional[str] = None
+    description: Optional[str] = None
+    category: Optional[str] = None
+    tags: Optional[List[str]] = None
+    status: IntegrationStatus = IntegrationStatus.DISCONNECTED
+    configuration: Optional[List["ConfigValue"]] = None
+
+class IntegrationCreate(IntegrationBase):
+    agent_instance_id: UUID
+    organization_id: UUID
+
+class IntegrationUpdate(BaseModel):
+    name: Optional[str] = None
+    label: Optional[str] = None
+    icon: Optional[str] = None
+    description: Optional[str] = None
+    category: Optional[str] = None
+    tags: Optional[List[str]] = None
+    status: Optional[IntegrationStatus] = None
+    configuration: Optional[List["ConfigValue"]] = None
+
+class Integration(IntegrationBase):
+    id: UUID
+    agent_instance_id: UUID
+    organization_id: UUID
+    # changed from datetime -> milliseconds
+    created_at: Optional[int] = None
+
+    class Config:
+        from_attributes = True
+        populate_by_name = True
+
+Integration.model_rebuild()
+IntegrationUpdate.model_rebuild()

dhisana-0.0.1.dev18/src/dhisana/schemas/sales.py

@@ -0,0 +1,276 @@
+from uuid import UUID
+from pydantic import BaseModel, Field
+from typing import List, Optional, Dict, Any
+from enum import Enum
+from typing import Optional, List, Dict, Literal
+
+from dhisana.schemas.common import User
+
+# -----------------------------
+# Lead-List-Specific Schemas
+# -----------------------------
+
+class Lead(BaseModel):
+    id: Optional[str] = None
+    full_name: Optional[str] = None
+    first_name: Optional[str] = None
+    last_name: Optional[str] = None
+    email: Optional[str] = None
+    user_linkedin_url: Optional[str] = None
+    primary_domain_of_organization: Optional[str] = None
+    job_title: Optional[str] = None
+    phone: Optional[str] = None
+    headline: Optional[str] = None
+    lead_location: Optional[str] = None
+    organization_name: Optional[str] = None
+    organization_website: Optional[str] = None
+    summary_about_lead: Optional[str] = None
+    keywords: Optional[List[str]] = None
+    workflow_stage: Optional[str] = None
+    assigned_to: Optional[str] = None
+    engaged: Optional[bool] = None
+    last_contact: Optional[int] = None
+    additional_properties: Optional[Dict[str, str]] = None
+    research_summary: Optional[str] = None
+    task_ids: Optional[List[str]] = None
+    email_validation_status: Optional[
+        Literal["not_started", "in_progress", "valid", "invalid"]
+    ] = None
+    linkedin_validation_status: Optional[
+        Literal["not_started", "in_progress", "valid", "invalid"]
+    ] = None
+    research_status: Optional[
+        Literal["not_started", "in_progress", "done", "failed"]
+    ] = None
+    enchrichment_status: Optional[
+        Literal["not_started", "in_progress", "done", "failed"]
+    ] = None
+
+
+class LeadList(BaseModel):
+    id: Optional[str] = None
+    name: str
+    sources: List[str]
+    tags: List[str]
+    category: str
+    leads_count: int
+    assigned_users: List[str]
+    updated_at: int
+    status: Literal["connected", "disconnected", "coming soon"]
+    leads: Optional[List[Lead]] = None
+    public: Optional[bool] = None
+
+# -----------------------------
+# Task-Specific Schemas
+# -----------------------------
+
+class TaskStatus(str, Enum):
+    PENDING = "PENDING"
+    RUNNING = "RUNNING"
+    FAILED = "FAILED"
+    COMPLETED = "COMPLETED"
+
+class TaskBase(BaseModel):
+    name: str
+    task_type: str
+    data_id: Optional[UUID] = None
+    data_type: Optional[str] = None
+
+    inputs: Optional[List[Dict[str, Any]]] = []
+    outputs: Optional[List[Dict[str, Any]]] = []
+
+class TaskCreate(TaskBase):
+    pass
+
+class TaskUpdate(BaseModel):
+    status: Optional[TaskStatus] = None
+    logs: Optional[List[Any]] = None
+    metrics: Optional[Dict[str, Any]] = None
+    outputs: Optional[List[Dict[str, Any]]] = None
+
+class Task(TaskBase):
+    id: UUID
+    status: TaskStatus
+    logs: List[Any] = []
+    metrics: Dict[str, Any] = {}
+    created_at: int                 # store as ms since epoch
+    updated_at: int
+    completed_at: Optional[int] = None
+
+    class Config:
+        from_attributes = True
+
+
+# -----------------------------
+# Campaign-Specific Schemas
+# -----------------------------
+
+class SendRules(BaseModel):
+    daily_send_limit: Optional[int] = None
+    concurrency_limit: Optional[int] = None
+    time_window_start: Optional[str] = None  # "HH:MM" string
+    time_window_end: Optional[str] = None
+    block_weekends: Optional[bool] = False
+
+class Touch(BaseModel):
+    type: str            # e.g. 'email', 'linkedin', ...
+    action: str          # e.g. 'view_linked_in_profile', 'send_connection_request'
+    details: str
+    delay_days: int
+    template_id: Optional[str] = None
+
+class PromptEngineeringGuidance(BaseModel):
+    tone: str
+    word_count: int
+    paragraphs: int
+
+class LeadLog(BaseModel):
+    id: Optional[str] = None
+    message: str
+    channel: Optional[str] = None
+    timestamp: int                   # ms since epoch
+    status: Optional[str] = None
+
+class CampaignLead(BaseModel):
+    id: Optional[str] = None
+    campaign_id: str
+    lead_list_id: Optional[str] = None
+    lead_id: Optional[str] = None
+    lead_name: str
+
+    status: Optional[str] = None  # 'PENDING', 'WAITING_APPROVAL', 'OUTBOUND_PENDING', 'COMPLETED'
+    current_step: Optional[int] = 0
+    total_steps: Optional[int] = 0
+    engaged: Optional[bool] = False
+    last_touch: Optional[str] = None
+    created_at: Optional[int] = None      # ms since epoch
+    updated_at: Optional[int] = None
+
+    logs: Optional[List[LeadLog]] = None
+
+class CampaignCounter(BaseModel):
+    id: Optional[str] = None
+    campaign_id: str
+    date: str                 # "YYYY-MM-DD"
+    daily_sends: int
+    current_concurrency: int
+
+class CampaignStatus(str, Enum):
+    DRAFT = "DRAFT"
+    ACTIVE = "ACTIVE"
+    PAUSED = "PAUSED"
+    COMPLETED = "COMPLETED"
+
+class PendingEvent(BaseModel):
+    event_id: str
+    lead_id: str
+    touch_index: int
+    channel: str
+    action: str
+    subject: str
+    message: str
+    created_at: int             # ms since epoch
+
+class Campaign(BaseModel):
+    id: str
+    name: str
+    description: str
+    lead_lists: List[str]
+    run_mode: str
+    updated_at: int                       # ms since epoch
+
+    tags: Optional[List[str]] = None
+    channel: Optional[str] = None
+    mission_objective: Optional[str] = None
+    mission_progress: Optional[int] = None
+    ai_alerts: Optional[List[str]] = None
+    automatic_adjustments: Optional[List[str]] = None
+
+    product_name: Optional[str] = None
+    value_prop: Optional[str] = None
+    call_to_action: Optional[str] = None
+    pain_points: Optional[List[str]] = None
+    proof_points: Optional[List[str]] = None
+    prompt_engineering_guidance: Optional[PromptEngineeringGuidance] = None
+    prompt_templates: Optional[List[Dict[str, Any]]] = None
+    touches: Optional[List[Touch]] = None
+    send_rules: Optional[SendRules] = None
+
+    status: Optional[CampaignStatus] = None
+    start_date: Optional[int] = None      # ms since epoch
+    pause_date: Optional[int] = None
+    end_date: Optional[int] = None
+
+    leads: Optional[List[CampaignLead]] = None
+    counter: Optional[CampaignCounter] = None
+    pending_events: Optional[List[PendingEvent]] = None
+
+    class Config:
+        from_attributes = True
+
+class EmailContentContext(BaseModel):
+    product_name: Optional[str] = None
+    value_prop: Optional[str] = None
+    call_to_action: Optional[str] = None
+    pain_points: Optional[List[str]] = None
+    proof_points: Optional[List[str]] = None
+    prompt_engineering_guidance: Optional[PromptEngineeringGuidance] = None
+
+    lead_info: Lead
+    
+    sender_full_name: str
+    sender_first_name: str
+    sender_last_name: str
+
+    email_subject_template: str
+    email_body_template: str
+    external_source_fileIds: Dict[str, Any]
+    external_openai_vector_store_id: Optional[str]
+    additional_instructions: str
+    
+# --------------------------------------------------------------------
+# 1. Define your HubSpotLeadInformation model
+# --------------------------------------------------------------------
+class HubSpotLeadInformation(BaseModel):
+    full_name: str = Field("", description="Full name of the lead")
+    first_name: str = Field("", description="First name of the lead")
+    last_name: str = Field("", description="Last name of the lead")
+    email: str = Field("", description="Email address of the lead")
+    user_linkedin_url: str = Field("", description="LinkedIn URL of the lead")
+    primary_domain_of_organization: str = Field("", description="Primary domain of the organization")
+    job_title: str = Field("", description="Job Title of the lead")
+    phone: str = Field("", description="Phone number of the lead")
+    headline: str = Field("", description="Headline of the lead")
+    lead_location: str = Field("", description="Location of the lead")
+    organization_name: str = Field("", description="Current Company where lead works")
+    organization_website: str = Field("", description="Current Company website of the lead")
+    # additional_properties is dict of dict-of-strings. 
+    # We store all unmapped HubSpot fields as string => string.
+    additional_properties: Dict[str, Dict[str, str]] = Field(
+        default_factory=lambda: {"hubspot_lead_information": {}},
+        description="Store all extra HubSpot fields not mapped to standard ones"
+    )
+
+# --------------------------------------------------------------------
+# 2. Map HubSpot property names -> HubSpotLeadInformation fields
+# --------------------------------------------------------------------
+HUBSPOT_TO_LEAD_MAPPING = {
+    "firstname": "first_name",
+    "lastname": "last_name",
+    "email": "email",
+    "phone": "phone",
+    "jobtitle": "job_title",            # Default HubSpot job title property
+    "company": "organization_name",     # Map "company" -> "organization_name"
+    "website": "organization_website",  # Map "website" -> "organization_website"
+    "address": "lead_location",         # You can choose "city", "state", etc. if you prefer
+    "city": "lead_location",
+    "domain": "primary_domain_of_organization",
+}
+
+# Potential LinkedIn fallback properties from hubspot.
+LINKEDIN_FALLBACK_FIELDS = [
+    "person_profile_url",
+    "user_linkedin_url",
+    "lead_linkedin_url",
+    "contact_linkedin_url",
+]