ai-parrot 0.17.2__cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

parrot/utils/types.pyx

@@ -0,0 +1,213 @@
+# cython: language_level=3, embedsignature=True, boundscheck=False, wraparound=True, initializedcheck=False
+# Copyright (C) 2018-present Jesus Lara
+#
+import time
+import orjson
+from typing import (
+    Dict
+)
+from collections.abc import Sequence
+from decimal import Decimal
+from numpy import int64, ndarray
+from datetime import timezone
+from cpython cimport datetime
+from zoneinfo import ZoneInfo
+from uuid import UUID
+from dateutil.parser import parse, ParserError
+
+
+
+cpdef object strtobool(str val):
+    """Convert a string representation of truth to true (1) or false (0).
+
+    True values are 'y', 'yes', 't', 'true', 'on', and '1'; false values
+    are 'n', 'no', 'f', 'false', 'off', and '0'.  Raises ValueError if
+    'val' is anything else.
+    """
+    val = val.lower()
+    if val in ('y', 'yes', 't', 'true', 'on', '1'):
+        return True
+    elif val in ('n', 'no', 'f', 'false', 'off', '0', 'null'):
+        return False
+    else:
+        raise ValueError(
+            f"invalid truth value for {val}"
+        )
+
+
+cpdef datetime.date to_date(object value, str mask = "%Y-%m-%d %H:%M:%S", str tz = None):
+    if isinstance(value, datetime.datetime):
+        return value
+    else:
+        try:
+            result = datetime.datetime.strptime(str(value), mask)
+            if tz is not None:
+                zone = ZoneInfo(key=tz)
+                result = result.replace(tzinfo=zone)
+            return result
+        except (TypeError, ValueError, AttributeError):
+            return parse(str(value))
+
+
+cpdef datetime.time to_time(object value, str mask = "%H:%M:%S"):
+    if value == 0:
+        return datetime.datetime.now().replace(
+            hour=0, minute=0, second=0, microsecond=0
+        )
+    if isinstance(value, datetime.time):
+        return value
+    elif isinstance(value, datetime.datetime):
+        return value.time()
+    else:
+        if len(str(value)) < 6:
+            value = str(value).zfill(6)
+        try:
+            return datetime.datetime.strptime(str(value), mask)
+        except ValueError:
+            return datetime.datetime.strptime(str(value), "%H:%M:%S")
+
+
+cpdef datetime.datetime to_datetime(object value, str mask = "%Y-%m-%d %H:%M:%S"):
+    if isinstance(value, datetime.datetime):
+        return value
+    elif isinstance(value, list):
+        dt = to_date(value[0], mask=mask[0])
+        mt = to_time(value[1], mask=mask[1]).time()
+        return datetime.datetime.combine(dt, mt)
+    else:
+        if value is None:
+            return datetime.datetime.now().replace(
+                hour=0, minute=0, second=0, microsecond=0
+            )
+        else:
+            return datetime.datetime.strptime(str(value), mask)
+
+
+cpdef datetime.datetime epoch_to_date(object value, str tz = None):
+    if value is None:
+        return None
+    if len(str(value)) == 10:
+        s = value
+    elif len(str(value)) == 19:
+        s = value / 1e9
+    else:
+        s, _ = divmod(value, 1000.0)
+    if tz is not None:
+        zone = ZoneInfo(key=tz)
+    else:
+        zone = timezone.utc
+    return datetime.datetime.fromtimestamp(s, zone)
+
+
+cpdef object to_boolean(object value):
+    if isinstance(value, bool):
+        return value
+    elif value is None:
+        return False
+    else:
+        try:
+            return strtobool(value)
+        except ValueError:
+            return False
+
+
+cpdef object is_date(object value):
+        if isinstance(value, (datetime.datetime, time, datetime.time, datetime.timedelta)):
+            return True
+        else:
+            try:
+                val = parse(value)
+                if val:
+                    return True
+                return False
+            except (ParserError, ValueError):
+                return False
+
+cdef class Entity:
+    """Entity.
+    Used to convert entities (string, number, dates) to appropiated string on SQL queries.
+    """
+
+    @classmethod
+    def is_integer(cls, _type):
+        return _type in (int, int64)
+
+    @classmethod
+    def is_number(cls, _type):
+        return _type in (int, int64, float, Decimal, bytes, bool)
+
+    @classmethod
+    def is_string(cls, _type):
+        return isinstance(_type, (
+            str,
+            datetime.datetime,
+            datetime.time,
+            datetime.timedelta,
+            UUID
+        ))
+
+    @classmethod
+    def is_date(cls, _type):
+        return _type in (datetime.datetime, time, datetime.time, datetime.timedelta)
+
+    @classmethod
+    def is_array(cls, t):
+        return isinstance(t,(list, dict, Sequence, ndarray))
+
+    @classmethod
+    def is_bool(cls, _type):
+        return isinstance(_type, bool)
+
+    @classmethod
+    def toSQL(cls, value, _type, dbtype: str = None):
+        v = f"{value!r}" if Entity.is_date(_type) else value
+        v = f"{value!s}" if Entity.is_string(_type) and value is not None else v
+        v = value if Entity.is_number(_type) else v
+        v = str(value) if isinstance(value, UUID) else v
+        # json conversion
+        v = orjson.dumps(value) if _type in [dict, Dict] else v
+        v = f"{value!s}" if dbtype == "array" and value is not None else v
+        # formatting htstore column
+        v = (
+            ",".join({"{}=>{}".format(k, v) for k, v in value.items()})
+            if isinstance(value, dict) and dbtype == "hstore"
+            else v
+        )
+        v = "NULL" if (value in ["None", "null"]) else v
+        v = "NULL" if value is None else v
+        return v
+
+    @classmethod
+    def escapeLiteral(cls, value, _type, dbtype: str = None):
+        v = value if value != "None" or value is not None else ""
+        v = f"{value!r}" if Entity.is_string(_type) else v
+        v = value if Entity.is_number(_type) else f"{value!r}"
+        v = f"array{value!s}" if dbtype == "array" else v
+        v = f"{value!r}" if dbtype == "hstore" else v
+        v = value if (value in ["None", "null", "NULL"]) else v
+        return v
+
+    @classmethod
+    def escapeString(cls, value):
+        v = value if value != "None" else ""
+        v = str(v).replace("'", "''")
+        v = "'{}'".format(v) if Entity.is_string(type(value)) else v
+        return v
+
+    @classmethod
+    def quoteString(cls, value):
+        v = value if value != "None" else ""
+        v = "'{}'".format(v) if type(v) == str else v
+        return v
+
+class SafeDict(dict):
+    """
+    SafeDict.
+
+    Allow to using partial format strings
+
+    """
+
+    def __missing__(self, key):
+        """Missing method for SafeDict."""
+        return "{" + key + "}"

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 any aiohttp applications."
+__version__ = "0.17.2"
+__author__ = "Jesus Lara"
+__author_email__ = "jesuslarag@gmail.com"
+__license__ = "MIT"
+__copyright__ = "Copyright (c) 2020-2024 Jesus Lara"

