From a96f40f7a208c56bd72844d17b6eacd1831afdc9 Mon Sep 17 00:00:00 2001
From: hibobmaster <bobmaster@hibobmaster.com>
Date: Sun, 16 Apr 2023 10:41:07 +0800
Subject: [PATCH] fix type check

---
 BingImageGen.py |  2 +-
 bot.py          | 18 ++++++------------
 2 files changed, 7 insertions(+), 13 deletions(-)

diff --git a/BingImageGen.py b/BingImageGen.py
index 6071de6..b3c3a4f 100644
--- a/BingImageGen.py
+++ b/BingImageGen.py
@@ -52,7 +52,7 @@ class ImageGenAsync:
 
     def __del__(self):
         try:
-            loop = asyncio.get_event_loop()
+            loop = asyncio.get_running_loop()
         except RuntimeError as e:
             loop = asyncio.new_event_loop()
             asyncio.set_event_loop(loop)
diff --git a/bot.py b/bot.py
index 3944cf0..ade0edf 100644
--- a/bot.py
+++ b/bot.py
@@ -3,7 +3,7 @@ import os
 import re
 import sys
 import traceback
-from typing import Optional, Union
+from typing import Union
 
 import aiohttp
 from nio import (AsyncClient, AsyncClientConfig, InviteMemberEvent, JoinError,
@@ -33,7 +33,7 @@ class Bot:
         device_id: str,
         chatgpt_api_endpoint: str = os.environ.get(
             "CHATGPT_API_ENDPOINT") or "https://api.openai.com/v1/chat/completions",
-        api_key: Optional[str] = os.environ.get("OPENAI_API_KEY") or "",
+        api_key: Union[str, None] = None,
         room_id: Union[str, None] = None,
         bing_api_endpoint: Union[str, None] = None,
         password: Union[str, None] = None,
@@ -116,7 +116,7 @@ class Bot:
         self.help_prog = re.compile(r"^\s*!help\s*.*$")
 
         # initialize chatbot and chatgpt_api_endpoint
-        if self.api_key != '':
+        if self.api_key is not None:
             self.chatbot = Chatbot(api_key=self.api_key, timeout=120)
 
             self.chatgpt_api_endpoint = self.chatgpt_api_endpoint
@@ -125,11 +125,6 @@ class Bot:
                 "Content-Type": "application/json",
                 "Authorization": f"Bearer {self.api_key}",
             }
-        else:
-            self.chatgpt_api_endpoint = self.chatgpt_api_endpoint
-            self.headers = {
-                "Content-Type": "application/json",
-            }
 
         # initialize askGPT class
         self.askgpt = askGPT(self.session)
@@ -149,7 +144,7 @@ class Bot:
 
     def __del__(self):
         try:
-            loop = asyncio.get_event_loop()
+            loop = asyncio.get_running_loop()
         except RuntimeError as e:
             loop = asyncio.new_event_loop()
             asyncio.set_event_loop(loop)
@@ -192,7 +187,7 @@ class Bot:
             n = self.chat_prog.match(content_body)
             if n:
                 prompt = n.group(1)
-                if self.api_key != '':
+                if self.api_key is not None:
                     try:
                         asyncio.create_task(self.chat(room_id,
                                                       reply_to_event_id,
@@ -289,9 +284,8 @@ class Bot:
     # invite_callback event
     async def invite_callback(self, room: MatrixRoom, event: InviteMemberEvent) -> None:
         """Handle an incoming invite event.
-        https://github.com/8go/matrix-eno-bot/blob/ad037e02bd2960941109e9526c1033dd157bb212/callbacks.py#L104
         If an invite is received, then join the room specified in the invite.
-        code copied from: 
+        code copied from: https://github.com/8go/matrix-eno-bot/blob/ad037e02bd2960941109e9526c1033dd157bb212/callbacks.py#L104
         """
         logger.debug(f"Got invite to {room.room_id} from {event.sender}.")
         # Attempt to join 3 times before giving up