unique_toolkit 0.0.1__py3-none-any.whl → 0.5.0__py3-none-any.whl

unique_toolkit/llm/service.py

@@ -1,163 +0,0 @@
-import unique_sdk
-
-from unique_toolkit.chat.state import ChatState
-from unique_toolkit.performance.async_wrapper import to_async
-
-from .models import LLMModelName
-
-
-class LLMService:
-    def __init__(self, state: ChatState):
-        self.state = state
-
-    def stream_complete(
-        self,
-        *,
-        messages: list,
-        model_name: LLMModelName,
-        search_contexts: list = [],
-        debug_info: dict = {},
-        timeout: int = 100_000,
-        temperature: float = 0.25,
-    ):
-        """
-        Streams a completion in the chat session synchronously.
-
-        Args:
-            messages (list): The messages to stream.
-            search_contexts (list): The search context.
-            model_name (LLMModelName): The language model to use for the completion.
-            debug_info (dict): The debug information.
-            timeout (int, optional): The timeout value in milliseconds. Defaults to 100_000.
-            temperature (float, optional): The temperature value for the completion. Defaults to 0.25.
-
-        Returns:
-            A generator yielding streamed completion chunks.
-        """
-        return self._trigger_stream_complete(
-            messages,
-            search_contexts,
-            model_name.name,
-            debug_info,
-            timeout,
-            temperature,
-        )
-
-    @to_async
-    def async_stream_complete(
-        self,
-        *,
-        messages: list,
-        model_name: LLMModelName,
-        search_contexts: list = [],
-        debug_info: dict = {},
-        timeout: int = 100_000,
-        temperature: float = 0.25,
-    ):
-        """
-        Streams a completion in the chat session asynchronously.
-
-        Args:
-            messages (list[dict[str, str]]): The messages to stream.
-            search_contexts (list): The search context.
-            debug_info (dict): The debug information.
-            model_name (LLMModelName): The language model to use for the completion.
-            timeout (int, optional): The timeout value in milliseconds. Defaults to 100_000.
-            temperature (float, optional): The temperature value for the completion. Defaults to 0.25.
-
-        Returns:
-            A generator yielding streamed completion chunks.
-        """
-        return self._trigger_stream_complete(
-            messages, search_contexts, model_name, debug_info, timeout, temperature
-        )
-
-    def complete(
-        self,
-        *,
-        messages: list,
-        model_name: LLMModelName,
-        temperature: float = 0,
-        timeout: int = 240000,
-    ) -> str:
-        """
-        Calls the completion endpoint synchronously without streaming the response.
-
-        Args:
-            messages (list[dict[str, str]]): The messages to complete.
-            model_name (LLMModelName): The model name.
-            temperature (float, optional): The temperature value. Defaults to 0.
-            timeout (int, optional): The timeout value in milliseconds. Defaults to 240000.
-
-        Returns:
-            str: The completed message content.
-        """
-        return self._trigger_complete(messages, model_name, temperature, timeout)
-
-    @to_async
-    def async_complete(
-        self,
-        *,
-        messages: list,
-        model_name: LLMModelName,
-        temperature: float = 0,
-        timeout: int = 240000,
-    ) -> str:
-        """
-        Calls the completion endpoint asynchronously without streaming the response.
-
-        Args:
-            messages (list[dict[str, str]]): The messages to complete.
-            model_name (LLMModelName): The model name.
-            temperature (float, optional): The temperature value. Defaults to 0.
-            timeout (int, optional): The timeout value in milliseconds. Defaults to 240000.
-
-        Returns:
-            str: The completed message content.
-        """
-        return self._trigger_complete(
-            messages,
-            model_name.name,
-            temperature,
-            timeout,
-        )
-
-    def _trigger_stream_complete(
-        self,
-        messages: list,
-        search_contexts: list,
-        model_name: str,
-        debug_info: dict,
-        timeout: int,
-        temperature: float,
-    ):
-        return unique_sdk.Integrated.chat_stream_completion(
-            user_id=self.state.user_id,
-            company_id=self.state.company_id,
-            assistantMessageId=self.state.assistant_message_id,
-            userMessageId=self.state.user_message_id,
-            messages=messages,
-            chatId=self.state.chat_id,
-            searchContext=search_contexts,
-            debugInfo=debug_info,
-            model=model_name,  # type: ignore
-            timeout=timeout,
-            temperature=temperature,
-            assistantId=self.state.assistant_id,
-        )
-
-    def _trigger_complete(
-        self,
-        messages: list,
-        model_name: str,
-        temperature: float,
-        timeout: int,
-    ) -> str:
-        result = unique_sdk.ChatCompletion.create(
-            company_id=self.state.company_id,
-            model=model_name,  # type: ignore
-            messages=messages,
-            timeout=timeout,
-            temperature=temperature,
-        )
-        return result.choices[-1]["message"]["content"]

unique_toolkit/performance/__init__.py

@@ -1 +0,0 @@
-from .async_wrapper import to_async  # noqa: F401

unique_toolkit/performance/async_wrapper.py

