Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-18 09:25:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

cohere[patch]: Allow overriding of the base URL in Cohere Client (#19766)

Browse Source 
This PR adds the ability for a user to override the base API url for the
Cohere client for embeddings and chat llm.
This commit is contained in:
BeatrixCohere 2024-03-29 21:22:30 +00:00 committed by GitHub
parent 1252ccce6f
commit bd02b83acd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/partners/cohere/langchain_cohere/embeddings.py
View File

@ -45,6 +45,9 @@ class CohereEmbeddings(BaseModel, Embeddings):
user_agent: str = "langchain"
"""Identifier for the application making the request."""
base_url: Optional[str] = None
"""Override the default Cohere API URL."""
class Config:
"""Configuration for this pydantic object."""
@ -64,11 +67,13 @@ class CohereEmbeddings(BaseModel, Embeddings):
cohere_api_key,
timeout=request_timeout,
client_name=client_name,
base_url=values["base_url"],
)
values["async_client"] = cohere.AsyncClient(
cohere_api_key,
timeout=request_timeout,
client_name=client_name,
base_url=values["base_url"],
)
return values

5
libs/partners/cohere/langchain_cohere/llms.py
View File

@ -69,6 +69,9 @@ class BaseCohere(Serializable):
user_agent: str = "langchain"
"""Identifier for the application making the request."""
base_url: Optional[str] = None
"""Override the default Cohere API URL."""
@root_validator()
def validate_environment(cls, values: Dict) -> Dict:
"""Validate that api key and python package exists in environment."""
@ -79,10 +82,12 @@ class BaseCohere(Serializable):
values["client"] = cohere.Client(
api_key=values["cohere_api_key"].get_secret_value(),
client_name=client_name,
base_url=values["base_url"],
)
values["async_client"] = cohere.AsyncClient(
api_key=values["cohere_api_key"].get_secret_value(),
client_name=client_name,
base_url=values["base_url"],
)
return values