🐛 fix(vertex/base.py): add artifacts attribute to Vertex class to store additional data

🐛 fix(vertex/base.py): update instantiation logic to handle tuple result from loading.instantiate_class()
🐛 fix(loading.py): update return value of instantiate_prompt() to return a tuple of prompt and format_kwargs
The Vertex class now has a new attribute called artifacts, which is a dictionary used to store additional data related to the vertex. The instantiation logic in the Vertex class has been updated to handle the case where loading.instantiate_class() returns a tuple containing the built object and additional artifacts. The loading.instantiate_prompt() function now returns a tuple containing the prompt and format_kwargs. These changes fix issues related to storing and handling additional data in the vertex and loading modules.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-06-28 17:26:57 -03:00
commit 03349bf999
2 changed files with 9 additions and 2 deletions

View file

@ -124,7 +124,7 @@ def instantiate_prompt(node_type, class_object, params):
if format_kwargs:
prompt = prompt.partial(**format_kwargs)
return prompt
return prompt, format_kwargs
def instantiate_tool(node_type, class_object, params):