🚧 chore(types.py): add OutputParserVertex class to support output parsing functionality

The OutputParserVertex class is added to the types.py file. This class extends the Vertex class and is used to represent a vertex in the graph that performs output parsing. It takes a data parameter of type Dict in its constructor and calls the super constructor with the data and a base_type of "output_parsers". This change is made to support the implementation of output parsing functionality in the graph.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-06-29 12:56:50 -03:00
commit c6d8d62ffd

View file

@ -194,3 +194,8 @@ class PromptVertex(Vertex):
self._build()
return self._built_object
class OutputParserVertex(Vertex):
def __init__(self, data: Dict):
super().__init__(data, base_type="output_parsers")