Merge branch 'CambioML:main' into main

example/chatbot/demo_launch_app_cpu_openai.py

@@ -6,24 +6,29 @@
     You can follow [the installation guide](https://github.com/CambioML/pykoi/tree/install#option-1-rag-cpu)
     to set up the environment.
 - Run the demo:
-    1. Enter your OpenAI API key in the `api_key` below.
+    1. Enter your OpenAI API key a .env file in the `~/pykoi` directory with the name OPEN_API_KEY, e.g.
+        ```
+        OPENAI_API_KEY=your_api_key
+        ```
     2. On terminal and `~/pykoi` directory, run
         ```
         python -m example.chatbot.demo_launch_app_cpu_openai
         ```
 """
+import os
+
+from dotenv import load_dotenv
 
 from pykoi import Application
 from pykoi.chat import ModelFactory
 from pykoi.chat import QuestionAnswerDatabase
 from pykoi.component import Chatbot, Dashboard
 
-
 ##########################################################
 # Creating an OpenAI model (requires an OpenAI API key) #
 ##########################################################
-# enter openai api key here
-api_key = "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"
+load_dotenv()
+api_key = os.getenv("OPENAI_API_KEY")
 
 # Creating an OpenAI model
 model = ModelFactory.create_model(

example/comparator/demo_model_comparator_cpu_openai.py

@@ -2,26 +2,34 @@
 Demo for the chatbot application using multiple OpenAI models.
 
 - Prerequisites:
-    To run this jupyter notebook, you need a `pykoi` environment with the `rag` option. 
-    You can follow [the installation guide](https://github.com/CambioML/pykoi/tree/install#option-1-rag-cpu) 
-    to set up the environment. 
+    To run this jupyter notebook, you need a `pykoi` environment with the `rag` option.
+    You can follow [the installation guide](https://github.com/CambioML/pykoi/tree/install#option-1-rag-cpu)
+    to set up the environment.
 - Run the demo:
-    1. Enter your OpenAI API key in the `api_key` below.
-    2. On terminal and `~/pykoi` directory, run 
+    1. Enter your OpenAI API key a .env file in the `~/pykoi` directory with the name OPEN_API_KEY, e.g.
+        ```
+        OPENAI_API_KEY=your_api_key
+        ```
+    2. On terminal and `~/pykoi` directory, run
         ```
         python -m example.comparator.demo_model_comparator_cpu_openai
         ```
 """
 
+import os
+
+from dotenv import load_dotenv
+
 from pykoi import Application
 from pykoi.chat import ModelFactory
 from pykoi.component import Compare
 
+
 ##########################################################
 # Creating an OpenAI model (requires an OpenAI API key) #
 ##########################################################
-# enter openai api key here
-api_key = ""
+load_dotenv()
+api_key = os.getenv("OPENAI_API_KEY")
 
 # Creating an OpenAI model
 openai_model_1 = ModelFactory.create_model(

pykoi/__init__.py

@@ -1,3 +1,3 @@
 from pykoi.application import Application
 
-__version__ = "0.0.7"
+__version__ = "0.0.9"