From 9c5b7070100b772b0f4870cc1903d289b5a878b3 Mon Sep 17 00:00:00 2001 From: "Duc Nguyen (john)" Date: Sun, 21 Jan 2024 14:36:07 +0700 Subject: [PATCH] Customize application settings (#132) * Allow customizing the base application * Make the core llms and embeddings customizable * Make the settings, reasoning and index customizable * Import from langchain_openai --- knowledgehub/embeddings/langchain_based.py | 4 ++-- knowledgehub/llms/chats/langchain_based.py | 2 +- knowledgehub/llms/completions/langchain_based.py | 2 +- pyproject.toml | 1 + 4 files changed, 5 insertions(+), 4 deletions(-) diff --git a/knowledgehub/embeddings/langchain_based.py b/knowledgehub/embeddings/langchain_based.py index 0aef4f1..9bd0e7b 100644 --- a/knowledgehub/embeddings/langchain_based.py +++ b/knowledgehub/embeddings/langchain_based.py @@ -122,7 +122,7 @@ class OpenAIEmbeddings(LCEmbeddingMixin, BaseEmbeddings): def _get_lc_class(self): try: - from langchain_community.embeddings import OpenAIEmbeddings + from langchain_openai import OpenAIEmbeddings except ImportError: from langchain.embeddings import OpenAIEmbeddings @@ -152,7 +152,7 @@ class AzureOpenAIEmbeddings(LCEmbeddingMixin, BaseEmbeddings): def _get_lc_class(self): try: - from langchain_community.embeddings import AzureOpenAIEmbeddings + from langchain_openai import AzureOpenAIEmbeddings except ImportError: from langchain.embeddings import AzureOpenAIEmbeddings diff --git a/knowledgehub/llms/chats/langchain_based.py b/knowledgehub/llms/chats/langchain_based.py index 9a815ad..c5c2469 100644 --- a/knowledgehub/llms/chats/langchain_based.py +++ b/knowledgehub/llms/chats/langchain_based.py @@ -163,7 +163,7 @@ class AzureChatOpenAI(LCChatMixin, ChatLLM): def _get_lc_class(self): try: - from langchain_community.chat_models import AzureChatOpenAI + from langchain_openai import AzureChatOpenAI except ImportError: from langchain.chat_models import AzureChatOpenAI diff --git a/knowledgehub/llms/completions/langchain_based.py b/knowledgehub/llms/completions/langchain_based.py index 65122dd..8ad5a59 100644 --- a/knowledgehub/llms/completions/langchain_based.py +++ b/knowledgehub/llms/completions/langchain_based.py @@ -187,7 +187,7 @@ class AzureOpenAI(LCCompletionMixin, LLM): def _get_lc_class(self): try: - from langchain_community.llms import AzureOpenAI + from langchain_openai import AzureOpenAI except ImportError: from langchain.llms import AzureOpenAI diff --git a/pyproject.toml b/pyproject.toml index fdc9e33..1a05069 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -50,6 +50,7 @@ dev = [ "sphinx", "coverage", "openai", + "langchain-openai", "chromadb", "wikipedia", "duckduckgo-search",