Add _is_chat_input() method to Vertex and ChatVertex classes
This commit is contained in:
parent
4dbe66ae06
commit
20ffe250a9
2 changed files with 8 additions and 2 deletions
|
|
@ -618,6 +618,9 @@ class Vertex:
|
||||||
self.steps_ran = []
|
self.steps_ran = []
|
||||||
self._build_params()
|
self._build_params()
|
||||||
|
|
||||||
|
def _is_chat_input(self):
|
||||||
|
return False
|
||||||
|
|
||||||
def build_inactive(self):
|
def build_inactive(self):
|
||||||
# Just set the results to None
|
# Just set the results to None
|
||||||
self._built = True
|
self._built = True
|
||||||
|
|
@ -640,7 +643,7 @@ class Vertex:
|
||||||
return self.get_requester_result(requester)
|
return self.get_requester_result(requester)
|
||||||
self._reset()
|
self._reset()
|
||||||
|
|
||||||
if self.is_input and inputs is not None:
|
if self._is_chat_input() and inputs is not None:
|
||||||
self.update_raw_params(inputs)
|
self.update_raw_params(inputs)
|
||||||
|
|
||||||
# Run steps
|
# Run steps
|
||||||
|
|
|
||||||
|
|
@ -6,7 +6,7 @@ import yaml
|
||||||
from langchain_core.messages import AIMessage
|
from langchain_core.messages import AIMessage
|
||||||
from loguru import logger
|
from loguru import logger
|
||||||
|
|
||||||
from langflow.graph.schema import INPUT_FIELD_NAME
|
from langflow.graph.schema import INPUT_FIELD_NAME, InterfaceComponentTypes
|
||||||
from langflow.graph.utils import UnbuiltObject, flatten_list, serialize_field
|
from langflow.graph.utils import UnbuiltObject, flatten_list, serialize_field
|
||||||
from langflow.graph.vertex.base import StatefulVertex, StatelessVertex
|
from langflow.graph.vertex.base import StatefulVertex, StatelessVertex
|
||||||
from langflow.interface.utils import extract_input_variables_from_prompt
|
from langflow.interface.utils import extract_input_variables_from_prompt
|
||||||
|
|
@ -455,6 +455,9 @@ class ChatVertex(StatelessVertex):
|
||||||
async for _ in self.stream():
|
async for _ in self.stream():
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
def _is_chat_input(self):
|
||||||
|
return self.vertex_type == InterfaceComponentTypes.ChatInput and self.is_input
|
||||||
|
|
||||||
|
|
||||||
class RoutingVertex(StatelessVertex):
|
class RoutingVertex(StatelessVertex):
|
||||||
def __init__(self, data: Dict, graph):
|
def __init__(self, data: Dict, graph):
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue