Update CohereEmbeddingsComponent and ChatOpenAIComponent configurations

This commit is contained in:
anovazzi1 2024-01-16 11:40:35 -03:00
commit d4b0706580
2 changed files with 4 additions and 1 deletions

View file

@ -20,12 +20,14 @@ class CohereEmbeddingsComponent(CustomComponent):
def build(
self,
request_timeout: Optional[float] = None,
cohere_api_key: Optional[str] = None,
cohere_api_key: str = None,
max_retries: Optional[int] = None,
model: str = "embed-english-v2.0",
truncate: Optional[str] = None,
user_agent: Optional[str] = "langchain",
) -> CohereEmbeddings:
return CohereEmbeddings(
max_retries=max_retries,
user_agent=user_agent,
request_timeout=request_timeout,
cohere_api_key=cohere_api_key,

View file

@ -51,6 +51,7 @@ class ChatOpenAIComponent(CustomComponent):
"field_type": "str",
"advanced": False,
"required": False,
"password": True,
},
"temperature": {
"display_name": "Temperature",