First_agent_template / healthcare_prompts.yaml
DocUA's picture
якась відносно робоча версія
ff6980a
from smolagents.tools import Tool
class HealthcareLLMVisualizerTool(Tool):
name = "healthcare_llm_visualizer"
description = "Creates interactive visualizations for analyzing LLM applications in Healthcare"
inputs = {
'data': {
'type': 'object',
'description': 'Data for visualization in format: {"items": [{"category": "name", "value": number}]}'
}
}
output_type = "string"
def forward(self, data):
"""Creates a visualization from the provided data"""
try:
# Create React component
chart_code = """
import React from 'react';
import { BarChart, Bar, XAxis, YAxis, CartesianGrid, Tooltip, Legend, ResponsiveContainer } from 'recharts';
const HealthcareLLMChart = () => {
const data = DATA_PLACEHOLDER;
return (
<div className="w-full max-w-4xl mx-auto p-4">
<h2 className="text-2xl font-bold mb-4">LLM Applications in Healthcare</h2>
<div className="h-96">
<ResponsiveContainer width="100%" height="100%">
<BarChart data={data.items}>
<CartesianGrid strokeDasharray="3 3" />
<XAxis dataKey="category" />
<YAxis />
<Tooltip />
<Legend />
<Bar dataKey="value" fill="#8884d8" />
</BarChart>
</ResponsiveContainer>
</div>
</div>
);
};
export default HealthcareLLMChart;
""".replace('DATA_PLACEHOLDER', str(data))
return chart_code
except Exception as e:
return f"Error creating visualization: {str(e)}"