yptools 1.0.16 → 1.0.18

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9cb3e39055391054fe343255192ab05476b7e579002f2749e52b664293c47347
-  data.tar.gz: df8a95a939692f3f62da1e04a8240c84feadd2adb2da6a3110445dbb2144a9ce
+  metadata.gz: 5b0c0ac65980ab31a4c668af14aea11d9c2d7ab7dcf0b8c20b072c797b0bea35
+  data.tar.gz: 7451143c8b5d160ff9176532a2570cfef65461e5f78dfa25f0b3627f28bf3653
 SHA512:
-  metadata.gz: 19f54c0bc031663974d6115540de9bf6f0001b12815a9691a2d85774e90618888d1e7674704778d10d90f38f1e695bd5a20c606ebe0d9c505f4e5b10815b2aaf
-  data.tar.gz: eba0155e8c4030b679fb400efd9397aae196fbfdef9523cf074f2c7195ac83ab0d4a2f177a270eaf4c3d9858452e4cdbe35a01e2f4b48370621ba2fff3f214e8
+  metadata.gz: 6664fa97f413646ca8926eef8df0d4a227e46b87af6f87ed53b96b0900e5596931fbc3f1cafb2e4bb4afccc1cbaefdd9c6e54788afeaf5c0e48c93314f603445
+  data.tar.gz: 05036757b34371ad309a41e42b8d2835720571aea0063efeff899dda29d163960c8a4adf6b99a9960aa1800abd04d750a2670ca9b311381001e375f213651c9e

data/lib/yptools/chatai/yp_chatai.rb

@@ -5,49 +5,172 @@ 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
 
+        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 
+                message = yp_message_response["error"]["message"]
+                yp_log_fail  "请求失败，请稍后再试！！！\n" + message
+                return ""
+            end
+        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
+
+    def self.openaiimg(message) 
+        yp_chataiURL = 'https://api.openai.com/v1/images/generations'
+        # 开始发送网络请求
+        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 = message
+
+        # 设置请body
+        data = {
+            # 'model' => 'text-davinci-003', # 然后GPT-3模型会根据您的输入文本自动生成文本补全或摘要。
+            'prompt' => question, # 问的问题
+            "n": 3, #几个回答
+            "size": "512x512"
+        }
+        
         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"]
+                data = yp_message_response["data"]
+                puts ""
+                if !data.empty?
+                    index = 1
+                    for item in data
+                        yp_log_success "图#{index} 【复制下面链接到浏览器打开或者 command + 鼠标左键快速打开 】"
+                        yp_log_msg item["url"]
+                        index = index + 1
+                        puts ""
+                    end
+                else 
+                    yp_log_fail "请求失败，" + yp_message_response
+                end
             else 
-                yp_log_fail "请求失败，" + yp_message_response
+                message = yp_message_response["error"]["message"]
+                yp_log_fail  "请求失败，请稍后再试！！！\n" + message
             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
 

data/lib/yptools/help/yp_help.rb

@@ -3,7 +3,9 @@ class YPHelp
   def self.message
     puts %q{
       
-      chatai: use [yptools chatai ...] 快速与 chatgpt 沟通（科学上网）
+      chatgpt: use [yptools chatgpt] 创建会话列表与 chatgpt 聊天，会记录上下内容（科学上网）
+                   [yptools chatgpt ...] 快速与 chatgpt 沟通，不会记录上下内容
+      openai:  use [yptools openaiimg ...] 根据文本描述生成图像（eg: yptools openaiimg '高冷 少妇 黑丝 短裤 真人照片' ）
 
       autocre: use [yptools autocre ...] 自动化工具命令
                use [yptools autocre -objc ...] 根据 json 自动创建 Objective-C 数据库操作文件 .h|.m 文件。（依赖三方库 FMDB ）

data/lib/yptools.rb

@@ -18,12 +18,20 @@ class YPTools
         cmd = argvs[0]
         case cmd
 
-        when 'chatai'
+        when 'chatgpt'
             if argvs.size > 1
                 name = argvs[1]
                 self.chatai name
             else
-                yp_log_fail "'yptools chatai ..' 参数缺失"
+                self.startChat
+            end
+        when 'openaiimg'
+            if argvs.size > 1
+                name = argvs[1..-1]
+                name = name.join(' ')
+                self.openaiimg name
+            else
+                yp_log_fail "'yptools openaiimg ..' 参数缺失"
                 self.help
             end
         when 'autocre'
@@ -141,6 +149,14 @@ class YPTools
     def self.autoinit
         YPAutoInit.createObjcInitJson
     end
+
+    def self.startChat 
+        YPChatAI.startChatAI()
+    end
+
+    def self.openaiimg(message)
+        YPChatAI.openaiimg(message)
+    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.18
 platform: ruby
 authors:
 - chenghengsheng
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-03-05 00:00:00.000000000 Z
+date: 2023-03-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler