Add 'check_api_key'

1. Verify if the API key is entered.
2. Determine if the API key is usable.
3. Display the corresponding error or warning.

Author: Lin-jun-xiang

app.py

@@ -87,7 +87,7 @@ def load_api_key() -> None:
         if temp_file_path:
             os.remove(temp_file_path)
 
-        docGPT_tool, calculate_tool, search_tool, llm_tool = None, None, None, None
+        docGPT_tool, calculate_tool, search_tool, llm_tool = [None] * 4
 
         try:
             agent_ = AgentHelper()

docGPT/__init__.py

@@ -1 +1,2 @@
 from .docGPT import DocGPT
+from .check_api_key import OpenAiAPI, SerpAPI

docGPT/check_api_key.py

@@ -0,0 +1,60 @@
+import os
+from abc import ABC, abstractmethod
+
+import openai
+import streamlit as st
+from langchain import SerpAPIWrapper
+
+
+class ApiKey(ABC):
+    """Check the Api key is valid or not"""
+    query = 'This is a test.'
+
+    @classmethod
+    @abstractmethod
+    def is_valid(cls):
+        pass
+
+
+class OpenAiAPI(ApiKey):
+    @classmethod
+    def is_valid(cls) -> str:
+        if not st.session_state['openai_api_key']:
+            st.error('⚠️ :red[You have not pass OpenAI API key.] Necessary Pass')
+            return
+
+        openai.api_key = os.getenv('OPENAI_API_KEY')
+        try:
+            response = openai.Completion.create(
+                engine='davinci',
+                prompt=cls.query,
+                max_tokens=5
+            )
+            return response
+        except Exception as e:
+            st.error(
+                '🚨 :red[Your OpenAI API key has a problem.] '
+                '[Check your usage](https://platform.openai.com/account/usage)'
+            )
+            print(f'Test error\n{e}')
+
+
+class SerpAPI(ApiKey):
+    @classmethod
+    def is_valid(cls) -> str:
+        if not st.session_state['serpapi_api_key']:
+            st.warning('⚠️ You have not pass SerpAPI key. (You cannot ask current events.)')
+            return
+
+        os.environ['SERPAPI_API_KEY'] = os.getenv('SERPAPI_API_KEY')
+        try:
+            search = SerpAPIWrapper()
+            response = search.run(cls.query)
+            return response
+        except Exception as e:
+            st.error(
+                '🚨 :red[Your SerpAPI key has a problem.] '
+                '[Check your usage](https://serpapi.com/dashboard)'
+            )
+            print(f'Test error\n{e}')
+