dhisana 0.0.1.dev7__tar.gz → 0.0.1.dev9__tar.gz

{dhisana-0.0.1.dev7 → dhisana-0.0.1.dev9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dhisana
-Version: 0.0.1.dev7
+Version: 0.0.1.dev9
 Summary: A Python SDK for Dhisana AI Platform
 Home-page: https://github.com/dhisana-ai/dhisana-python-sdk
 Author: Admin
@@ -23,3 +23,5 @@ Requires-Dist: uvicorn[standard]
 Requires-Dist: aiohttp
 Requires-Dist: openapi_pydantic
 Requires-Dist: pandas
+Requires-Dist: simple_salesforce
+Requires-Dist: backoff

{dhisana-0.0.1.dev7 → dhisana-0.0.1.dev9}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='dhisana',  
-    version='0.0.1-dev7',
+    version='0.0.1-dev9',
     description='A Python SDK for Dhisana AI Platform',
     author='Admin',
     author_email='contact@dhisana.ai',
@@ -22,7 +22,9 @@ setup(
         'uvicorn[standard]',
         'aiohttp',
         'openapi_pydantic',
-        'pandas'
+        'pandas',
+        'simple_salesforce',
+        'backoff'
     ],
     entry_points={
         'console_scripts': [

{dhisana-0.0.1.dev7 → dhisana-0.0.1.dev9}/src/dhisana/ui/components.py

@@ -1,11 +1,9 @@
-from typing import List, Dict, Any, Optional
-
+from typing import List, Dict, Any, Optional, Union
 
 class Component:
     def to_dict(self) -> Dict[str, Any]:
         raise NotImplementedError("Must implement to_dict method.")
 
-
 class Header(Component):
     def __init__(self, title: str, subtitle: Optional[str] = None, logo: Optional[str] = None):
         self.title = title
@@ -22,7 +20,6 @@ class Header(Component):
             },
         }
 
-
 class Footer(Component):
     def __init__(self, content: str):
         self.content = content
@@ -35,7 +32,6 @@ class Footer(Component):
             },
         }
 
-
 class Sidebar(Component):
     def __init__(self, items: List[str]):
         self.items = items
@@ -48,19 +44,6 @@ class Sidebar(Component):
             },
         }
 
-class Text(Component):
-    def __init__(self, content: str):
-        self.content = content
-
-    def to_dict(self):
-        return {
-            'type': 'text',
-            'properties': {
-                'content': self.content,
-            },
-        }
-    
-
 class MainContent(Component):
     def __init__(self, children: List[Component]):
         self.children = children
@@ -88,29 +71,39 @@ class ChatWindow(Component):
             },
         }
 
-
 class DataTable(Component):
     def __init__(
-        self,
-        columns: List[Dict[str, Any]],
-        data_source: str,
-        actions: Optional[List[Dict[str, Any]]] = None,
-    ):
+            self,             
+            columns: List[Dict[str, Any]], 
+            data_source: str, 
+            actions: Optional[List[Dict[str, Any]]] = None,
+            row_selection: Optional[bool] = False,
+            selected_row_keys: Optional[List[str]] = None,
+            on_selection_change: Optional[str] = None,
+            title: Optional[str] = None,
+        ):
+        self.title = title
         self.columns = columns
-        self.data_source = data_source
+        self.data_source = data_source  # Should be a reference to data in dataContext
         self.actions = actions or []
+        self.row_selection = row_selection
+        self.selected_row_keys = selected_row_keys
+        self.on_selection_change = on_selection_change
 
     def to_dict(self):
         return {
             'type': 'data-table',
             'properties': {
+                'title': self.title,
                 'columns': self.columns,
-                'dataSource': self.data_source,
+                'dataSource': self.data_source,  # Should be in the form '{{dataKey}}'
                 'actions': self.actions,
+                'rowSelection': self.row_selection,
+                'selectedRowKeys': self.selected_row_keys,
+                'onSelectionChange': self.on_selection_change,
             },
         }
 
