yptools 1.0.16 → 1.0.17

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9cb3e39055391054fe343255192ab05476b7e579002f2749e52b664293c47347
-  data.tar.gz: df8a95a939692f3f62da1e04a8240c84feadd2adb2da6a3110445dbb2144a9ce
+  metadata.gz: 05c1d548417627949ac94ebf616c8fa43f617effa7abe15c6696f0fd7a4ebc98
+  data.tar.gz: 77b047b1c9e4a24dbe2ca490884459a715a4f522d25b6ade0d117de59495745f
 SHA512:
-  metadata.gz: 19f54c0bc031663974d6115540de9bf6f0001b12815a9691a2d85774e90618888d1e7674704778d10d90f38f1e695bd5a20c606ebe0d9c505f4e5b10815b2aaf
-  data.tar.gz: eba0155e8c4030b679fb400efd9397aae196fbfdef9523cf074f2c7195ac83ab0d4a2f177a270eaf4c3d9858452e4cdbe35a01e2f4b48370621ba2fff3f214e8
+  metadata.gz: 30bb630cb7d84c1e80dec1bf59f0c667b53f21ba96ee0284c92013af3421eed6bea1290ddccd6fd85cf3d9d0f9ac2ddf34a4a2ebba9c323b8a7e34a4dc2c8380
+  data.tar.gz: fa7e14ba1b2a495849921bb73b64d71446a0bed3b4c09a50bafa928c415e325f994c8ee1208810f04fb54b73f2fd0f53ef065dcbc06e679e5710dfe18ff301f3

data/lib/yptools/chatai/yp_chatai.rb

@@ -5,50 +5,114 @@ require 'base64'
 
 class YPChatAI
     
-    def self.message(message)
-        yp_log_doing "你的问题：#{message}"
-        yp_log_msg "请求中，请等待..."
-        yp_chataiURL = 'https://api.openai.com/v1/completions'
+    def self.startChatAI
+        system('clear')
+        yp_log_msg "\n欢迎使用 YPTools & ChatGPT"
+        yp_log_msg "YPTools源码地址：https://github.com/HansenCCC/YPTools"
+        yp_log_msg "OpenAI地址：https://github.com/openai\n"
+        yp_log_success "ChatGPT"
+        yp_log_doing "输入 'q' 或者 'quit' 退出当前会话"
+        yp_log_msg '我是一个非常聪明的ChatGPT机器人。如果您问我一个根源于真理的问题，我会给您答案。'
+        yp_log_msg 'Q: 全球人类的平均寿命是多少？'
+        yp_log_msg 'A: 根据世界卫生组织公布的数据，2019年全球人类的平均寿命为71.4岁。'
+
+        yp_contents = Array.new
+        while true
+            Encoding.default_external = Encoding::UTF_8
+            Encoding.default_internal = Encoding::UTF_8
+            # 使用while循环
+            print "Q: "
+            yp_question = STDIN.gets.chomp
+            if (yp_question.upcase == 'Q' || yp_question.upcase == 'QUIT')
+                break
+            end
+            yp_contents.push({
+                'USER' => yp_question,
+                'AI' => ""
+            })
+            yp_answer = self.chatGPTWithQuestion(yp_contents)
+            print "A: " + yp_answer + "\n"
+            yp_item = yp_contents.pop
+            yp_item["AI"] = yp_answer
+            yp_contents.push(yp_item)
+        end
+    end
 
+    def self.chatGPTWithQuestion(content)
+        yp_chataiURL = 'https://api.openai.com/v1/completions'
+        # yp_chataiURL = 'https://api.openai.com/v1/engines/davinci-codex/completions'
         # 开始发送网络请求
         url = URI.parse(yp_chataiURL)
         http = Net::HTTP.new(url.host, url.port)
         http.use_ssl = true 
-        # 设置请求头
+        # 简单了加了下密，免费的apikey，就不要扒去用了（我的代码是开源放到GitHub，加密的目的是ChatGPT会检测秘钥是否在网络泄露）
         yp_token = "c2stTVRVMVZqeUdpNWxzYlU1TmNlU1pUM0JsYmtGSmNYam5iUk5ROENVYUd2QVR4WXpp"
         # 将Base64字符串解码为原始二进制数据
         decoded_data = Base64.decode64(yp_token).force_encoding("UTF-8")
