publish 0.0.7

Author: heyWFeng

README.md

@@ -22,8 +22,8 @@
     	<a target="_blank" href='https://gitee.com/CoderWanFeng/python-office'>
 		<img src='https://gitee.com/CoderWanFeng//python-office/badge/star.svg?theme=dark' alt='gitee star'/>
 	</a>
-  	<a href="https://mp.weixin.qq.com/s/Jf_EVdKlVnHhK68fW5OA6A">
-	<img src="https://img.shields.io/badge/QQ-1090738447-orange"/>
+  	<a href="https://mp.weixin.qq.com/s/yaSmFKO3RrBpyanW3nvRAQ">
+	<img src="https://img.shields.io/badge/QQ-163434413-orange"/>
   </a>
     	<a href="https://mp.weixin.qq.com/s/wx-JkgOUoJhb-7ZESxl93w">
 	<img src="https://img.shields.io/badge/%E5%BE%AE%E4%BF%A1-%E4%BA%A4%E6%B5%81%E7%BE%A4-brightgreen"/>
@@ -70,4 +70,4 @@ pip install -i https://pypi.tuna.tsinghua.edu.cn/simple poai -U
 ## 📌进交流群
 
 
-![CoderWanFeng](https://python-office-1300615378.cos.ap-chongqing.myqcloud.com/python-office-qr.jpg)
+![CoderWanFeng](https://python-office-1300615378.cos.ap-chongqing.myqcloud.com/python-office-qr.jpg)

demo/result.txt

@@ -0,0 +1,19 @@
+，我是李红英。
+
+Hi, I'm Li Hongying.
+
+我的电话号码是：
+
+My phone number is:
+
+13670067700。
+
+13670067700.
+
+请记住我的电话号码。
+
+Please remember my phone number.
+
+谢谢！
+
+Thank you!

poai/__init__.py

@@ -14,5 +14,5 @@
 print('使用有问题 or 提交你的功能需求 or 参与项目开发')
 print('1、请+【项目交流群】：http://t.cn/A65MiFvH')
 print('2、本开源项目的【代码仓库】：https://github.com/CoderWanFeng/poai')
-print('3、免费的微信机器人：https://b23.tv/jDVRtU5')
+print('3、免费的微信机器人：https://mp.weixin.qq.com/s/9aspEHdCiAdXK17AvHlu9Q')
 print(SPLIT_LINE)

poai/api/chatgpt.py

@@ -9,19 +9,21 @@
 import openai
 
 
-def chat(api_key, prompt, model_engine="text-davinci-002"):
+def chat(api_key, prompt, model_engine="text-davinci-002", max_tokens=1024, n=1, stop=None, temperature=0.5, top_p=1,
+         frequency_penalty=0.0, presence_penalty=0.6):
     # 设置 API Key
     openai.api_key = api_key  # your_api_key
-
     completions = openai.Completion.create(
         engine=model_engine,
         prompt=prompt,
-        max_tokens=1024,
-        n=1,
-        stop=None,
-        temperature=0.5,
+        max_tokens=max_tokens,  # 生成结果时的最大 tokens 数。平均一个汉字是 2 个 tokens，text-davinci-003 最多是 4000 个 tokens，也就是 2000 个汉字左右
+        n=n,
+        stop=stop,
+        temperature=temperature,  # 控制结果的随机性，如果希望结果更有差异性 0.9，或者希望有固定结果可以尝试 0.0
+        top_p=top_p,  # 一个可用于代替 temperature 的参数，对应机器学习中 nucleus sampling，如果设置 0.1 意味着只考虑构成前 10% 概率质量的 tokens
+        frequency_penalty=frequency_penalty,  # 控制字符的重复度，取值为 -2.0 ~ 2.0 之间的数字
+        presence_penalty=presence_penalty  # 控制主题的重复度，取值为 -2.0 ~ 2.0 之间的数字
     )
-
     # 获取 ChatGPT 的回复
     message = completions.choices[0].text
     print(message)

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = poai
-version = 0.0.6
+version = 0.0.7
 description = pip install poai
 long_description = file: README.md
 long_description_content_type = text/markdown