-
 class Chart(Component):
     def __init__(self, chart_type: str, data_source: str, options: Optional[Dict[str, Any]] = None):
         self.chart_type = chart_type
@@ -127,7 +120,6 @@ class Chart(Component):
             },
         }
 
-
 class Form(Component):
     def __init__(self, children: List[Component], on_submit: List[str]):
         self.children = children
@@ -142,43 +134,51 @@ class Form(Component):
             'children': [child.to_dict() for child in self.children],
         }
 
-
 class FormItem(Component):
-    def __init__(self, label: str, children: List[Component]):
+    def __init__(self, label: str, children: List[Component], visible: bool = True):
         self.label = label
         self.children = children
+        self.visible = visible
 
     def to_dict(self):
         return {
             'type': 'form-item',
             'properties': {
                 'label': self.label,
+                'visible': self.visible,
             },
             'children': [child.to_dict() for child in self.children],
         }
 
-
 class Input(Component):
-    def __init__(self, name: str, placeholder: str = '', required: bool = False):
+    def __init__(self, name: str, type: str = 'text', placeholder: str = '', required: bool = False, value: str = None, checked: bool = False):
         self.name = name
+        self.type = type
         self.placeholder = placeholder
         self.required = required
+        self.value = value
+        self.checked = checked
 
     def to_dict(self):
         return {
             'type': 'input',
             'properties': {
                 'name': self.name,
+                'type': self.type,
                 'placeholder': self.placeholder,
                 'required': self.required,
+                'value': self.value,
+                'checked': self.checked,
             },
         }
 
-
 class TextArea(Component):
-    def __init__(self, name: str, placeholder: str = ''):
+    def __init__(self, name: str, placeholder: str = '', value: str = None, required: bool = False, rows: int = 3):
         self.name = name
         self.placeholder = placeholder
+        self.value = value
+        self.required = required
+        self.rows = rows
 
     def to_dict(self):
         return {
@@ -186,14 +186,17 @@ class TextArea(Component):
             'properties': {
                 'name': self.name,
                 'placeholder': self.placeholder,
+                'value': self.value,
+                'required': self.required,
+                'rows': self.rows,
             },
         }
 
-
 class Upload(Component):
-    def __init__(self, name: str, required: bool = False):
+    def __init__(self, name: str, required: bool, multiple: bool = False):
         self.name = name
         self.required = required
+        self.multiple = multiple
 
     def to_dict(self):
         return {
@@ -201,16 +204,17 @@ class Upload(Component):
             'properties': {
                 'name': self.name,
                 'required': self.required,
+                'multiple': self.multiple,
             },
         }
 
-
 class Button(Component):
-    def __init__(self, label: str, button_type: str = 'button', disabled: bool = False, on_click: Optional[str] = None):
+    def __init__(self, label: str, button_type: str = 'button', disabled: bool = False, on_click: Optional[str] = None, style: Optional[Dict[str, Any]] = None):
         self.label = label
         self.button_type = button_type
         self.disabled = disabled
         self.on_click = on_click
+        self.style = style
 
     def to_dict(self):
         return {
@@ -220,9 +224,74 @@ class Button(Component):
                 'type': self.button_type,
                 'disabled': self.disabled,
                 'onClick': self.on_click,
+                'style': self.style,
+            },
+        }
+
+class Text(Component):
+    def __init__(self, content: Union[str, List[Dict[str, Any]]]):
+        self.content = content
+
+    def to_dict(self):
+        return {
+            'type': 'text',
+            'properties': {
+                'content': self.content,
             },
         }
 
