ai-parrot 0.1.0__cp311-cp311-manylinux_2_28_x86_64.whl

parrot/utils/toml.py

@@ -0,0 +1,11 @@
+from .parsers import TOMLParser
+
+
+async def parse_toml_config(config_dir: str) -> dict:
+    try:
+        parser = TOMLParser()
+        return await parser.parse(config_dir)
+    except Exception as exc:
+        raise ValueError(
+            f"Error Parsing TOML Config on {config_dir}: {exc}"
+        )

parrot/utils/uv.py

@@ -0,0 +1,11 @@
+import asyncio
+
+
+def install_uvloop():
+    """install uvloop and set as default loop for asyncio."""
+    try:
+        import uvloop  # noqa # pylint: disable=import-outside-toplevel
+        asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
+        uvloop.install()
+    except ImportError:
+        pass

parrot/version.py

@@ -0,0 +1,10 @@
+"""Nav Parrot Meta information."""
+
+__title__ = "ai-parrot"
+__description__ = "Live Chatbots based on Langchain chatbots and Agents \
+    Integrated into Navigator Framework or used into aiohttp applications."
+__version__ = "0.1.0"
+__author__ = "Jesus Lara"
+__author_email__ = "jesuslarag@gmail.com"
+__license__ = "MIT"
+__copyright__ = 'Copyright (c) 2020-2024 Jesus Lara'

resources/users/__init__.py

@@ -0,0 +1,5 @@
+from .models import User, UserIdentity
+from .handlers import ADUserHandler, ADPeopleHandler
+
+
+__all__ = ('User', 'ADUserHandler', 'ADPeopleHandler', )

resources/users/handlers.py

@@ -0,0 +1,13 @@
+from datamodel import BaseModel
+from navigator.views import ModelView
+from .models import ADUser, ADPeople
+
+
+class ADUserHandler(ModelView):
+    model: BaseModel = ADUser
+    pk: str = 'people_id'
+
+
+class ADPeopleHandler(ModelView):
+    model: BaseModel = ADPeople
+    pk: str = 'people_id'

resources/users/models.py

@@ -0,0 +1,205 @@
+from typing import Optional, List
+from datetime import datetime, timedelta
+from uuid import UUID
+from datamodel.types import Text
+from asyncdb.models import Column, Model
+from navigator_auth.conf import AUTH_DB_SCHEMA, AUTH_USERS_VIEW
+
+
+class User(Model):
+    """Basic User notation."""
+
+    user_id: int = Column(
+        required=False,
+        primary_key=True,
+        db_default="auto"
+    )
+    first_name: str
+    last_name: str
+    display_name: str
+    email: str = Column(required=False, max=254)
+    alt_email: str = Column(required=False, max=254)
+    password: str = Column(required=False, max=128)
+    last_login: datetime = Column(required=False)
+    username: str = Column(required=False)
+    is_superuser: bool = Column(required=True, default=False)
+    is_active: bool = Column(required=True, default=True)
+    is_new: bool = Column(required=True, default=True)
+    is_staff: bool = Column(required=False, default=True)
+    title: str = Column(equired=False, max=90)
+    avatar: str = Column(max=512)
+    associate_id: str = Column(required=False)
+    associate_oid: str = Column(required=False)
+    department_code: str = Column(required=False)
+    job_code: str = Column(required=False)
+    position_id: str = Column(required=False)
+    group_id: list = Column(required=False)
+    groups: list = Column(required=False)
+    program_id: list = Column(required=False)
+    programs: list = Column(required=False)
+    start_date: datetime = Column(required=False)
+    birthday: str = Column(required=False)
+    worker_type: str = Column(required=False)
+    created_at: datetime = Column(required=False)
+
+    def birth_date(self):
+        if self.birthday:
+            _, month, day = self.birthday.split('-')
+            # Get the current year
+            current_year = datetime.now().year
+            # Create a new date string with the current year
+            new_date_str = f"{current_year}-{month}-{day}"
+            # Convert the new date string to a datetime object
+            return datetime.strptime(new_date_str, "%Y-%m-%d").date()
+        return None
+
+    def employment_duration(self):
+        if not self.start_date:
+            return None, None, None
+        # Get today's date
+        today = datetime.now().date()
+        # employment:
+        employment = self.start_date
+
+        # Calculate the difference in years, months, days
+        years = today.year - employment.year
+        months = today.month - employment.month
+        days = today.day - employment.day
+
+        # Adjust for cases where the current month is before the start month
+        if months < 0:
+            years -= 1
+            months += 12
+
+        # Adjust for cases where the current day
+        # is before the start day in the month
+        if days < 0:
+            # Subtract one month and calculate days based on the previous month
+            months -= 1
+            if months < 0:
+                years -= 1
+                months += 12
+            # Calculate the last day of the previous month
+            last_day_of_prev_month = (
+                today.replace(day=1) - timedelta(days=1)
+            ).day
+            days += last_day_of_prev_month
+
+        # Adjust months and years again if necessary
+        if months < 0:
+            years -= 1
+            months += 12
+
+        return years, months, days
+
+    class Meta:
+        driver = "pg"
+        name = AUTH_USERS_VIEW
+        schema = AUTH_DB_SCHEMA
+        description = 'View Model for getting Users.'
+        strict = True
+        frozen = False
+
+
+class UserIdentity(Model):
+
+    identity_id: UUID = Column(
+        required=False,
+        primary_key=True,
+        db_default="auto",
+        repr=False
+    )
+    display_name: str = Column(required=False)
+    title: str = Column(required=False)
+    nickname: str = Column(required=False)
+    email: str = Column(required=False)
+    phone: str = Column(required=False)
+    short_bio: Text = Column(required=False)
+    avatar: str = Column(required=False)
+    user_id: User = Column(required=False, repr=False)
+    auth_provider: str = Column(required=False)
+    auth_data: Optional[dict] = Column(required=False, repr=False)
+    attributes: Optional[dict] = Column(required=False, repr=False)
+    created_at: datetime = Column(
+        required=False,
+        default=datetime.now(),
+        repr=False
+    )
+
+    class Meta:
+        driver = "pg"
+        name = "user_identities"
+        description = 'Manage User Identities.'
+        schema = AUTH_DB_SCHEMA
+        strict = True
+
+class ADUser(Model):
+    """Active Directory Users."""
+    people_id: UUID = Column(
+        required=False,
+        primary_key=True,
+        db_default="auto",
+        repr=False
+    )
+    username: str = Column(required=False)
+    display_name: str = Column(required=False)
+    given_name: str = Column(required=False)
+    last_name: str = Column(required=False)
+    phones: Optional[list] = Column(required=False)
+    mobile: str = Column(required=False)
+    job_title: str = Column(required=False)
+    email: str = Column(required=False)
+    office_location: str = Column(required=False)
+    preferred_language: str = Column(required=False)
+    associate_id: str = Column(required=False)
+    associate_oid: str = Column(required=False)
+    job_code_title: str = Column(required=False)
+    position_id: str = Column(required=False)
+    zammad_created: bool = Column(required=False, default=False)
+    created_at: datetime = Column(
+        required=False,
+        default=datetime.now(),
+        repr=False
+    )
+
+    class Meta:
+        name = 'people'
+        schema = 'troc'
+        strict = True
+
+
+class ADPeople(Model):
+    """Active Directory Users."""
+    people_id: UUID = Column(
+        required=False,
+        primary_key=True,
+        db_default="auto",
+        repr=False
+    )
+    user_id: int = Column(required=True)
+    userid: UUID = Column(required=False)
+    username: str = Column(required=False)
+    display_name: str = Column(required=False)
+    given_name: str = Column(required=False)
+    last_name: str = Column(required=False)
+    phones: Optional[list] = Column(required=False)
+    mobile: str = Column(required=False)
+    job_title: str = Column(required=False)
+    email: str = Column(required=False)
+    alt_email: str = Column(required=False)
+    office_location: str = Column(required=False)
+    preferred_language: str = Column(required=False)
+    associate_id: str = Column(required=False)
+    associate_oid: str = Column(required=False)
+    job_code_title: str = Column(required=False)
+    position_id: str = Column(required=False)
+    created_at: datetime = Column(
+        required=False,
+        default=datetime.now(),
+        repr=False
+    )
+
+    class Meta:
+        name = 'vw_people'
+        schema = 'troc'
+        strict = True

