Update import statement in Cohere.py

This commit is contained in:
carlosrcoelho 2024-01-10 11:39:11 -03:00
commit a24e0314bc

View file

@ -1,6 +1,6 @@
from langflow import CustomComponent from langflow import CustomComponent
from langchain.llms import BaseLanguageModel from langchain_core.language_models.base import BaseLanguageModel
from typing import Optional from typing import Optional
class CohereComponent(CustomComponent): class CohereComponent(CustomComponent):
@ -13,7 +13,7 @@ class CohereComponent(CustomComponent):
"cohere_api_key": { "cohere_api_key": {
"display_name": "Cohere API Key", "display_name": "Cohere API Key",
"type": "password", "type": "password",
"show": True "password": True
}, },
"max_tokens": { "max_tokens": {
"display_name": "Max Tokens", "display_name": "Max Tokens",