49ed3f6994
* Create a script to auto-generate markdown docs from pipeline * Clean up documentation for Chain-of-Thought
26 lines
616 B
Python
26 lines
616 B
Python
from langchain.embeddings import OpenAIEmbeddings as LCOpenAIEmbeddings
|
|
|
|
from .base import LangchainEmbeddings
|
|
|
|
|
|
class OpenAIEmbeddings(LangchainEmbeddings):
|
|
"""OpenAI embeddings.
|
|
|
|
This method is wrapped around the Langchain OpenAIEmbeddings class.
|
|
"""
|
|
|
|
_lc_class = LCOpenAIEmbeddings
|
|
|
|
|
|
class AzureOpenAIEmbeddings(LangchainEmbeddings):
|
|
"""Azure OpenAI embeddings.
|
|
|
|
This method is wrapped around the Langchain OpenAIEmbeddings class.
|
|
"""
|
|
|
|
_lc_class = LCOpenAIEmbeddings
|
|
|
|
def __init__(self, **params):
|
|
params["openai_api_type"] = "azure"
|
|
super().__init__(**params)
|