settings/settings.py

@@ -0,0 +1,51 @@
+"""
+Settings Configuration for Parrot-AI.
+
+Basic Configuration.
+"""
+from navconfig import config, BASE_DIR
+from navconfig.logging import logging
+
+
+logging.getLogger(name='sentence_transformers').setLevel(logging.WARNING)
+
+AUTHENTICATION_BACKENDS = (
+    'navigator_auth.backends.AzureAuth',
+    'navigator_auth.backends.APIKeyAuth',
+    'navigator_auth.backends.TokenAuth',
+    'navigator_auth.backends.TrocToken',
+    'navigator_auth.backends.ADFSAuth',
+    'navigator_auth.backends.Oauth2Provider',
+    'navigator_auth.backends.BasicAuth'
+)
+
+# Azure Bot:
+MS_TENANT_ID = config.get('MS_TENANT_ID')
+MS_CLIENT_ID = config.get('MS_CLIENT_ID')
+MS_CLIENT_SECRET = config.get('MS_CLIENT_SECRET')
+
+# New Bot:
+NEW_CLIENT_ID = config.get('NEW_CLIENT_ID')
+NEW_CLIENT_SECRET = config.get('NEW_CLIENT_SECRET')
+
+# Other Bots:
+BOSE_CLIENT_ID = config.get('BOSE_CLIENT_ID')
+BOSE_CLIENT_SECRET = config.get('BOSE_CLIENT_SECRET')
+ODOO_CLIENT_ID = config.get('ODOO_CLIENT_ID')
+ODOO_CLIENT_SECRET = config.get('ODOO_CLIENT_SECRET')
+ASKBRETT_CLIENT_ID = config.get('ASKBRETT_CLIENT_ID')
+ASKBRETT_CLIENT_SECRET = config.get('ASKBRETT_CLIENT_SECRET')
+TROCERS_CLIENT_ID = config.get('TROCERS_CLIENT_ID')
+TROCERS_CLIENT_SECRET = config.get('TROCERS_CLIENT_SECRET')
+BOTTROCDEV_CLIENT_ID = config.get('BOTTROCDEV_CLIENT_ID')
+BOTTROCDEV_CLIENT_SECRET = config.get('BOTTROCDEV_CLIENT_SECRET')
+ATTBOT_CLIENT_ID = config.get('ATTBOT_CLIENT_ID')
+ATTBOT_CLIENT_SECRET = config.get('ATTBOT_CLIENT_SECRET')
+
+
+# ScyllaDB Configuration:
+SCYLLADB_HOST = config.get('SCYLLADB_HOST', fallback='localhost')
+SCYLLADB_PORT = int(config.get('SCYLLADB_PORT', fallback=9042))
+SCYLLADB_USERNAME = config.get('SCYLLADB_USERNAME', fallback='navigator')
+SCYLLADB_PASSWORD = config.get('SCYLLADB_PASSWORD', fallback='navigator')
+SCYLLADB_KEYSPACE = config.get('SCYLLADB_KEYSPACE', fallback='navigator')