pycoze 0.1.223__py3-none-any.whl → 0.1.225__py3-none-any.whl

pycoze/ai/__init__.py

@@ -1,2 +1,2 @@
 from .vram_reserve import reserve_vram, reserve_vram_retry, unreserve_vram
-from .llm import think, think_stream, extract, yes_or_no, extract_code
+from .llm import chat, chat_stream, extract, yes_or_no, extract_code, text_to_image_prompt

pycoze/ai/llm/__init__.py

@@ -1,2 +1,2 @@
 from .text_to_image_prompt import text_to_image_prompt
-from .think import think, think_stream, extract, yes_or_no, extract_code
+from .chat import chat, chat_stream, extract, yes_or_no, extract_code

pycoze/ai/llm/{think.py → chat.py}

@@ -13,7 +13,7 @@ def never_retry_on_rate_limit_error(exception):
     return not isinstance(exception, openai.RateLimitError)
 
 @retry(retry_on_exception=never_retry_on_rate_limit_error, wait_exponential_multiplier=500, stop_max_attempt_number=5)
-def think(user_text, history, temperature=0.2, stop=None, **kwargs):
+def chat(user_text, history, temperature=0.2, stop=None, **kwargs):
     user_msg = {"role": "user", "content": user_text}
     cfg = utils.read_json_file("llm.json")
 
@@ -33,7 +33,7 @@ def think(user_text, history, temperature=0.2, stop=None, **kwargs):
 
 
 @retry(retry_on_exception=never_retry_on_rate_limit_error, wait_exponential_multiplier=500, stop_max_attempt_number=5)
-def think_stream(user_text, history, temperature=0.2, stop=None, **kwargs):
+def chat_stream(user_text, history, temperature=0.2, stop=None, **kwargs):
     user_msg = {"role": "user", "content": user_text}
     cfg = utils.read_json_file("llm.json")
 
@@ -73,7 +73,7 @@ def extract(response_data, text: str, temperature=0, **kwargs):
         json_text += f'    "{res[0]}": {res[1]}{comma}\n'
 
     # Combine the provided text with the formatted JSON schema
-    think_text = f"""
+    chat_text = f"""
 The output should be a markdown code snippet formatted in the following schema, including the leading and trailing "```json" and "```" tags:
 ```json
 {{
@@ -85,7 +85,7 @@ Request:
 {text}
 """
     # text放后面,当翻译等情况时,不会把"The output should"之类翻译了,导致错误
-    markdown = think(think_text, [], temperature=temperature, **kwargs)
+    markdown = chat(chat_text, [], temperature=temperature, **kwargs)
     pattern = r'```json(.*?)```'
     matches = re.findall(pattern, markdown, re.DOTALL)
     if matches:
@@ -109,10 +109,10 @@ def yes_or_no(question, temperature=0, **kwargs):
 @retry(retry_on_exception=never_retry_on_rate_limit_error, wait_exponential_multiplier=500, stop_max_attempt_number=3)
 def extract_code(text: str, temperature=0, language="python", markdown_word='python', **kwargs):
     """print(extract_code("sum 1~100"))"""
-    think_text = text + f"""
+    chat_text = text + f"""
 The output should be a complete and usable {language} code snippet, including the leading and trailing "```{markdown_word}" and "```":
 """
-    markdown = think(think_text, [], temperature=temperature, **kwargs)
+    markdown = chat(chat_text, [], temperature=temperature, **kwargs)
     # 使用正则表达式匹配围绕在```{markdown_word} 和 ```之间的文本
     pattern = rf'```{markdown_word}(.*?)```'
     matches = re.findall(pattern, markdown, re.DOTALL)
@@ -124,8 +124,8 @@ The output should be a complete and usable {language} code snippet, including th
 
 
 if __name__ == "__main__":
-    print(think("你好", []))
-    for chunk in think_stream("你好", []):
+    print(chat("你好", []))
+    for chunk in chat_stream("你好", []):
         print(chunk)
     print(extract({"name": "lowercase"}, "hello XiaoMing"))
     print(extract_code("sum 1~100"))

pycoze/ai/llm/text_to_image_prompt.py

@@ -1,5 +1,5 @@
 import re
-from .think import yes_or_no, extract
+from .chat import yes_or_no, extract
 
 
 def contains_chinese(text):

{pycoze-0.1.223.dist-info → pycoze-0.1.225.dist-info}/METADATA

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

{pycoze-0.1.223.dist-info → pycoze-0.1.225.dist-info}/RECORD

@@ -1,9 +1,9 @@
 pycoze/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pycoze/ai/__init__.py,sha256=NhIqfUR_01rlZ8hsxi4aUwQ78tiMGmcUluiB5E4Xy6M,145
+pycoze/ai/__init__.py,sha256=uhG4Z3HSTp_o6H4ouNy8gW1hzcU0B8zDB0nJgbmnHpk,165
 pycoze/ai/vram_reserve.py,sha256=brgXP42yj3yaZRgW8pfgc4Jg9EivAhcbp5W4igVHcow,4256
-pycoze/ai/llm/__init__.py,sha256=Pc7eYhrzkZqNXNPKkbd1tyoizdlGm9a60J9NGr1v3RQ,128
-pycoze/ai/llm/text_to_image_prompt.py,sha256=WnC7AoPxvEGrSftTdUytMwEr7-ETZjLy4ztUi9s7Ouo,3334
-pycoze/ai/llm/think.py,sha256=sUgTBdGzcZtL3r-Wx8M3lDuVUmDVz8g3qC0VU8uiKAI,5143
+pycoze/ai/llm/__init__.py,sha256=B4apuRE3PDputKTELVYWNP-V5GhdAOBq9BTjzP1Ne3c,125
+pycoze/ai/llm/chat.py,sha256=izriC7nCp5qeJRqcUVQBVqTHiH6MJS77ROzGBJufdNI,5133
+pycoze/ai/llm/text_to_image_prompt.py,sha256=NdfhPTOTSR4whGgd10UtyQjzQCat0wdcX94OLSt4gUU,3333
 pycoze/automation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/automation/browser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/automation/browser/edge_driver_manager.py,sha256=gpgseunph5owZH6EskSYthuhey2SU3UP204gY0yIcuI,3022
@@ -31,8 +31,8 @@ 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/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.223.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.223.dist-info/METADATA,sha256=0FkJta7KAhCl11FGPipeoyHMHcHWGfXJmkxtaHPa9NM,726
-pycoze-0.1.223.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
-pycoze-0.1.223.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.223.dist-info/RECORD,,
+pycoze-0.1.225.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.225.dist-info/METADATA,sha256=N4r76t8hsUkcTCQzuJ_M1WM1L6-rG6V6qUqi5DzrK9M,726
+pycoze-0.1.225.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+pycoze-0.1.225.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.225.dist-info/RECORD,,