jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update CohereEmbeddingsComponent and ChatOpenAIComponent configurations

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/components/embeddings/CohereEmbeddings.py
View file

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

1
src/backend/langflow/components/llms/ChatOpenAI.py
View file

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