-
+        # 设置请求头
         headers = {
             'Content-Type' => 'application/json',
             'Authorization' => 'Bearer ' + decoded_data
         }
+        # 设置ai根据上下文
+        question = ''
+        for key in content
+            user = key['USER']
+            ai = key['AI']
+            question += '\nUSER: ' + user + '\nAI: ' + ai
+        end
+
         # 设置请body
         data = {
+            # 'engine' => 'davinci',
             'model' => 'text-davinci-003', # 然后GPT-3模型会根据您的输入文本自动生成文本补全或摘要。
-            'prompt' => message, # 问的问题
+            'prompt' => question, # 问的问题
             'max_tokens' => 999, # 设置回答最多多少个字符 
-            'temperature' => 0.9, # 文本创意度，默认 1
+            'temperature' => 0.7, # 文本创意度，默认 1
+            "n": 1, #几个回答
+            "stop": "\n"
         }
-
+        
         request = Net::HTTP::Post.new(url.path, headers)
         request.body = data.to_json
-        response = http.request(request)
 
-        yp_message_response = JSON(response.body)
-        if !yp_message_response["error"]
-            created = yp_message_response["created"]
-            choices = yp_message_response["choices"]
-            if !choices.empty?
-                text =  choices.first["text"]
-                yp_log_success "chatGPT：" + text.gsub(/\n+/, "\n")
+        begin
+            response = http.request(request)
+            # 处理响应数据
+            yp_message_response = JSON(response.body)
+            if !yp_message_response["error"]
+                created = yp_message_response["created"]
+                choices = yp_message_response["choices"]
+                if !choices.empty?
+                    text = choices.first["text"]
+                    text = text.gsub(/\n+/, "")
+                    return text
+                else 
+                    yp_log_fail "请求失败，" + yp_message_response
+                    return ""
+                end
             else 
-                yp_log_fail "请求失败，" + yp_message_response
+                message = yp_message_response["error"]["message"]
+                yp_log_fail  "请求失败，请稍后再试！！！\n" + message
+                return ""
             end
-        else 
-            message = yp_message_response["error"]["message"]
-            yp_log_fail  "请求失败，请稍后再试！！！\n" + message
+        rescue StandardError => e
+            # 处理异常
+            yp_log_fail "请求的次数太多了，请稍后再试！！！"
+            yp_log_fail "发生异常：#{e.message}"
         end
     end
 
+    def self.message(message)
+        yp_log_doing "你的问题：#{message}"
+        yp_log_msg "请求中，请等待..."
+        yp_contents = Array.new
+        yp_contents.push({
+            'USER' => message,
+            'AI' => ""
+        })
+        yp_answer = self.chatGPTWithQuestion(yp_contents)
+        yp_log_success "chatGPT：" + yp_answer.gsub(/\n+/, "\n")
+    end
+
 end

data/lib/yptools/help/yp_help.rb

@@ -3,7 +3,8 @@ class YPHelp
   def self.message
     puts %q{
       
-      chatai: use [yptools chatai ...] 快速与 chatgpt 沟通（科学上网）
+      chatai: use [yptools chatai] 创建会话列表与 chatgpt 聊天，会记录上下内容（科学上网）
+                  [yptools chatai ...] 快速与 chatgpt 沟通，不会记录上下内容
 
       autocre: use [yptools autocre ...] 自动化工具命令
                use [yptools autocre -objc ...] 根据 json 自动创建 Objective-C 数据库操作文件 .h|.m 文件。（依赖三方库 FMDB ）

data/lib/yptools.rb

@@ -23,8 +23,7 @@ class YPTools
                 name = argvs[1]
                 self.chatai name
             else
-                yp_log_fail "'yptools chatai ..' 参数缺失"
-                self.help
+                self.startChat
             end
         when 'autocre'
             if argvs.size > 1
@@ -141,6 +140,10 @@ class YPTools
     def self.autoinit
         YPAutoInit.createObjcInitJson
     end
+
+    def self.startChat 
+        YPChatAI.startChatAI()
+    end
     
 end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: yptools
 version: !ruby/object:Gem::Version
-  version: 1.0.16
+  version: 1.0.17
 platform: ruby
 authors:
 - chenghengsheng
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-03-05 00:00:00.000000000 Z
+date: 2023-03-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler