pycoze 0.1.253__py3-none-any.whl → 0.1.254__py3-none-any.whl

pycoze/ai/__init__.py

@@ -1,2 +1,14 @@
 from .vram_reserve import reserve_vram, reserve_vram_retry, unreserve_vram
 from .llm import chat, chat_stream, extract, yes_or_no, extract_code, text_to_image_prompt
+
+__all__ = [
+    reserve_vram,
+    reserve_vram_retry,
+    unreserve_vram,
+    chat,
+    chat_stream,
+    extract,
+    yes_or_no,
+    extract_code,
+    text_to_image_prompt
+]

pycoze/ai/llm/__init__.py

@@ -1,2 +1,11 @@
 from .text_to_image_prompt import text_to_image_prompt
 from .chat import chat, chat_stream, extract, yes_or_no, extract_code
+
+__all__ = [
+    text_to_image_prompt,
+    chat,
+    chat_stream,
+    extract,
+    yes_or_no,
+    extract_code
+]

pycoze/ai/vram_reserve.py

@@ -8,7 +8,7 @@ import sys
 from pycoze import utils
 
 # 定义数据库连接和初始化
-params = utils.read_params_file()
+params = utils.params
 if params:
     DATABASE_PATH = params["appPath"] + "/vram_usage.db"
 else:

pycoze/bot/__init__.py

@@ -1 +1,3 @@
 from .bot import chat, get_chat_response
+
+__all__ = [chat, get_chat_response]

pycoze/bot/agent/agent_types/openai_func_call_agent.py

@@ -106,7 +106,6 @@ def create_openai_func_call_agent_executor(
         if last_message.content.strip().endswith("```"):
             last_message.content = last_message.content + "\n\n"  # 避免影响阅读
         tools = get_tools(last_message)
-        print("tools", tools)
         if last_message.tool_calls or tools:
             return 'continue'
         return 'end'

pycoze/bot/agent_chat.py

@@ -1,6 +1,6 @@
 import json
 from langchain_openai import ChatOpenAI
-from .agent import run_agent, Runnable, output, CHAT_DATA, clear_chat_data
+from .agent import run_agent, Runnable,CHAT_DATA, clear_chat_data
 import asyncio
 from pycoze import utils
 from pycoze.reference.bot import ref_bot
@@ -100,7 +100,7 @@ async def agent_chat(bot_setting_file, history):
         assistant_message=prompt,
         tool_compatibility_mode=cfg["toolCompatibilityMode"],
     )
-    params = utils.read_params_file()
+    params = utils.params
     if "interruptFile" in params:
         interrupt_file_path = params["interruptFile"]
         result = await run_with_interrupt_check(agent, history, cfg["toolCompatibilityMode"], interrupt_file_path)

pycoze/bot/bot.py

@@ -1,5 +1,4 @@
 from langchain_core.messages import HumanMessage, AIMessage
-import threading
 import json
 from .agent import INPUT_MESSAGE, output, CHAT_DATA, clear_chat_data
 from .agent_chat import agent_chat

pycoze/ui/__init__.py

@@ -25,3 +25,31 @@ from .ui_def import (
     seed,
     button
 )
+
+__all__ = [
+    get_ui,
+    get_ui_text,
+    label,
+    number,
+    text,
+    textarea,
+    password,
+    tool,
+    color,
+    checkbox,
+    single_select,
+    multi_select,
+    single_file_select,
+    multi_file_select,
+    single_folder_select,
+    multi_folder_select,
+    folder_tree,
+    single_image_select,
+    multi_image_select,
+    single_audio_select,
+    multi_audio_select,
+    single_video_select,
+    multi_video_select,
+    seed,
+    button
+]

pycoze/utils/__init__.py

@@ -1,3 +1,5 @@
 from .arg import read_arg
-from .env import read_params_file, read_json_file
+from .env import read_params_file, params, read_json_file
 from .text_or_file import to_text
+
+__all__ = [read_arg, read_params_file, params, read_json_file, to_text]

pycoze/utils/arg.py

@@ -1,6 +1,4 @@
 import sys
-import json
-import os
 
 
 def read_arg(param, is_path=False, posix=True):

pycoze/utils/env.py

@@ -13,9 +13,13 @@ def read_params_file():
             print(e)
     return params
 
+
+params = read_params_file()
+
+
 def read_json_file(filename):
-    params = read_params_file()
     json_file = os.path.join(params['appPath'], 'JsonStorage', filename)
     with open(json_file, encoding='utf-8') as f:
         return json.load(f)
-        
+
+

pycoze/utils/socket.py

@@ -0,0 +1,72 @@
+import socket
+import json
+import uuid
+
+
+tcp_port = read_arg("tcp_port")
+
+
+
+class TcpSocket:
+    def __init__(self):
+        self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        self.socket.connect(("localhost", int(tcp_port)))
+        self.is_connected = True
+
+    def post(self, subject, message):
+        subject = "python-event:" + subject
+        message = message.copy()
+        message["subject"] = subject
+        message = json.dumps(message)
+        message_bytes = message.encode()
+        message_length = len(message_bytes)
+        message_length_bytes = message_length.to_bytes(4, byteorder='big')
+
+        self.socket.sendall(message_length_bytes + message_bytes)
+
+    def recv(self):
+        # 接收消息长度
+        message_length_bytes = self.socket.recv(4)
+        message_length = int.from_bytes(message_length_bytes, byteorder='big')
+        # 初始化接收到的消息
+        message = b''
+        # 循环接收直到收到完整的消息
+        while len(message) < message_length:
+            # 接收剩余的消息
+            chunk = self.socket.recv(message_length - len(message))
+            if not chunk:  # 如果没有接收到数据，则断开连接
+                raise RuntimeError("Socket connection broken")
+            message += chunk
+        message = json.loads(message.decode())
+        return message
+
+    def post_and_recv_result(self, subject, message):
+        message = message.copy()
+        message["uid"] = str(uuid.uuid4())
+        socket_subscribe(subject + ":" +  message["uid"]) # 先订阅,再发送,防止消息丢失
+        self.post(subject, message)
+        ret_val = self.recv()        
+        result = ret_val["result"] if "result" in ret_val else None # 传输时，可能会丢失result字段，因此需要判断
+        return result
+
+    def close(self):
+        if self.is_connected:
+            self.is_connected = False
+
+    def __del__(self):
+        # 垃圾回收时自动关闭socket
+        if self.is_connected:
+            self.close()
+
+    def __exit__(self, exc_type, exc_value, traceback):
+        # with 语句块结束时自动触发的
+        if self.is_connected:
+            self.close()
+
+
+socket = TcpSocket()
+
+def socket_subscribe(subject: str):
+    subject = "python-event:" + subject
+    socket.post(subject, {"subscribe": subject})
+    return socket

{pycoze-0.1.253.dist-info → pycoze-0.1.254.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.253
+Version: 0.1.254
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.253.dist-info → pycoze-0.1.254.dist-info}/RECORD

@@ -1,36 +1,37 @@
 pycoze/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycoze/ai/__init__.py,sha256=NY6treq6PVA52c62RaP3S3v9Xsmj5xE1YOl169Fr0pg,168
-pycoze/ai/vram_reserve.py,sha256=brgXP42yj3yaZRgW8pfgc4Jg9EivAhcbp5W4igVHcow,4256
-pycoze/ai/llm/__init__.py,sha256=kAXcQ7SefJYysgKeVInlwYZoDk0BPuEnUuixy-quD_A,127
+pycoze/ai/__init__.py,sha256=amfVImcuiRRnGbmOjIqjXKiuH1H3zZPHOrJue2dEB8U,355
+pycoze/ai/vram_reserve.py,sha256=DRxKzqf89fLAts0DzU8e19z9kecIF8OdMkQnJlCKZV0,4244
+pycoze/ai/llm/__init__.py,sha256=5_AnOrzXI2V6ZZsLUWW1v5ATRWmJy53JLN9jfSZQXCg,249
 pycoze/ai/llm/chat.py,sha256=izriC7nCp5qeJRqcUVQBVqTHiH6MJS77ROzGBJufdNI,5133
 pycoze/ai/llm/text_to_image_prompt.py,sha256=0bx2C_YRvjAo7iphHGp1-pmGKsKqwur7dM0t3SiA8kA,3398
 pycoze/ai/llm/think.py,sha256=sUgTBdGzcZtL3r-Wx8M3lDuVUmDVz8g3qC0VU8uiKAI,5143