+class Select(Component):
+    def __init__(
+            self, 
+            name: str, 
+            options: List[Dict[str, Any]], 
+            multiple: bool = False, 
+            required: bool = False,
+            placeholder: Optional[str] = None,
+            label_field: Optional[str] = 'label',
+            value_field: Optional[str] = 'value',
+            on_change: Optional[str] = None,
+        ):
+        self.name = name
+        self.options = options
+        self.multiple = multiple
+        self.required = required
+        self.placeholder = placeholder
+        self.value_field = value_field
+        self.label_field = label_field
+        self.on_change = on_change
+
+
+    def to_dict(self):
+        return {
+            'type': 'select',
+            'properties': {
+                'name': self.name,
+                'options': self.options,
+                'multiple': self.multiple,
+                'required': self.required,
+                'placeholder': self.placeholder,
+                'labelField': self.label_field,
+                'valueField': self.value_field,
+                'onChange': self.on_change,
+            },
+        }
+
+class Checkbox(Component):
+    def __init__(self, name: str, label: Optional[str] = None, checked: bool = False):
+        self.name = name
+        self.label = label
+        self.checked = checked
+
+    def to_dict(self):
+        return {
+            'type': 'checkbox',
+            'properties': {
+                'name': self.name,
+                'label': self.label,
+                'checked': self.checked,
+            },
+        }
 
 class Tabs(Component):
     def __init__(self, children: List['Tab']):
@@ -234,7 +303,6 @@ class Tabs(Component):
             'children': [child.to_dict() for child in self.children],
         }
 
-
 class Tab(Component):
     def __init__(self, label: str, children: List[Component]):
         self.label = label
@@ -249,7 +317,6 @@ class Tab(Component):
             'children': [child.to_dict() for child in self.children],
         }
 
-
 class ModalDialog(Component):
     def __init__(
         self,
@@ -259,7 +326,7 @@ class ModalDialog(Component):
         visible: bool = False,
         on_close: Optional[str] = None,
     ):
-        self.name = name 
+        self.name = name
         self.title = title
         self.content = content
         self.visible = visible
@@ -277,10 +344,11 @@ class ModalDialog(Component):
             'children': [component.to_dict() for component in self.content],
         }
 
-
 class Page(Component):
-    def __init__(self, name: str, path: str, components: List[Component]):
+    def __init__(self, name: str, label: str, main: bool, path: str, components: List[Component]):
         self.name = name
+        self.label = label
+        self.main = main
         self.path = path
         self.components = components
 
@@ -290,11 +358,12 @@ class Page(Component):
             'properties': {
                 'name': self.name,
                 'path': self.path,
+                'label': self.label,
+                'main': self.main,
             },
             'children': [component.to_dict() for component in self.components],
         }
 
-
 class Action:
     def __init__(
         self,
@@ -302,15 +371,25 @@ class Action:
         method: str,
         url: Optional[str] = None,
         data: Optional[Any] = None,
+        content_type: Optional[str] = None,
+        params: Optional[Dict[str, Any]] = None,
         state: Optional[str] = None,
-        on_success: Optional[str] = None,
+        before_action: Optional[str] = None,
+        on_success: Optional[Any] = None,
+        on_error: Optional[Any] = None,
+        actions: Optional[List[Union['Action', str]]] = None,
     ):
         self.action_type = action_type
         self.method = method
         self.url = url
         self.data = data
+        self.content_type = content_type
+        self.params = params
         self.state = state
+        self.before_action = before_action
         self.on_success = on_success
+        self.on_error = on_error
+        self.actions = actions or []
 
     def to_dict(self):
         return {
@@ -318,8 +397,13 @@ class Action:
             'method': self.method,
             'url': self.url,
             'data': self.data,
+            'contentType': self.content_type,
+            'params': self.params,
             'state': self.state,
+            'beforeAction': self.before_action,
             'onSuccess': self.on_success,
+            'onError': self.on_error,
+            'actions': [action.to_dict() if isinstance(action, Action) else action for action in self.actions],
         }
 
 

dhisana-0.0.1.dev9/src/dhisana/utils/agent_tools.py

