refactor: rename libro config key

difizen · Oct 8, 2024 · 7e93bf1 · 7e93bf1
1 parent 35f143b
commit 7e93bf1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/libro-ai/src/libro_ai/chat/openai_chat_executor.py b/libro-ai/src/libro_ai/chat/openai_chat_executor.py
@@ -17,7 +17,7 @@ class OpenAIChat(LLMChat):
     def load(self):
         if is_langchain_installed():
             extra_params = {}
-            libro_ai_config = libro_config.get_config().get("libro-ai")
+            libro_ai_config = libro_config.get_config().get("llm")
             if libro_ai_config is not None:
                 if api_key := libro_ai_config.get("OPENAI_API_KEY"):
                     extra_params["api_key"] = api_key

diff --git a/libro-sql/src/libro_sql/database.py b/libro-sql/src/libro_sql/database.py
@@ -75,11 +75,11 @@ class DatabaseManager():
     db: Optional[Database] = None
 
     def __init__(self) -> None:
-        libro_sql_config = libro_config.get_config().get("libro-sql")
+        libro_sql_config = libro_config.get_config().get("db")
         if libro_sql_config is not None:
             self.config(libro_sql_config)
         else:
-            raise ValueError(f"Can not find config of libro-sql")
+            raise ValueError(f"Can not find config of db")
 
     def config(self, c: dict):
         config = DatabaseConfig.model_validate(c)