reykit 1.0.1__py3-none-any.whl → 1.1.1__py3-none-any.whl

reyweb/rbaidu/rbaidu_chat.py

@@ -1,299 +0,0 @@
-# !/usr/bin/env python
-# -*- coding: utf-8 -*-
-
-"""
-@Time    : 2024-01-11 22:00:14
-@Author  : Rey
-@Contact : reyxbo@163.com
-@Explain : Baidu API chat methods.
-"""
-
-
-from typing import TypedDict, Literal, Union, Optional
-from datetime import datetime, timedelta
-from reykit.rexception import catch_exc
-from reykit.rrandom import randi
-from reykit.rtime import now
-
-from .rbaidu_base import RAPIBaidu
-
-
-__all__ = (
-    'RAPIBaiduChat',
-)
-
-
-ChatRecord = TypedDict('ChatRecord', {'time': datetime, 'send': str, 'receive': str})
-HistoryMessage = TypedDict('HistoryMessage', {'role': str, 'content': str})
-
-
-class RAPIBaiduChat(RAPIBaidu):
-    """
-    Rey's `Baidu API chat` type.
-    """
-
-    # Character.
-    characters = (
-        '善良', '淳厚', '淳朴', '豁达', '开朗', '体贴', '活跃', '慈祥', '仁慈', '温和',
-        '温存', '和蔼', '和气', '直爽', '耿直', '憨直', '敦厚', '正直', '爽直', '率直',
-        '刚直', '正派', '刚正', '纯正', '廉政', '清廉', '自信', '信心', '新年', '相信',
-        '老实', '谦恭', '谦虚', '谦逊', '自谦', '谦和', '坚强', '顽强', '建议', '刚毅',
-        '刚强', '倔强', '强悍', '刚毅', '减震', '坚定', '坚韧', '坚决', '坚忍', '勇敢',
-        '勇猛', '勤劳', '勤恳', '勤奋', '勤勉', '勤快', '勤俭', '辛勤', '刻苦', '节约',
-        '狂妄', '骄横', '骄纵', '窘态', '窘迫', '困窘', '难堪', '害羞', '羞涩', '赧然',
-        '无语', '羞赧'
-    )
-
-
-    def __init__(
-        self,
-        key: str,
-        secret: str,
-        character: Optional[str] = None
-    ) -> None:
-        """
-        Build `Baidu API chat` attributes.
-
-        Parameters
-        ----------
-        key : API key.
-        secret : API secret.
-        Character : Character of language model.
-        """
-
-        # Set attribute.
-        super().__init__(key, secret)
-        self.chat_records: dict[str, ChatRecord] = {}
-        self.character=character
-
-
-    def chat(
-        self,
-        text: str,
-        character: Optional[Union[str, Literal[False]]] = None,
-        history_key: Optional[str] = None,
-        history_recent_seconds: float = 1800,
-        history_max_word: int = 400
-    ) -> bytes:
-        """
-        Chat with language model.
-
-        Parameters
-        ----------
-        text : Text.
-        Character : Character of language model.
-            - `None`, Use `self.character`: attribute.
-            - `str`: Use this value.
-            - `Literal[False]`: Do not set.
-        Character : Character of language model.
-        history_key : Chat history records key.
-        history_recent_seconds : Limit recent seconds of chat history.
-        history_max_word : Limit maximum word of chat history.
-
-        Returns
-        -------
-        Reply text.
-        """
-
-        # Get parameter.
-        url = 'https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/chat/completions_pro'
-        params = {'access_token': self.token}
-        headers = {'Content-Type': 'application/json'}
-        if history_key is None:
-            messages = []
-        else:
-            messages = self.history_messages(
-                history_key,
-                history_recent_seconds,
-                history_max_word
-            )
-        message = {'role': 'user', 'content': text}
-        messages.append(message)
-        json = {'messages': messages}
-        match character:
-            case None:
-                character = self.character
-            case False:
-                character = None
-        if character is not None:
-            json['system'] = character
-
-        # Request.
-        try:
-            response = self.request(
-                url,
-                params=params,
-                json=json,
-                headers=headers
-            )
-
-        ## Parameter 'system' error.
-        except:
-            *_, exc_instance, _ = catch_exc()
-            error_code = exc_instance.args[1]['error_code']
-            if error_code == 336104:
-                result = self.chat(
-                    text,
-                    False,
-                    history_key,
-                    history_recent_seconds,
-                    history_max_word
-                )
-                return result
-            else:
-                raise exc_instance
-
-        # Extract.
-        response_json: dict = response.json()
-        result: str = response_json['result']
-
-        # Record.
-        self.record_call(
-            messages=messages,
-            character=character
-        )
-        if history_key is not None:
-            self.record_chat(
-                text,
-                result,
-                history_key
-            )
-
-        return result
-
-
-    def record_chat(
-        self,
-        send: str,
-        receive: str,
-        key: str
-    ) -> None:
-        """
-        Record chat.
-
-        Parameters
-        ----------
-        send : Send text.
-        receive : Receive text.
-        key : Chat history records key.
-        """
-
-        # Generate.
-        record = {
-            'time': now(),
-            'send': send,
-            'receive': receive
-        }
-
-        # Record.
-        reocrds = self.chat_records.get(key)
-        if reocrds is None:
-            self.chat_records[key] = [record]
-        else:
-            reocrds.append(record)
-
-
-    def history_messages(
-        self,
-        key: str,
-        recent_seconds: float,
-        max_word: int
-    ) -> list[HistoryMessage]:
-        """
-        Return history messages.
-
-        Parameters
-        ----------
-        key : Chat history records key.
-        recent_seconds : Limit recent seconds of chat history.
-        max_word : Limit maximum word of chat history.
-
-        Returns
-        -------
-        History messages.
-        """
-
-        # Get parameter.
-        records = self.chat_records.get(key, [])
-        now_time = now()
-
-        # Generate.
-        messages = []
-        word_count = 0
-        for record in records:
-
-            ## Limit time.
-            interval_time: timedelta = now_time - record['time']
-            interval_seconds = interval_time.total_seconds()
-            if interval_seconds > recent_seconds:
-                break
-
-            ## Limit word.
-            word_len = len(record['send']) + len(record['receive'])
-            character_len = len(self.character)
-            word_count += word_len
-            if word_count + character_len > max_word:
-                break
-
-            ## Append.
-            message = [
-                {'role': 'user', 'content': record['send']},
-                {'role': 'assistant', 'content': record['receive']}
-            ]
-            messages.extend(message)
-
-        return messages
-
-
-    def interval_chat(
-        self,
-        key: str
-    ) -> float:
-        """
-        Return the interval seconds from last chat.
-        When no record, then return the interval seconds from start.
-
-        Parameters
-        ----------
-        key : Chat history records key.
-
-        Returns
-        -------
-        Interval seconds.
-        """
-
-        # Get parameter.
-        records = self.chat_records.get(key)
-        if records is None:
-            last_time = self.start_time
-        else:
-            last_time: datetime = records[-1]['time']
-        if self.call_records == []:
-            last_time = self.start_time
-        else:
-            last_time: datetime = self.call_records[-1]['time']
-
-        # Count.
-        now_time = now()
-        interval_time = now_time - last_time
-        interval_seconds = interval_time.total_seconds()
-
-        return interval_seconds
-
-
-    def modify(
-        self,
-        text: str
-    ) -> str:
-        """
-        Modify text.
-        """
-
-        # Get parameter.
-        character = randi(self.characters)
-
-        # Modify.
-        text = '用%s的语气，润色以下这句话\n%s' % (character, text)
-        text_modify = self.chat(text)
-
-        return text_modify

