feat(ai): use libro config for libro ai

difizen · Oct 8, 2024 · 35f143b · 35f143b
1 parent 860d955
commit 35f143b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/libro-ai/pyproject.toml b/libro-ai/pyproject.toml
@@ -13,6 +13,7 @@ dependencies = [
     "pydantic>=2.6.4",
     "requests>=2.31.0",
     "nbclient>=0.10.0",
+    "libro-core>=0.1.1",
 ]
 readme = "README.md"
 requires-python = ">= 3.10"

diff --git a/libro-ai/src/libro_ai/chat/openai_chat_executor.py b/libro-ai/src/libro_ai/chat/openai_chat_executor.py
@@ -7,7 +7,7 @@
 from langchain.callbacks import get_openai_callback
 from langchain_community.utilities.dalle_image_generator import DallEAPIWrapper
 from IPython.display import display
-
+from libro_core.config import libro_config
 
 class OpenAIChat(LLMChat):
     name: str = "chatgpt"
@@ -16,7 +16,12 @@ class OpenAIChat(LLMChat):
 
     def load(self):
         if is_langchain_installed():
-            self.chat = ChatOpenAI(model_name=self.model)
+            extra_params = {}
+            libro_ai_config = libro_config.get_config().get("libro-ai")
+            if libro_ai_config is not None:
+                if api_key := libro_ai_config.get("OPENAI_API_KEY"):
+                    extra_params["api_key"] = api_key
+            self.chat = ChatOpenAI(model_name=self.model,**extra_params)
             return True
         return False