parrot/yaml-rs/Cargo.lock

@@ -0,0 +1,350 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 4
+
+[[package]]
+name = "autocfg"
+version = "1.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c08606f8c3cbf4ce6ec8e28fb0014a2c086708fe954eaa885384a6165172e7e8"
+
+[[package]]
+name = "bitflags"
+version = "2.10.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "812e12b5285cc515a9c72a5c1d3b6d46a19dac5acfef5265968c166106e31dd3"
+
+[[package]]
+name = "cfg-if"
+version = "1.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9330f8b2ff13f34540b44e946ef35111825727b38d33286ef986142615121801"
+
+[[package]]
+name = "equivalent"
+version = "1.0.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "877a4ace8713b0bcf2a4e7eec82529c029f1d0619886d18145fea96c3ffe5c0f"
+
+[[package]]
+name = "hashbrown"
+version = "0.16.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5419bdc4f6a9207fbeba6d11b604d481addf78ecd10c11ad51e76c2f6482748d"
+
+[[package]]
+name = "heck"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "95505c38b4572b2d910cecb0281560f54b440a19336cbbcb27bf6ce6adc6f5a8"
+
+[[package]]
+name = "indexmap"
+version = "2.12.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6717a8d2a5a929a1a2eb43a12812498ed141a0bcfb7e8f7844fbdbe4303bba9f"
+dependencies = [
+ "equivalent",
+ "hashbrown",
+]
+
+[[package]]
+name = "indoc"
+version = "2.0.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "79cf5c93f93228cf8efb3ba362535fb11199ac548a09ce117c9b1adc3030d706"
+dependencies = [
+ "rustversion",
+]
+
+[[package]]
+name = "itoa"
+version = "1.0.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4a5f13b858c8d314ee3e8f639011f7ccefe71f97f96e50151fb991f267928e2c"
+
+[[package]]
+name = "libc"
+version = "0.2.177"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2874a2af47a2325c2001a6e6fad9b16a53b802102b528163885171cf92b15976"
+
+[[package]]
+name = "lock_api"
+version = "0.4.14"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "224399e74b87b5f3557511d98dff8b14089b3dadafcab6bb93eab67d3aace965"
+dependencies = [
+ "scopeguard",
+]
+
+[[package]]
+name = "memchr"
+version = "2.7.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f52b00d39961fc5b2736ea853c9cc86238e165017a493d1d5c8eac6bdc4cc273"
+
+[[package]]
+name = "memoffset"
+version = "0.9.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "488016bfae457b036d996092f6cb448677611ce4449e970ceaf42695203f218a"
+dependencies = [
+ "autocfg",
+]
+
+[[package]]
+name = "once_cell"
+version = "1.21.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "42f5e15c9953c5e4ccceeb2e7382a716482c34515315f7b03532b8b4e8393d2d"
+
+[[package]]
+name = "parking_lot"
+version = "0.12.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "93857453250e3077bd71ff98b6a65ea6621a19bb0f559a85248955ac12c45a1a"
+dependencies = [
+ "lock_api",
+ "parking_lot_core",
+]
+
+[[package]]
+name = "parking_lot_core"
+version = "0.9.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2621685985a2ebf1c516881c026032ac7deafcda1a2c9b7850dc81e3dfcb64c1"
+dependencies = [
+ "cfg-if",
+ "libc",
+ "redox_syscall",
+ "smallvec",
+ "windows-link",
+]
+
+[[package]]
+name = "portable-atomic"
+version = "1.11.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f84267b20a16ea918e43c6a88433c2d54fa145c92a811b5b047ccbe153674483"
+
+[[package]]
+name = "proc-macro2"
+version = "1.0.103"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5ee95bc4ef87b8d5ba32e8b7714ccc834865276eab0aed5c9958d00ec45f49e8"
+dependencies = [
+ "unicode-ident",
+]
+
+[[package]]
+name = "pyo3"
+version = "0.20.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "53bdbb96d49157e65d45cc287af5f32ffadd5f4761438b527b055fb0d4bb8233"
+dependencies = [
+ "cfg-if",
+ "indoc",
+ "libc",
+ "memoffset",
+ "parking_lot",
+ "portable-atomic",
+ "pyo3-build-config",
+ "pyo3-ffi",
+ "pyo3-macros",
+ "unindent",
+]
+
+[[package]]
+name = "pyo3-build-config"
+version = "0.20.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "deaa5745de3f5231ce10517a1f5dd97d53e5a2fd77aa6b5842292085831d48d7"
+dependencies = [
+ "once_cell",
+ "target-lexicon",
+]
+
+[[package]]
+name = "pyo3-ffi"
+version = "0.20.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "62b42531d03e08d4ef1f6e85a2ed422eb678b8cd62b762e53891c05faf0d4afa"
+dependencies = [
+ "libc",
+ "pyo3-build-config",
+]
+
+[[package]]
+name = "pyo3-macros"
+version = "0.20.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7305c720fa01b8055ec95e484a6eca7a83c841267f0dd5280f0c8b8551d2c158"
+dependencies = [
+ "proc-macro2",
+ "pyo3-macros-backend",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "pyo3-macros-backend"
+version = "0.20.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7c7e9b68bb9c3149c5b0cade5d07f953d6d125eb4337723c4ccdb665f1f96185"
+dependencies = [
+ "heck",
+ "proc-macro2",
+ "pyo3-build-config",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "quote"
+version = "1.0.41"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ce25767e7b499d1b604768e7cde645d14cc8584231ea6b295e9c9eb22c02e1d1"
+dependencies = [
+ "proc-macro2",
+]
+
+[[package]]
+name = "redox_syscall"
+version = "0.5.18"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ed2bf2547551a7053d6fdfafda3f938979645c44812fbfcda098faae3f1a362d"
+dependencies = [
+ "bitflags",
+]
+
+[[package]]
+name = "rustversion"
+version = "1.0.22"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b39cdef0fa800fc44525c84ccb54a029961a8215f9619753635a9c0d2538d46d"
+
+[[package]]
+name = "ryu"
+version = "1.0.20"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "28d3b2b1366ec20994f1fd18c3c594f05c5dd4bc44d8bb0c1c632c8d6829481f"
+
+[[package]]
+name = "scopeguard"
+version = "1.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "94143f37725109f92c262ed2cf5e59bce7498c01bcc1502d7b9afe439a4e9f49"
+
+[[package]]
+name = "serde"
+version = "1.0.228"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9a8e94ea7f378bd32cbbd37198a4a91436180c5bb472411e48b5ec2e2124ae9e"
+dependencies = [
+ "serde_core",
+ "serde_derive",
+]
+
+[[package]]
+name = "serde_core"
+version = "1.0.228"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "41d385c7d4ca58e59fc732af25c3983b67ac852c1a25000afe1175de458b67ad"
+dependencies = [
+ "serde_derive",
+]
+
+[[package]]
+name = "serde_derive"
+version = "1.0.228"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d540f220d3187173da220f885ab66608367b6574e925011a9353e4badda91d79"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "serde_json"
+version = "1.0.145"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "402a6f66d8c709116cf22f558eab210f5a50187f702eb4d7e5ef38d9a7f1c79c"
+dependencies = [
+ "itoa",
+ "memchr",
+ "ryu",
+ "serde",
+ "serde_core",
+]
+
+[[package]]
+name = "serde_yaml"
+version = "0.9.34+deprecated"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6a8b1a1a2ebf674015cc02edccce75287f1a0130d394307b36743c2f5d504b47"
+dependencies = [
+ "indexmap",
+ "itoa",
+ "ryu",
+ "serde",
+ "unsafe-libyaml",
+]
+
+[[package]]
+name = "smallvec"
+version = "1.15.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03"
+
+[[package]]
+name = "syn"
+version = "2.0.108"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "da58917d35242480a05c2897064da0a80589a2a0476c9a3f2fdc83b53502e917"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "unicode-ident",
+]
+
+[[package]]
+name = "target-lexicon"
+version = "0.12.16"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "61c41af27dd6d1e27b1b16b489db798443478cef1f06a660c96db617ba5de3b1"
+
+[[package]]
+name = "unicode-ident"
+version = "1.0.20"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "462eeb75aeb73aea900253ce739c8e18a67423fadf006037cd3ff27e82748a06"
+
+[[package]]
+name = "unindent"
+version = "0.2.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7264e107f553ccae879d21fbea1d6724ac785e8c3bfc762137959b5802826ef3"
+
+[[package]]
+name = "unsafe-libyaml"
+version = "0.2.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "673aac59facbab8a9007c7f6108d11f63b603f7cabff99fabf650fea5c32b861"
+
+[[package]]
+name = "windows-link"
+version = "0.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f0805222e57f7521d6a62e36fa9163bc891acd422f971defe97d64e70d0a4fe5"
+
+[[package]]
+name = "yaml-rs"
+version = "0.1.0"
+dependencies = [
+ "pyo3",
+ "serde",
+ "serde_json",
+ "serde_yaml",
+]

parrot/yaml-rs/Cargo.toml

@@ -0,0 +1,19 @@
+[package]
+name = "yaml-rs"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+name = "yaml_rs"
+crate-type = ["cdylib"]
+
+[dependencies]
+pyo3 = { version = "0.20", features = ["extension-module"] }
+serde = { version = "1.0", features = ["derive"] }
+serde_json = "1.0"
+serde_yaml = "0.9"
+
+[profile.release]
+opt-level = 3
+lto = true
+codegen-units = 1

parrot/yaml-rs/pyproject.toml

@@ -0,0 +1,19 @@
+[build-system]
+requires = ["maturin>=1.0,<2.0"]
+build-backend = "maturin"
+
+[project]
+name = "yaml_rs"
+version = "0.1.0"
+description = "Fast YAML serialization using Rust and PyO3"
+requires-python = ">=3.8"
+classifiers = [
+"Programming Language :: Rust",
+"Programming Language :: Python :: Implementation :: CPython",
+]
+
+[tool.maturin]
+python-source = "python"
+module-name = "yaml_rs"
+bindings = "pyo3"
+features = ["pyo3/extension-module"]