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

lint

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-04 22:10:30 -03:00
commit e7b4c2f4b2
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/langflow/interface/agents/base.py
View file

@ -18,7 +18,8 @@ class AgentCreator(LangChainTypeCreator):
self.type_dict = loading.AGENT_TO_CLASS self.type_dict = loading.AGENT_TO_CLASS
# Add JsonAgent to the list of agents # Add JsonAgent to the list of agents
for name, agent in CUSTOM_AGENTS.items(): for name, agent in CUSTOM_AGENTS.items():
self.type_dict[name] = agent # TODO: validate AgentType
self.type_dict[name] = agent # type: ignore
return self.type_dict return self.type_dict
def get_signature(self, name: str) -> Optional[Dict]: def get_signature(self, name: str) -> Optional[Dict]:

4
src/backend/langflow/interface/prompts/base.py
View file

@ -1,4 +1,4 @@
from typing import Dict, List, Optional from typing import Dict, List, Optional, Type
from langchain import prompts from langchain import prompts
from langchain.prompts import loading from langchain.prompts import loading
@ -15,7 +15,7 @@ class PromptCreator(LangChainTypeCreator):
type_name: str = "prompts" type_name: str = "prompts"
@property @property
def frontend_node_class(self) -> str: def frontend_node_class(self) -> Type[PromptFrontendNode]:
return PromptFrontendNode return PromptFrontendNode
@property @property