From d2e3b7503b3f4dd045dbc35a4662cfdb937499a0 Mon Sep 17 00:00:00 2001
From: dece <shgck@pistache.land>
Date: Tue, 29 Nov 2022 12:53:47 +0100
Subject: [PATCH] gpt3: add plugin

---
 Pipfile                     |  1 +
 config.json.example         |  2 ++
 edmond/plugins/gpt3.py      | 48 +++++++++++++++++++++++++++++++++++++
 edmond/plugins/wikipedia.py | 21 +++-------------
 edmond/utils.py             | 16 +++++++++++++
 5 files changed, 70 insertions(+), 18 deletions(-)
 create mode 100644 edmond/plugins/gpt3.py

diff --git a/Pipfile b/Pipfile
index 5ebf86a..81b7559 100644
--- a/Pipfile
+++ b/Pipfile
@@ -19,3 +19,4 @@ translate = "*"
 scaruffi = "==0.0.3"
 wolframalpha = "*"
 meteofrance-api = "~=1.0.2"
+openai = "~=0.25.0"
diff --git a/config.json.example b/config.json.example
index c0eaca9..d573096 100644
--- a/config.json.example
+++ b/config.json.example
@@ -70,6 +70,8 @@
     },
     "gpt3": {
       "openai_key": ""
+      "join_lines": "; ",
+      "computing_replies": ["Hmm…"]
     },
     "horoscope": {
       "commands": ["horoscope"],
diff --git a/edmond/plugins/gpt3.py b/edmond/plugins/gpt3.py
new file mode 100644
index 0000000..3cc2a98
--- /dev/null
+++ b/edmond/plugins/gpt3.py
@@ -0,0 +1,48 @@
+import random
+import re
+from typing import Optional
+
+import openai
+
+from edmond.plugin import Plugin
+from edmond.utils import limit_text_length
+
+
+class Gpt3Plugin(Plugin):
+
+    REQUIRED_CONFIGS = ["openai_key", "computing_replies", "join_lines"]
+
+    def __init__(self, bot):
+        super().__init__(bot)
+        openai.api_key = self.config["openai_key"]
+
+    def reply(self, prompt: str, target: str):
+        computing_reply = random.choice(self.config["computing_replies"])
+        self.bot.say(target, computing_reply)
+
+        completion = self.complete(prompt)
+        if completion and (reply := self.sanitize(completion)):
+            self.bot.say(target, reply)
+        else:
+            self.bot.signal_failure(target)
+
+    def complete(self, prompt: str) -> Optional[str]:
+        try:
+            completion = openai.Completion.create(
+                model="text-davinci-002",
+                prompt=prompt,
+                temperature=0.7,
+                max_tokens=128,
+                top_p=1,
+                frequency_penalty=0.5,
+                presence_penalty=0
+            )
+        except openai.error.OpenAIError:
+            return None
+        return completion.choices[0].text
+
+    def sanitize(self, text: str) -> str:
+        text = text.strip()
+        text = re.sub(r"\n+", self.config["join_lines"], text)
+        text = limit_text_length(text)
+        return text
diff --git a/edmond/plugins/wikipedia.py b/edmond/plugins/wikipedia.py
index 774179a..8fb9911 100644
--- a/edmond/plugins/wikipedia.py
+++ b/edmond/plugins/wikipedia.py
@@ -5,6 +5,7 @@ import wikipedia
 
 from edmond.plugin import Plugin
 from edmond.plugins.plus import PlusPlugin
+from edmond.utils import limit_text_length
 
 
 class WikipediaPlugin(Plugin):
@@ -47,7 +48,7 @@ class WikipediaPlugin(Plugin):
                 self.bot.log_d(f"Wikipedia exception: {exc}")
             retries -= 1
         if page:
-            reply = WikipediaPlugin.limit_text_length(page.summary)
+            reply = limit_text_length(page.summary)
             self.register_url_for_plus(page.url, event.target)
             self.bot.say(event.target, reply)
 
@@ -71,7 +72,7 @@ class WikipediaPlugin(Plugin):
                 time.sleep(1)
             retries -= 1
         if page:
-            reply = WikipediaPlugin.limit_text_length(page.summary)
+            reply = limit_text_length(page.summary)
             self.register_url_for_plus(page.url, event.target)
         self.bot.say(event.target, reply)
 
@@ -80,19 +81,3 @@ class WikipediaPlugin(Plugin):
             def handler(plus_event):
                 self.bot.say(plus_event.target, url)
             cast(PlusPlugin, plus_plugin).add_handler(target, handler)
-
-    @staticmethod
-    def limit_text_length(text, max_length=200):
-        """Limit text size to 200 characters max."""
-        words = text.split(" ")
-        cut_text = ""
-        while words:
-            next_word = words.pop(0)
-            if len(cut_text) + len(next_word) + 1 >= max_length:
-                break
-            cut_text += next_word + " "
-        if len(cut_text) < len(text):
-            cut_text = cut_text[:-1] + "…"
-        else:
-            cut_text = cut_text.rstrip()
-        return cut_text
diff --git a/edmond/utils.py b/edmond/utils.py
index 410b9a2..514c032 100644
--- a/edmond/utils.py
+++ b/edmond/utils.py
@@ -13,3 +13,19 @@ def http_get(url: str) -> Optional[str]:
 
 def proc(proba_percentage: int) -> bool:
     return random.random() < (proba_percentage / 100.0)
+
+
+def limit_text_length(text, max_length=400):
+    """Limit text size to 400 characters max."""
+    words = text.split(" ")
+    cut_text = ""
+    while words:
+        next_word = words.pop(0)
+        if len(cut_text) + len(next_word) + 1 >= max_length:
+            break
+        cut_text += next_word + " "
+    if len(cut_text) < len(text):
+        cut_text = cut_text[:-1] + "…"
+    else:
+        cut_text = cut_text.rstrip()
+    return cut_text