diff --git a/src/backend/langflow/interface/initialize/llm.py b/src/backend/langflow/interface/initialize/llm.py index 2b9fa3b2e..eaed04b77 100644 --- a/src/backend/langflow/interface/initialize/llm.py +++ b/src/backend/langflow/interface/initialize/llm.py @@ -1,6 +1,6 @@ def initialize_vertexai(class_object, params): if credentials_path := params.get("credentials"): - from google.oauth2 import service_account + from google.oauth2 import service_account # type: ignore credentials_object = service_account.Credentials.from_service_account_file( filename=credentials_path diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index b5a901068..a48ae28ec 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -96,6 +96,7 @@ def instantiate_llm(node_type, class_object, params: Dict): if node_type == "VertexAI": return initialize_vertexai(class_object=class_object, params=params) + def instantiate_memory(node_type, class_object, params): try: return class_object(**params) @@ -125,9 +126,6 @@ def instantiate_retriever(node_type, class_object, params): return class_object(**params) - - - def instantiate_chains(node_type, class_object: Type[Chain], params: Dict): if "retriever" in params and hasattr(params["retriever"], "as_retriever"): params["retriever"] = params["retriever"].as_retriever()