@@ -0,0 +1,48 @@
+# Global List of tools that can be used in the assistant
+# Only functions marked with @assistant_tool will be available in the allowed list
+# This is in addition to the tools from OpenAPI Spec add to allowed tools
+# These tools are loaded in the agent like below
+
+# async def load_global_tool_function():
+#     # Iterate over all modules in the package
+#     for loader, module_name, is_pkg in pkgutil.walk_packages(global_tools.__path__):
+#         module = importlib.import_module(f"{global_tools.__name__}.{module_name}")
+#         for name in dir(module):
+#             func = getattr(module, name)
+#             if callable(func) and getattr(func, 'is_assistant_tool', False):
+#                 GLOBAL_TOOLS_FUNCTIONS[name] = func
+
+# Global Data Models Used for Data Extraction. These can be referenced in workflows.
+GLOBAL_DATA_MODELS = []
+
+# Global Functions used in workflows. Like CRM, Email, etc. They can be invoked as Python functions.
+GLOBAL_TOOLS_FUNCTIONS = {}
+
+# GLOBAL_TOOLS_FUNCTIONS in OPENAI function spec format
+# src/dhisana/utils/agent_tools.py
+
+# Global List of tools that can be used in the assistant
+# Only functions marked with @assistant_tool will be available in the allowed list
+# This is in addition to the tools from OpenAPI Spec add to allowed tools
+# These tools are loaded in the agent like below
+
+# async def load_global_tool_function():
+#     # Iterate over all modules in the package
+#     for loader, module_name, is_pkg in pkgutil.walk_packages(global_tools.__path__):
+#         module = importlib.import_module(f"{global_tools.__name__}.{module_name}")
+#         for name in dir(module):
+#             func = getattr(module, name)
+#             if callable(func) and getattr(func, 'is_assistant_tool', False):
+#                 GLOBAL_TOOLS_FUNCTIONS[name] = func
+
+# Ensure GLOBAL_DATA_MODELS is only initialized once
+if 'GLOBAL_DATA_MODELS' not in globals():
+    GLOBAL_DATA_MODELS = []
+
+# Ensure GLOBAL_TOOLS_FUNCTIONS is only initialized once
+if 'GLOBAL_TOOLS_FUNCTIONS' not in globals():
+    GLOBAL_TOOLS_FUNCTIONS = {}
+
+# Ensure GLOBAL_OPENAI_ASSISTANT_TOOLS is only initialized once
+if 'GLOBAL_OPENAI_ASSISTANT_TOOLS' not in globals():
+    GLOBAL_OPENAI_ASSISTANT_TOOLS = []

dhisana-0.0.1.dev9/src/dhisana/utils/apollo_tools.py

