Spaces:
Sleeping
Sleeping
File size: 1,038 Bytes
a26db82 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
from abc import abstractmethod, ABC
from langchain.embeddings.base import Embeddings as LCEmbeddings
from langchain.embeddings.openai import OpenAIEmbeddings
from langchain.llms import AzureOpenAI
from langchain.base_language import BaseLanguageModel
from core.lifecycle import Lifecycle
class BaseLangChainManager(Lifecycle, ABC):
def __init__(self) -> None:
super().__init__()
@abstractmethod
def get_embedding(self) -> LCEmbeddings:
pass
@abstractmethod
def get_llm(self) -> BaseLanguageModel:
pass
class LangChainAzureManager(BaseLangChainManager):
def __init__(self) -> None:
super().__init__()
# Override
def get_embedding(self) -> LCEmbeddings:
return OpenAIEmbeddings(client=None, chunk_size=1)
# Override
def get_llm(self) -> BaseLanguageModel:
return AzureOpenAI(
deployment_name="text-davinci-003",
# model_name="text-davinci-003",
model="text-davinci-003",
client=None,
)
|