xiaothink 1.0.0__py3-none-any.whl

xiaothink/__init__.py

@@ -0,0 +1,7 @@
+import xiaothink.openapi
+import xiaothink.openapi.chatbot_old
+import xiaothink.openapi.drawer
+import xiaothink.openapi.writer
+import xiaothink.openapi.yanzhi
+import xiaothink.llm.inference.test_formal
+import xiaothink.llm.inference.test

xiaothink/openapi/chatbot_old.py

@@ -0,0 +1,29 @@
+import requests
+import os
+#import set_conf
+def get_key(key):
+    p='C:'+os.environ['HOMEPATH']
+    try:
+        #open(os.path.join(p,'xiaothink_conf.conf','rb'))
+        with open(os.path.join(p,'xiaothink_conf.conf'),'r',encoding='utf-8')as f:
+            data=eval(f.read())
+    except FileNotFoundError:
+        
+            data={}
+    if key in list(data.keys()):
+        return data[key]
+    else:
+        return None
+
+
+    
+def chat(text,usern=None):
+    """
+    text: 聊天内容
+    usern:你的用户名（暂时不需要注册，直接起一个就行）
+    """
+    if not usern:
+        usern=get_key('XIAOTHINK_USERNAME')
+    #print(usern)
+    r = requests.get(f'https://4147093qp2.imdo.co/openapi/user-chat?inp={text}&user={usern}')
+    return r.json()['data']['result'].split('\n')[0]

xiaothink/openapi/drawer.py

@@ -0,0 +1,63 @@
+import requests
+import urllib.parse
+
+
+class Drawer():
+    def __init__(self,return_type='bytes'):
+        self.return_type=return_type
+
+    def draw(self,description,style,filename=None):
+        
+
+        """
+        根据给定的艺术图像描述和风格编号，调用清韵AI艺术图创作API，并返回生成的图像内容。
+
+        :param description: str 艺术图像描述
+        :param style: int 图像风格编号（1-8）
+        :return: bytes 图像二进制数据
+        """
+
+        # 对描述进行URL编码
+        encoded_description = urllib.parse.quote(description)
+
+        # 构造请求URL
+        url = f"https://4147093qp2.imdo.co/aidrawnew?text={encoded_description}&syle={style}"#注意这里是syle，别问为什么
+
+        try:
+            # 发送GET请求
+            response = requests.get(url)
+            
+            # 检查请求是否成功
+            if response.status_code == 200:
+                # 获取并返回图像二进制数据
+                if self.return_type=='bytes':
+                    return response.content
+                elif self.return_type=='file':
+                    with open(filename, "wb") as f:
+                        f.write(response.content)
+                    return None
+                        
+            else:
+                print(f"请求失败，状态码：{response.status_code}")
+                return None
+
+        except requests.exceptions.RequestException as e:
+            print(f"请求过程中发生错误：{e}")
+            return None
+
+    def help(self):
+        print('''风格编号（1-8）：
+    风格1 - 糖果 （candy）
+    风格2 - 组成vii（composition vii）
+    风格3 - 羽毛 （feathers）
+    风格4 - la muse
+    风格5 - 马赛克（mosaic）
+    风格6 - 梵高·星空（starry_night）
+    风格7 - 奶油（the_scream）
+    风格8 - 波形（the_wave）
+——————————————————————————————————
+    支持的return_type：1.'bytes'——返回为图像的二进制数据
+                       2.'file'——直接保存为filename
+                       
+''')
+

xiaothink/openapi/set_conf.py

@@ -0,0 +1,30 @@
+import os
+def set_user(name):
+    p='C:'+os.environ['HOMEPATH']
+    try:
+        open(os.path.join(p,'xiaothink_conf.conf','rb'))
+        with open(os.path.join(p,'xiaothink_conf.conf'),'r',encoding='utf-8')as f:
+            data=eval(f.read())
+    except FileNotFoundError:
+        
+            data={}
+            
+    data['XIAOTHINK_USERNAME']=name
+    with open(os.path.join(p,'xiaothink_conf.conf'),'w',encoding='utf-8')as f:
+            f.write(str(data))
+
+def get_key(key):
+    p='C:'+os.environ['HOMEPATH']
+    try:
+        #open(os.path.join(p,'xiaothink_conf.conf','rb'))
+        with open(os.path.join(p,'xiaothink_conf.conf'),'r',encoding='utf-8')as f:
+            data=eval(f.read())
+    except FileNotFoundError:
+        
+            data={}
+    if key in list(data.keys()):
+        return data[key]
+    else:
+        return None
+    
+

xiaothink/openapi/writer.py

@@ -0,0 +1,23 @@
+import requests
+import urllib.parse
+
+
+def write(original_text):
+    # 对原文内容进行URL编码
+    encoded_text = urllib.parse.quote(original_text)
+
+    # 构造API请求URL
+    api_url = f"https://4147093qp2.imdo.co/zhxxapi?inp={encoded_text}"
+
+    # 发送GET请求并获取响应
+    response = requests.get(api_url)
+    
+    # 解析返回结果
+    result = response.json()
+
+    if result["Error"] == "None":
+        return original_text + result["text"]
+    else:
+        print("错误信息：", result["Error"])
+        return None
+

xiaothink/openapi/yanzhi.py

@@ -0,0 +1,19 @@
+import requests
+
+def get_score(image_url="https://4147093qp2.imdo.co/yanzhi/54.jpg"):
+    # API接口
+    api_url = f"https://4147093qp2.imdo.co/yanzhi?url={image_url}"
+    
+
+    # 发送请求
+    response = requests.get(api_url)
+    
+    # 检查请求是否成功
+    if response.status_code == 200:
+        # 从响应中获取颜值分数
+        beauty_score = float(response.json()['ret'])
+        return beauty_score
+    else:
+        print("Request failed with status code:", response.status_code)
+        return None
+

xiaothink-1.0.0.dist-info/LICENSE

@@ -0,0 +1,16 @@
+Copyright (c) 2018 The Python Packaging Authority
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

xiaothink-1.0.0.dist-info/METADATA

@@ -0,0 +1,186 @@
+Metadata-Version: 2.1
+Name: xiaothink
+Version: 1.0.0
+Summary: 一个AI工具包，帮助用户快速调用小思框架（Xiaothink）相关接口。
+Home-page: UNKNOWN
+Author: Ericsjq
+Author-email: xiaothink@foxmail.com
+License: UNKNOWN
+Platform: UNKNOWN
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: tensorflow
+Requires-Dist: requests
+
+# Xiaothink Python 模块使用文档
+
+Xiaothink 是一个以自然语言处理（NLP）为核心的AI研究组织，致力于提供高效、灵活的工具来满足各种应用场景的需求。Xiaothink Python 模块是该组织提供的核心工具包，涵盖了图像生成、文本续写、颜值评分以及对话模型等多种功能。以下是详细的使用指南和代码示例。
+
+## 目录
+1. [安装](#安装)
+2. [图像生成](#图像生成)
+3. [文本续写](#文本续写)
+4. [颜值评分](#颜值评分)
+5. [在线AI对话](#在线ai对话)
+6. [本地对话模型](#本地对话模型)
+
+---
+
+## 安装
+
+首先，您需要通过 pip 安装 Xiaothink 模块：
+
+```bash
+pip install xiaothink
+```
+
+---
+
+## 图像生成
+
+Xiaothink 提供了强大的图像生成功能，可以通过简单的 API 调用来生成高质量的图像。
+
+### 示例代码
+
+```python
+import xiaothink as xt
+
+drawer = xt.openapi.drawer.Drawer(return_type='bytes')
+with open('a.jpg', 'wb') as f:
+    f.write(drawer.draw('两只猫', style=6))
+```
+
+---
+
+## 文本续写
+
+您可以使用 Xiaothink 的文本续写功能来生成连贯且有意义的文本内容。
+
+### 示例代码
+
+```python
+import xiaothink as xt
+
+text = '从前，由一位老人，'
+out_ = xt.openapi.writer.write(text)
+print('续写内容：', out_)
+```
+
+---
+
+## 颜值评分
+
+Xiaothink 提供了基于 Web URL 的颜值评分服务。请注意，该服务仅支持传入网络图片 URL，不支持本地路径。
+
+### 示例代码
+
+```python
+import xiaothink as xt
+
+image_url = "https://example.com/image.jpg"
+num = xt.openapi.yanzhi.get_score(image_url)
+print('颜值得分：', num)
+```
+
+**重要提示**：`xt.openapi.yanzhi.get_score` 函数只支持传入 Web URL，不支持传入本地路径。
+
+---
+
+## 在线AI对话
+
+Xiaothink 提供了一个简便易用的在线 AI 对话接口。设置用户名后即可进行多轮对话。
+
+
+### 示例代码
+
+```python
+import xiaothink as xt
+
+xt.openapi.set_conf.set_user('YOUR_USERNAME')
+
+while True:
+    inp = input('【问】：')
+    re = xt.openapi.chatbot_old.chat(inp)
+    print('\n【答】：', re, '\n')
+```
+
+**重要提示**：在线 AI 对话与本地模型对话是两个完全不同的系统，毫无关系。`xt.openapi.chatbot_old.chat` 函数设置用户名后便支持多轮对话。
+
+---
+
+## 本地对话模型
+
+对于本地加载的对话模型，根据模型类型的不同，应调用相应的函数来进行对话。
+
+### 单轮对话
+
+适用于单轮对话场景。
+
+### 示例代码
+
+```python
+import xiaothink.llm.inference.test_formal as tf
+
+model = tf.QianyanModel(
+    ckpt_dir=r'E:\小思框架\论文\ganskchat\ckpt_test_40_2_3_1_formal_open',
+    MT=40.231,
+    vocab=r'E:\小思框架\论文\ganskchat\vocab_lx3.txt'
+)
+
+while True:
+    inp = input('【问】：')
+    if inp == '[CLEAN]':
+        print('【清空上下文】\n\n')
+        model.clean_his()
+        continue
+    re = model.chat_SingleTurn(inp)  # 使用 chat_SingleTurn 进行单轮对话
+    print('\n【答】：', re, '\n')
+```
+
+### 多轮对话
+
+适用于多轮对话场景。
+
+### 示例代码
+
+```python
+import xiaothink.llm.inference.test as test
+
+MT = 40.231
+m, d = test.load(
+    ckpt_dir=r'E:\小思框架\论文\ganskchat\ckpt_test_40_2_3_1_qas',
+    model_type=MT,
+    vocab=r'E:\小思框架\论文\ganskchat\vocab_lx3.txt',
+)
+
+b_chat = {
+    1: '给定一个数字列表，找出其中的最大值和最小值。\n数字列表：[3, 9, 2, 7, 5, 1, 8, 4, 6, 0]',
+    2: '请生成一段关于“春风”为主题的短故事，提供完整的答案。\n',
+    3: '请回答下列问题：“什么是全球变暖？”\n',
+    4: '请创建一个简单的Python程序，用于计算两数乘积\n',
+    5: '创建一个简单的计算器，并提供任何必需的数学原理。\n',
+    6: '根据给定的单词，生成与该单词相关的五个同义词。\n单词：友谊\n',
+}[6]
+
+belle_chat = '{"instruction": "{b_chat}", "input": "", "output": "'.replace('{b_chat}', b_chat)
+inp_m = belle_chat
+
+ret = test.generate_texts_loop(m, d, inp_m,
+                               num_generate=100,
+                               every=lambda a: print(a, end='', flush=True),
+                               temperature=0.55,
+                               pass_char=['▩'])
+```
+
+**重要提示**：对于本地模型，单论对话模型应调用 `model.chat_SingleTurn` 函数，多轮对话模型应调用 `model.chat` 函数。
+
+---
+
+以上就是 Xiaothink Python 模块的主要功能及使用方法。
+
+如有任何疑问或建议，请随时联系我们：xiaothink@foxmail.com。
+

xiaothink-1.0.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+xiaothink/__init__.py,sha256=Txr_4qpdb_LCW0PhWcJ-NCvuZ2kirL9BHSBjJFnShNA,242
+xiaothink/openapi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+xiaothink/openapi/chatbot_old.py,sha256=qgpN-Yxzj1h-zxt5tS44nKz7vfYp8f_QU5KjRcZL74Y,830
+xiaothink/openapi/drawer.py,sha256=gVHJIIRWZ1CNwPvqgd--2NWaPd9U7K9hlOc11dp06jY,2238
+xiaothink/openapi/set_conf.py,sha256=AY6CbPz4rIcVi3k1Km1MUQYPEQmuFjRNu__rNUERIms,881
+xiaothink/openapi/writer.py,sha256=yLIq1hoP_UFOsLbD9T8_95by4XfO4U3Ly1XFACd7j94,570
+xiaothink/openapi/yanzhi.py,sha256=-1NdT5fd2WesbKNCJ0V5-t_fSvXlcIUA4z5T_6yKhcg,547
+xiaothink-1.0.0.dist-info/LICENSE,sha256=GzHS9pi6vw_oGwKIGxAPch8iuVrHazOilEsYL7AlPK0,1085
+xiaothink-1.0.0.dist-info/METADATA,sha256=hz-B4o1Ov0GXclSRsqk3DxIh0GgpO3pc-JC71IzF9Bo,5151
+xiaothink-1.0.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+xiaothink-1.0.0.dist-info/top_level.txt,sha256=2Dpv6MYJdOSpU2IK6tlPtmzt28b11pOV72BPRH4tRpE,10
+xiaothink-1.0.0.dist-info/RECORD,,

xiaothink-1.0.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.37.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

xiaothink-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+xiaothink