@@ -0,0 +1,121 @@
+import os
+import aiohttp
+from dhisana.utils.assistant_tool_tag import assistant_tool
+from typing import List, Optional, Union
+import backoff
+
+@assistant_tool
+@backoff.on_exception(
+    backoff.expo,
+    aiohttp.ClientResponseError,
+    max_tries=2,
+    giveup=lambda e: e.status != 429,
+    factor=10,
+)
+async def enrich_person_info_from_apollo(
+    linkedin_url: Optional[str] = None,
+    email: Optional[str] = None,
+    phone: Optional[str] = None,
+):
+    """
+    Fetch a person's details from Apollo using LinkedIn URL, email, or phone number.
+    
+    Parameters:
+    - **linkedin_url** (*str*, optional): LinkedIn profile URL of the person.
+    - **email** (*str*, optional): Email address of the person.
+    - **phone** (*str*, optional): Phone number of the person.
+
+    Returns:
+    - **dict**: JSON response containing person information.
+    """
+    APOLLO_API_KEY = os.environ.get('APOLLO_API_KEY')
+    if not APOLLO_API_KEY:
+        return {'error': "Apollo API key not found in environment variables"}
+
+    if not linkedin_url and not email and not phone:
+        return {'error': "At least one of linkedin_url, email, or phone must be provided"}
+
+    headers = {
+        "X-Api-Key": f"{APOLLO_API_KEY}",
+        "Content-Type": "application/json"
+    }
+
+    data = {}
+    if linkedin_url:
+        data['linkedin_url'] = linkedin_url
+    if email:
+        data['email'] = email
+    if phone:
+        data['phone_numbers'] = [phone]  # Apollo expects a list for phone numbers
+
+    url = 'https://api.apollo.io/v1/people/match'  # Correct API endpoint
+
+    async with aiohttp.ClientSession() as session:
+        async with session.post(url, headers=headers, json=data) as response:
+            if response.status == 200:
+                return await response.json()
+            elif response.status == 429:
+                raise aiohttp.ClientResponseError(
+                    request_info=response.request_info,
+                    history=response.history,
+                    status=response.status,
+                    message="Rate limit exceeded",
+                    headers=response.headers
+                )
+            else:
+                result = await response.json()
+                return {'error': result}
+
+
+
+@assistant_tool
+@backoff.on_exception(
+    backoff.expo,
+    aiohttp.ClientResponseError,
+    max_tries=2,
+    giveup=lambda e: e.status != 429,
+    factor=10,
+)
+async def enrich_company_info_from_apollo(
+    company_domain: Optional[str] = None,
+):
+    """
+    Fetch a company's details from Apollo using the company domain.
+    
+    Parameters:
+    - **company_domain** (*str*, optional): Domain of the company.
+
+    Returns:
+    - **dict**: JSON response containing company information.
+    """
+    APOLLO_API_KEY = os.environ.get('APOLLO_API_KEY')
+    if not APOLLO_API_KEY:
+        return {'error': "Apollo API key not found in environment variables"}
+
+    if not company_domain:
+        return {'error': "Company domain must be provided"}
+
+    headers = {
+        "X-Api-Key": f"{APOLLO_API_KEY}",
+        "Content-Type": "application/json",
+        "Cache-Control": "no-cache",
+        "accept": "application/json"
+    }
+
+    url = f'https://api.apollo.io/api/v1/organizations/enrich?domain={company_domain}'
+
+    async with aiohttp.ClientSession() as session:
+        async with session.get(url, headers=headers) as response:
+            if response.status == 200:
+                return await response.json()
+            elif response.status == 429:
+                raise aiohttp.ClientResponseError(
+                    request_info=response.request_info,
+                    history=response.history,
+                    status=response.status,
+                    message="Rate limit exceeded",
+                    headers=response.headers
+                )
+            else:
+                result = await response.json()
+                return {'error': result}

dhisana-0.0.1.dev9/src/dhisana/utils/check_email_validity_tools.py

@@ -0,0 +1,36 @@
+# Check validity of email
+
+import os
+import os
+import aiohttp
+from dhisana.utils.assistant_tool_tag import assistant_tool    
+
+@assistant_tool
+async def check_email_validity_with_zero_bounce(email_id: str):
+    """
+    Validate a single email address using the ZeroBounce API.
+
+    This function sends an asynchronous GET request to the ZeroBounce API to validate the provided email address.
+    
+    Parameters:
+    email_id (str): The email address to be validated.
+
+    Returns:
+    dict: The JSON response from the ZeroBounce API containing the validation results.
+
+    Raises:
+    ValueError: If the ZeroBounce API key is not found in the environment variables.
+    Exception: If the response status code from the ZeroBounce API is not 200.
+    """
+    ZERO_BOUNCE_API_KEY = os.environ.get('ZERO_BOUNCE_API_KEY')
+    if not ZERO_BOUNCE_API_KEY:
+        raise ValueError("ZeroBounce API key not found in environment variables")
+
+    url = f"https://api.zerobounce.net/v2/validate?api_key={ZERO_BOUNCE_API_KEY}&email={email_id}"
+
+    async with aiohttp.ClientSession() as session:
+        async with session.get(url) as response:
+            if response.status != 200:
+                raise Exception(f"Error: Received status code {response.status}")
+            result = await response.json()
+            return result