diff --git a/libro-ai/src/libro_ai/chat/openai_chat_executor.py b/libro-ai/src/libro_ai/chat/openai_chat_executor.py index 463a452..65d96ac 100644 --- 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 index 577964d..226e313 100644 --- 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)