add __all__

iblai · Dec 12, 2022 · 1d15450 · 1d15450
1 parent 3af3f43
commit 1d15450
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/ibl_ai_chatgpt_langchain/base.py b/src/ibl_ai_chatgpt_langchain/base.py
@@ -95,6 +95,8 @@ def __call__(self, prompt: str, stop=None) -> str:
         return answer
 
 
+__all__ = ['IBLChatGPT']
+
 if __name__ == "__main__":
     llm = IBLChatGPT()
     response = llm(sys.argv[1])