mirror of
https://github.com/hwchase17/langchain.git
synced 2025-07-17 10:13:29 +00:00
use openai_organization as argument (#2566)
Added support for passing the openai_organization as an argument, as it was only supported by the environment variable but openai_api_key was supported by both environment variables and arguments. `ChatOpenAI(temperature=0, model_name="gpt-4", openai_api_key="sk-****", openai_organization="org-****")`
This commit is contained in:
parent
544cc7f395
commit
f7f118e021
@ -31,6 +31,7 @@ class OpenAIModerationChain(Chain):
|
||||
input_key: str = "input" #: :meta private:
|
||||
output_key: str = "output" #: :meta private:
|
||||
openai_api_key: Optional[str] = None
|
||||
openai_organization: Optional[str] = None
|
||||
|
||||
@root_validator()
|
||||
def validate_environment(cls, values: Dict) -> Dict:
|
||||
|
@ -44,6 +44,7 @@ class AzureChatOpenAI(ChatOpenAI):
|
||||
openai_api_base: str = ""
|
||||
openai_api_version: str = ""
|
||||
openai_api_key: str = ""
|
||||
openai_organization: str = ""
|
||||
|
||||
@root_validator()
|
||||
def validate_environment(cls, values: Dict) -> Dict:
|
||||
|
@ -115,6 +115,7 @@ class ChatOpenAI(BaseChatModel):
|
||||
model_kwargs: Dict[str, Any] = Field(default_factory=dict)
|
||||
"""Holds any model parameters valid for `create` call not explicitly specified."""
|
||||
openai_api_key: Optional[str] = None
|
||||
openai_organization: Optional[str] = None
|
||||
request_timeout: int = 60
|
||||
"""Timeout in seconds for the OpenAPI request."""
|
||||
max_retries: int = 6
|
||||
|
@ -98,6 +98,7 @@ class OpenAIEmbeddings(BaseModel, Embeddings):
|
||||
query_model_name: str = "text-embedding-ada-002"
|
||||
embedding_ctx_length: int = 8191
|
||||
openai_api_key: Optional[str] = None
|
||||
openai_organization: Optional[str] = None
|
||||
chunk_size: int = 1000
|
||||
"""Maximum number of texts to embed in each batch"""
|
||||
max_retries: int = 6
|
||||
|
@ -151,6 +151,7 @@ class BaseOpenAI(BaseLLM):
|
||||
model_kwargs: Dict[str, Any] = Field(default_factory=dict)
|
||||
"""Holds any model parameters valid for `create` call not explicitly specified."""
|
||||
openai_api_key: Optional[str] = None
|
||||
openai_organization: Optional[str] = None
|
||||
batch_size: int = 20
|
||||
"""Batch size to use when passing multiple documents to generate."""
|
||||
request_timeout: Optional[Union[float, Tuple[float, float]]] = None
|
||||
|
Loading…
Reference in New Issue
Block a user