github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-14 17:07:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Harrison/azure openai (#3942)

Browse Source 
Co-authored-by: Saverio Proto <zioproto@gmail.com>
This commit is contained in:
Harrison Chase 2023-05-01 21:34:16 -07:00 committed by GitHub
parent 48ea27ba60
commit 2366e71bed
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 30 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
langchain/embeddings/openai.py
View File

@ -94,7 +94,9 @@ class OpenAIEmbeddings(BaseModel, Embeddings):
from langchain.embeddings.openai import OpenAIEmbeddings from langchain.embeddings.openai import OpenAIEmbeddings
embeddings = OpenAIEmbeddings( embeddings = OpenAIEmbeddings(
deployment="your-embeddings-deployment-name", deployment="your-embeddings-deployment-name",
model="your-embeddings-model-name" model="your-embeddings-model-name",
api_base="https://your-endpoint.openai.azure.com/",
api_type="azure",
) )
text = "This is a test query." text = "This is a test query."
query_result = embeddings.embed_query(text) query_result = embeddings.embed_query(text)
@ -104,6 +106,11 @@ class OpenAIEmbeddings(BaseModel, Embeddings):
client: Any #: :meta private: client: Any #: :meta private:
model: str = "text-embedding-ada-002" model: str = "text-embedding-ada-002"
deployment: str = model # to support Azure OpenAI Service custom deployment names deployment: str = model # to support Azure OpenAI Service custom deployment names
openai_api_version: str = "2022-12-01"
# to support Azure OpenAI Service custom endpoints
openai_api_base: Optional[str] = None
# to support Azure OpenAI Service custom endpoints
openai_api_type: Optional[str] = None
embedding_ctx_length: int = 8191 embedding_ctx_length: int = 8191
openai_api_key: Optional[str] = None openai_api_key: Optional[str] = None
openai_organization: Optional[str] = None openai_organization: Optional[str] = None
@ -125,6 +132,23 @@ class OpenAIEmbeddings(BaseModel, Embeddings):
openai_api_key = get_from_dict_or_env( openai_api_key = get_from_dict_or_env(
values, "openai_api_key", "OPENAI_API_KEY" values, "openai_api_key", "OPENAI_API_KEY"
) )
openai_api_base = get_from_dict_or_env(
values,
"openai_api_base",
"OPENAI_API_BASE",
default="",
)
openai_api_type = get_from_dict_or_env(
values,
"openai_api_type",
"OPENAI_API_TYPE",
default="",
)
openai_api_version = get_from_dict_or_env(
values,
"openai_api_version",
"OPENAI_API_VERSION",
)
openai_organization = get_from_dict_or_env( openai_organization = get_from_dict_or_env(
values, values,
"openai_organization", "openai_organization",
@ -137,6 +161,11 @@ class OpenAIEmbeddings(BaseModel, Embeddings):
openai.api_key = openai_api_key openai.api_key = openai_api_key
if openai_organization: if openai_organization:
openai.organization = openai_organization openai.organization = openai_organization
if openai_api_base:
openai.api_base = openai_api_base
openai.api_version = openai_api_version
if openai_api_type:
openai.api_type = openai_api_type
values["client"] = openai.Embedding values["client"] = openai.Embedding
except ImportError: except ImportError:
raise ValueError( raise ValueError(