|
|
@ -123,7 +123,12 @@ class OpenAIEmbeddings(BaseModel, Embeddings):
|
|
|
|
retry_max_seconds: int = 20
|
|
|
|
retry_max_seconds: int = 20
|
|
|
|
"""Max number of seconds to wait between retries"""
|
|
|
|
"""Max number of seconds to wait between retries"""
|
|
|
|
http_client: Union[Any, None] = None
|
|
|
|
http_client: Union[Any, None] = None
|
|
|
|
"""Optional httpx.Client."""
|
|
|
|
"""Optional httpx.Client. Only used for sync invocations. Must specify
|
|
|
|
|
|
|
|
http_async_client as well if you'd like a custom client for async invocations.
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
http_async_client: Union[Any, None] = None
|
|
|
|
|
|
|
|
"""Optional httpx.AsyncClient. Only used for async invocations. Must specify
|
|
|
|
|
|
|
|
http_client as well if you'd like a custom client for sync invocations."""
|
|
|
|
|
|
|
|
|
|
|
|
class Config:
|
|
|
|
class Config:
|
|
|
|
"""Configuration for this pydantic object."""
|
|
|
|
"""Configuration for this pydantic object."""
|
|
|
@ -218,12 +223,17 @@ class OpenAIEmbeddings(BaseModel, Embeddings):
|
|
|
|
"max_retries": values["max_retries"],
|
|
|
|
"max_retries": values["max_retries"],
|
|
|
|
"default_headers": values["default_headers"],
|
|
|
|
"default_headers": values["default_headers"],
|
|
|
|
"default_query": values["default_query"],
|
|
|
|
"default_query": values["default_query"],
|
|
|
|
"http_client": values["http_client"],
|
|
|
|
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if not values.get("client"):
|
|
|
|
if not values.get("client"):
|
|
|
|
values["client"] = openai.OpenAI(**client_params).embeddings
|
|
|
|
sync_specific = {"http_client": values["http_client"]}
|
|
|
|
|
|
|
|
values["client"] = openai.OpenAI(
|
|
|
|
|
|
|
|
**client_params, **sync_specific
|
|
|
|
|
|
|
|
).embeddings
|
|
|
|
if not values.get("async_client"):
|
|
|
|
if not values.get("async_client"):
|
|
|
|
values["async_client"] = openai.AsyncOpenAI(**client_params).embeddings
|
|
|
|
async_specific = {"http_client": values["http_async_client"]}
|
|
|
|
|
|
|
|
values["async_client"] = openai.AsyncOpenAI(
|
|
|
|
|
|
|
|
**client_params, **async_specific
|
|
|
|
|
|
|
|
).embeddings
|
|
|
|
return values
|
|
|
|
return values
|
|
|
|
|
|
|
|
|
|
|
|
@property
|
|
|
|
@property
|
|
|
|