@@ -1,13 +0,0 @@
-import asyncio
-from functools import wraps
-from typing import Any, Callable, Coroutine, TypeVar
-
-T = TypeVar("T")
-
-
-def to_async(func: Callable[..., T]) -> Callable[..., Coroutine[Any, Any, T]]:
-    @wraps(func)
-    async def wrapper(*args, **kwargs) -> T:
-        return await asyncio.to_thread(func, *args, **kwargs)
-
-    return wrapper

unique_toolkit/sdk/init.py

@@ -1,19 +0,0 @@
-import os
-
-import unique_sdk
-
-
-def init_sdk():
-    """Initializes the SDK."""
-    unique_sdk.api_base = os.environ.get("API_BASE")  # Required
-
-    # Optional if being run locally
-    unique_sdk.api_key = os.environ.get(
-        "API_KEY", "dummy"
-    )  # Optional if being run locally
-    unique_sdk.app_id = os.environ.get(
-        "APP_ID", "dummy"
-    )  # Optional if being run locally
-    endpoint_secret = os.environ.get("ENDPOINT_SECRET", None)
-
-    return endpoint_secret

unique_toolkit/search/service.py

@@ -1,118 +0,0 @@
-from typing import Literal
-
-import unique_sdk
-from unique_sdk import Content
-
-from unique_toolkit.chat.state import ChatState
-from unique_toolkit.performance.async_wrapper import to_async
-
-
-class SearchService:
-    def __init__(self, state: ChatState):
-        self.state = state
-
-    def search(
-        self,
-        *,
-        search_string: str,
-        search_type: Literal["VECTOR", "COMBINED"],
-        limit: int,
-        scope_ids: list[str] = [],
-    ):
-        """
-        Performs a synchronous search in the knowledge base.
-
-        Args:
-            search_string (str): The search string.
-            search_type (Literal["VECTOR", "COMBINED"]): The type of search to perform.
-            limit (int): The maximum number of results to return.
-
-        Returns:
-            The search results.
-        """
-        return self._trigger_search(search_string, search_type, limit, scope_ids)
-
-    @to_async
-    def async_search(
-        self,
-        *,
-        search_string: str,
-        search_type: Literal["VECTOR", "COMBINED"],
-        limit: int,
-        scope_ids: list[str] = [],
-    ):
-        """
-        Performs an asynchronous search in the knowledge base.
-
-        Args:
-            search_string (str): The search string.
-            search_type (Literal["VECTOR", "COMBINED"]): The type of search to perform.
-            limit (int): The maximum number of results to return.
-
-        Returns:
-            The search results.
-        """
-        return self._trigger_search(search_string, search_type, limit, scope_ids)
-
-    def _trigger_search(
-        self,
-        search_string: str,
-        search_type: Literal["VECTOR", "COMBINED"],
-        limit: int,
-        scope_ids: list[str] = [],
-    ):
-        scope_ids = scope_ids or self.state.scope_ids or []
-        return unique_sdk.Search.create(
-            user_id=self.state.user_id,
-            company_id=self.state.company_id,
-            chatId=self.state.chat_id,
-            searchString=search_string,
-            searchType=search_type,
-            scopeIds=scope_ids,
-            limit=limit,
-            chatOnly=self.state.chat_only,
-        )
-
-    def search_content(
-        self,
-        *,
-        where: dict,
-    ) -> list[Content]:
-        """
-        Performs a search in the knowledge base by filter.
-
-        Args:
-            where (dict): The search criteria.
-
-        Returns:
-            The search results.
-        """
-        return self._trigger_search_content(where)
-
-    @to_async
-    def async_search_content(
-        self,
-        *,
-        where: dict,
-    ) -> list[Content]:
-        """
-        Performs an asynchronous search in the knowledge base by filter.
-
-        Args:
-            where (dict): The search criteria.
-
-        Returns:
-            The search results.
-        """
-        return self._trigger_search_content(where)
-
-    def _trigger_search_content(
-        self,
-        where: dict,
-    ) -> list[Content]:
-        return unique_sdk.Content.search(
-            user_id=self.state.user_id,
-            company_id=self.state.company_id,
-            chatId=self.state.chat_id,
-            where=where,  # type: ignore
-        )

unique_toolkit/security/verify.py

@@ -1,29 +0,0 @@
-from http import HTTPStatus
-
-import unique_sdk
-
-from unique_toolkit.event.schema import Event
-
-
-def verify_signature_and_construct_event(
-    headers, payload: bytes, endpoint_secret: str, logger
-) -> Event:
-    sig_header = headers.get("X-Unique-Signature")
-    timestamp = headers.get("X-Unique-Created-At")
-
-    if not sig_header or not timestamp:
-        logger.error("⚠️  Webhook signature or timestamp headers missing.")
-        return False, HTTPStatus.BAD_REQUEST
-
-    try:
-        event = unique_sdk.Webhook.construct_event(
-            payload,
-            sig_header,
-            timestamp,
-            endpoint_secret,
-        )
-        logger.info("✅  Webhook signature verification successful.")
-        return Event(**event)
-    except unique_sdk.SignatureVerificationError as e:
-        logger.error("⚠️  Webhook signature verification failed. " + str(e))
-        return False, HTTPStatus.BAD_REQUEST

