pycoze 0.1.223__tar.gz → 0.1.225__tar.gz

{pycoze-0.1.223 → pycoze-0.1.225}/PKG-INFO

@@ -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.225/pycoze/ai/__init__.py

@@ -0,0 +1,2 @@
+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

pycoze-0.1.225/pycoze/ai/llm/__init__.py

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

pycoze-0.1.223/pycoze/ai/llm/think.py → pycoze-0.1.225/pycoze/ai/llm/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-0.1.223 → pycoze-0.1.225}/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 → pycoze-0.1.225}/pycoze.egg-info/PKG-INFO

@@ -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 → pycoze-0.1.225}/pycoze.egg-info/SOURCES.txt

@@ -9,8 +9,8 @@ pycoze.egg-info/top_level.txt
 pycoze/ai/__init__.py
 pycoze/ai/vram_reserve.py
 pycoze/ai/llm/__init__.py
+pycoze/ai/llm/chat.py
 pycoze/ai/llm/text_to_image_prompt.py
-pycoze/ai/llm/think.py
 pycoze/automation/__init__.py
 pycoze/automation/browser/__init__.py
 pycoze/automation/browser/edge_driver_manager.py

{pycoze-0.1.223 → pycoze-0.1.225}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pycoze",
-    version="0.1.223",
+    version="0.1.225",
     packages=find_packages(),
     install_requires=[],
     author="Yuan Jie Xiong",

pycoze-0.1.223/pycoze/ai/__init__.py

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

pycoze-0.1.223/pycoze/ai/llm/__init__.py

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