griptape
griptape copied to clipboard
Update StructureVisualizer to be able to identify what model/prompt_driver is being used
- [X] I have read and agree to the contributing guidelines.
It would be incredibly useful to be able to visualize where your data is going - for example, what data is going to go to gpt-4o vs llama3.1.
Perhaps we could add the ability for StructureVisualizer to do this for Agents as well?
for example, for this code:
agent = Agent(
stream=True,
prompt_driver=OpenAiChatPromptDriver(model="gpt-4o"),
tools=[
WebScraperTool(off_prompt=True),
QueryTool(prompt_driver=OllamaPromptDriver(model="llama3.1")),
],
)
we could display something like:
This should also work for Workflows & Pipelines so you know what drivers various tasks are using.
For anyone interested: the easiest way to implement this would likely be to convert the Structure to a dictionary, and then build a graph from the dictionary fields.