unique_toolkit-0.0.1.dist-info/METADATA

@@ -1,33 +0,0 @@
-Metadata-Version: 2.1
-Name: unique_toolkit
-Version: 0.0.1
-Summary: 
-License: MIT
-Author: Unique Data Science
-Author-email: datascience@unique.ch
-Requires-Python: >=3.11,<4.0
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: pydantic (>=2.8.2,<3.0.0)
-Requires-Dist: pyhumps (>=3.8.0,<4.0.0)
-Requires-Dist: typing-extensions (>=4.9.0,<5.0.0)
-Requires-Dist: unique-sdk (>=0.8.10,<0.9.0)
-Description-Content-Type: text/markdown
-
-# Unique Tool Kit
-
-This package provides highlevel abstarctions and methods on top of `unique_sdk` to ease application development for the Unique Platform.
-
-
-# Development instructions
-
-1. Install poetry on your system (through `brew` or `pipx`).
-
-2. Install `pyenv` and install python 3.11. `pyenv` is recommended as otherwise poetry uses the python version used to install itself and not the user preferred python version.
-
-3. If you then run `python --version` in your terminal, you should be able to see python version as specified in `.python-version`.
-
-4. Then finally run `poetry install` to install the package and all dependencies.
-

unique_toolkit-0.0.1.dist-info/RECORD

@@ -1,23 +0,0 @@
-unique_toolkit/__init__.py,sha256=vObIHz3Vte5K6tlrytiNQQ1uqhibwnhSzB5WIt9cyIs,197
-unique_toolkit/chat/__init__.py,sha256=TocIEJrMO5_xJB4ehFhslqoxOsV2oiLlglUdIWVLa-s,166
-unique_toolkit/chat/messages.py,sha256=IuYPz8PoK6HyFsYigtfYkfz2wB8zhDJyjryWYC2D7k4,416
-unique_toolkit/chat/service.py,sha256=UigC5-1wdQYoWLfzAU7h8hl_SB05YSCOTrxhU7dgMG4,5511
-unique_toolkit/chat/state.py,sha256=7N9vsMl4mu1mbtFqg61T6-42UgnkXmLwhVfIwVnm6tc,1327
-unique_toolkit/event/__init__.py,sha256=fqfQeanYjxLFGsrX97Ti3HpzLRCE3kDUlMBVXEvzr0s,80
-unique_toolkit/event/constants.py,sha256=hwdCgSKBAZQYZSl1nFkP1yWiw6GBI7KbcNJrdxFxVm8,66
-unique_toolkit/event/schema.py,sha256=sT0gtJN2LT0ormP7HfVot11Oub1YlzfBCKFj3G0Or4w,879
-unique_toolkit/llm/__init__.py,sha256=3hT7PhdHnEQaqILJ6gGSe0cpkWEDMgZ_PHw1Q01W0e4,127
-unique_toolkit/llm/json_parser.py,sha256=sHTe6_VuqqwrC1xCg9ud3VOukEzVSoj0Uce_sv7GEYE,528
-unique_toolkit/llm/models.py,sha256=Gt691uSL74e3bdIm4UhAcp3Nw-jafnqJSgKdzhLJKww,1254
-unique_toolkit/llm/service.py,sha256=UXXMo7MKcj7fbDKlv0gk2J00ClL1F7n0_6uiwOLSXoY,5226
-unique_toolkit/observability/log_config.py,sha256=mfIu4cX95AiqOCdh2iW3iLoXtwrIrkPCXRtG_zYMhSs,679
-unique_toolkit/performance/__init__.py,sha256=TYPrsYdxXhhom12ub0dzvR6dHtFdjnF-P506LYitj8Y,50
-unique_toolkit/performance/async_executor.py,sha256=8N8GQn-MQN-8fX8DAbbYt7s2_91G0nchFQ25i9lIxsE,5555
-unique_toolkit/performance/async_wrapper.py,sha256=MB3cgqcl-Y7Ltc9a_l8bkhfOGrds1lUtici17b9dGuQ,339
-unique_toolkit/sdk/init.py,sha256=UQLbHxUjN_dttkiUznfJclYPHNVwANdf8w6H4SKs9Y4,482
-unique_toolkit/search/service.py,sha256=hGzfFAdBt9Y5pwPHXTEmzid4FnEsXu9DiSzpw5gPv1Y,3167
-unique_toolkit/security/verify.py,sha256=RxBw-gnWUsvqZT6rAt5Eq0vaR0rDY8AqUOTBZmQUItE,927
-unique_toolkit-0.0.1.dist-info/LICENSE,sha256=nojs8Z8soUMb9wg1sAUZdJxyhzRGrUIXFeq0_I-zejo,1065
-unique_toolkit-0.0.1.dist-info/METADATA,sha256=klnJs0EqSKVW0wz384xzI0PU-gaAbTIkb0EVHpLukhQ,1233
-unique_toolkit-0.0.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.0.1.dist-info/RECORD,,