reyweb/rbaidu/rbaidu_image.py

@@ -1,183 +0,0 @@
-# !/usr/bin/env python
-# -*- coding: utf-8 -*-
-
-"""
-@Time    : 2024-01-11 22:00:14
-@Author  : Rey
-@Contact : reyxbo@163.com
-@Explain : Baidu API image methods.
-"""
-
-
-from typing import Optional
-from reykit.ros import RFile
-from reykit.rtime import wait
-
-from .rbaidu_base import RAPIBaidu
-
-
-__all__ = (
-    'RAPIBaiduImage',
-)
-
-
-class RAPIBaiduImage(RAPIBaidu):
-    """
-    Rey's `Baidu API image` type.
-    """
-
-
-    def _to_url_create_task(
-        self,
-        text: str
-    ) -> str:
-        """
-        Create task of generate image URL from text.
-
-        Parameters
-        ----------
-        text : Text, length cannot exceed 60.
-
-        Returns
-        -------
-        Task ID.
-        """
-
-        # Get parameter.
-        url = 'https://aip.baidubce.com/rpc/2.0/ernievilg/v1/txt2imgv2'
-        params = {'access_token': self.token}
-        headers = {
-            'Accept': 'application/json',
-            'Content-Type': 'application/json'
-        }
-        json = {
-            'prompt': text,
-            'width': 1024,
-            'height': 1024
-        }
-
-        # Request.
-        response = self.request(
-            url,
-            params=params,
-            json=json,
-            headers=headers
-        )
-
-        # Record.
-        self.record_call(text=text)
-
-        # Extract.
-        response_json: dict = response.json()
-        task_id: str = response_json['data']['task_id']
-
-        return task_id
-
-
-    def _to_url_query_task(
-        self,
-        task_id: str
-    ) -> dict:
-        """
-        Query task of generate image URL from text.
-
-        Parameters
-        ----------
-        task_id : Task ID.
-
-        Returns
-        -------
-        Task information.
-        """
-
-        # Get parameter.
-        url = 'https://aip.baidubce.com/rpc/2.0/ernievilg/v1/getImgv2'
-        params = {'access_token': self.token}
-        headers = {
-            'Accept': 'application/json',
-            'Content-Type': 'application/json'
-        }
-        json = {'task_id': task_id}
-
-        # Request.
-        response = self.request(
-            url,
-            params=params,
-            json=json,
-            headers=headers
-        )
-
-        # Extract.
-        response_json: dict = response.json()
-        task_info: dict = response_json['data']
-
-        return task_info
-
-
-    def to_url(
-        self,
-        text: str,
-        path: Optional[str] = None
-    ) -> str:
-        """
-        Generate image URL from text.
-
-        Parameters
-        ----------
-        text : Text, length cannot exceed 60.
-        path : File save path.
-            - `None`: Not save.
-
-        Returns
-        -------
-        Image URL.
-        """
-
-        # Create.
-        task_id = self._to_url_create_task(text)
-
-        # Wait.
-        store = {}
-
-
-        ## Define.
-        def is_task_success() -> bool:
-            """
-            Whether if is task successed.
-
-            Returns
-            -------
-            Judge result.
-            """
-
-            # Query.
-            task_info = self._to_url_query_task(task_id)
-
-            # Judge.
-            match task_info['task_status']:
-                case 'RUNNING':
-                    return False
-                case 'SUCCESS':
-                    store['url'] = task_info['sub_task_result_list'][0]['final_image_list'][0]['img_url']
-                    return True
-                case _:
-                    raise AssertionError('Baidu API text to image task failed')
-
-
-        ## Start.
-        wait(
-            is_task_success,
-            _interval=0.5,
-            _timeout=600
-        )
-
-        ## Extract.
-        url = store['url']
-
-        # Save.
-        if path is not None:
-            response = self.request(url)
-            rfile = RFile(path)
-            rfile.write(response.content)
-
-        return url