-pycoze/bot/__init__.py,sha256=6HHMxDQVOyZM9dtSjQm9tjGnhj4h7CixD0JOvEwTi48,41
-pycoze/bot/agent_chat.py,sha256=ZQeuNHwFAKB8XO4_R1lytf_kE43hvtI7AzOK4NPysWE,4199
-pycoze/bot/bot.py,sha256=fmcgnWcYTFeOxDuAwqWhFhOJzv4mAyJGLqbod-nkhJE,862
+pycoze/bot/__init__.py,sha256=JxnRoCCqx_LFyVb3pLu0qYCsH8ZLuAaMXAtvVUuCHuE,78
+pycoze/bot/agent_chat.py,sha256=ARXXsIVCTpmBojz2C4BR69nB0QhM6gkEngL3iq34JPo,4178
+pycoze/bot/bot.py,sha256=_qmUTZ09FmRLifHrW5stDZWGVK6yuMEMBC3fmeYJnqk,844
 pycoze/bot/agent/__init__.py,sha256=3wE8_FFQS8j2BY-g9Cr-onV0POEvDRZaw_NCzpqrNus,265
 pycoze/bot/agent/agent.py,sha256=chUgNZh6v6375L_Y2dBEAaLJyfmw4SygYjVVrDN8VIk,3548
 pycoze/bot/agent/assistant.py,sha256=5LIgPIVVzx6uIOWT5S_XDDyPPjPHRBBNpIU3GiOkVHc,1186
 pycoze/bot/agent/chat.py,sha256=mubOCAHvA6VtyE6N40elI6KrP6A69uB_G6ihE3G_Vi4,860
 pycoze/bot/agent/agent_types/__init__.py,sha256=zmU2Kmrv5mCdfg-QlPn2H6pWxbGeq8s7YTqLhpzJC6k,179
 pycoze/bot/agent/agent_types/const.py,sha256=BfUKPrhAHREoMLHuFNG2bCIEkC1-f7K0LEqNg4RwiRE,70
-pycoze/bot/agent/agent_types/openai_func_call_agent.py,sha256=OudI1xFDrBa-69JX-FrWmIauNG2y_Eco3HY1WEMf-fI,6731
+pycoze/bot/agent/agent_types/openai_func_call_agent.py,sha256=6aKnUQDINyUaCW24oa9Qjkm5w3ctZ6lxAgcE4m9YHwE,6701
 pycoze/reference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/reference/bot.py,sha256=BDflTV3zYoZqWnJpD5lMM_1vU_5b20M3XiFt1p-RHWM,2427
 pycoze/reference/lib.py,sha256=0xQJTLTHedGzQBsjuTFNBVqYc4-8Yl65gGCrAhWyOX8,2155
 pycoze/reference/tool.py,sha256=9OxHawxjkh6S8eAOlGNrbfhkk5oRlRc7LBpnV6RmhYI,1722
 pycoze/reference/workflow.py,sha256=xIxv4EAeIR663bRfXkLhgBC0OhfMs5RdfvdORI80CoQ,2068
-pycoze/ui/__init__.py,sha256=RTEUVXdMo7AYC1wsyS1yl5sLi5vf4bNFv4iggghGEgg,469
+pycoze/ui/__init__.py,sha256=uaXet23wUk64TcZjpBX8qOx4aUhwA_ucrmcxy7Q4Qr4,929
 pycoze/ui/base.py,sha256=sbBZGMUtlosWHQJpxMULa1bGByeSlcldtE9QXNyiJmM,1093
 pycoze/ui/color.py,sha256=cT9Ib8uNzkOKxyW0IwVj46o4LwdB1xgNCj1_Rou9d_4,854
 pycoze/ui/typ.py,sha256=NpT0FrbHvByOszBZMFtroRp7I7pN-38tYz_zPOPejF4,1723
 pycoze/ui/ui_def.py,sha256=lGWZGpzRoegP34D562PvK0EJHrmVZrlHW1JjsIG9A9Q,4521
-pycoze/utils/__init__.py,sha256=Gi5EnrWZGMD2JRejgV4c_VLCXyvA2wwBFI_niDF5MUE,110
-pycoze/utils/arg.py,sha256=GtfGbMTMdaK75Fwh6MpUe1pCA5X6Ep4LFG7a72YrzjI,525
-pycoze/utils/env.py,sha256=W04lhvTHhAAC6EldP6kk2xrctqtu8K6kl1vDLZDNeh8,561
+pycoze/utils/__init__.py,sha256=6bQ3erQHKVOUk4ylyC-6PCASbAV0pYFQfC_f1KqkbUw,191
+pycoze/utils/arg.py,sha256=jop1tBfe5hYkHW1NSpCeaZBEznkgguBscj_7M2dWfrs,503
+pycoze/utils/env.py,sha256=5pWlXfM1F5ZU9hhv1rHlDEanjEW5wf0nbyez9bNRqqA,559
+pycoze/utils/socket.py,sha256=CpyyC-zgibi_GvTeHP14VlqIfC7QuRE6BNbQH3Hb45c,2370
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.253.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.253.dist-info/METADATA,sha256=9mtMC2PMjn5O5S6AUPKXRs7WOeGOvCGebM67Wu0YNyM,699
-pycoze-0.1.253.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-pycoze-0.1.253.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.253.dist-info/RECORD,,
+pycoze-0.1.254.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.254.dist-info/METADATA,sha256=hD65Lc7e26JX0Th3OYDX8OFcJPZcQCR90fy9doVnkno,699
+pycoze-0.1.254.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+pycoze-0.1.254.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.254.dist-info/RECORD,,