pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-25 15:17:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

lint

Browse Source
This commit is contained in:
Gabriel Almeida
2023-04-04 22:10:30 -03:00
parent 41f05b2e85
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
# Add JsonAgent to the list of agents
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
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.prompts import loading
@ -15,7 +15,7 @@ class PromptCreator(LangChainTypeCreator):
type_name: str = "prompts"
@property
def frontend_node_class(self) -> str:
def frontend_node_class(self) -> Type[PromptFrontendNode]:
return PromptFrontendNode
@property