<script> import { getContext, createEventDispatcher, onMount, tick } from 'svelte'; const i18n = getContext('i18n'); import CodeEditor from '$lib/components/common/CodeEditor.svelte'; import { goto } from '$app/navigation'; import ConfirmDialog from '$lib/components/common/ConfirmDialog.svelte'; import Badge from '$lib/components/common/Badge.svelte'; import ChevronLeft from '$lib/components/icons/ChevronLeft.svelte'; import Tooltip from '$lib/components/common/Tooltip.svelte'; const dispatch = createEventDispatcher(); let formElement = null; let loading = false; let showConfirm = false; export let edit = false; export let clone = false; export let id = ''; export let name = ''; export let meta = { description: '' }; export let content = ''; let _content = ''; $: if (content) { updateContent(); } const updateContent = () => { _content = content; }; $: if (name && !edit && !clone) { id = name.replace(/\s+/g, '_').toLowerCase(); } let codeEditor; let boilerplate = `import os import requests from datetime import datetime class Tools: def __init__(self): pass # Add your custom tools using pure Python code here, make sure to add type hints # Use Sphinx-style docstrings to document your tools, they will be used for generating tools specifications # Please refer to function_calling_filter_pipeline.py file from pipelines project for an example def get_user_name_and_email_and_id(self, __user__: dict = {}) -> str: """ Get the user name, Email and ID from the user object. """ # Do not include :param for __user__ in the docstring as it should not be shown in the tool's specification # The session user object will be passed as a parameter when the function is called print(__user__) result = "" if "name" in __user__: result += f"User: {__user__['name']}" if "id" in __user__: result += f" (ID: {__user__['id']})" if "email" in __user__: result += f" (Email: {__user__['email']})" if result == "": result = "User: Unknown" return result def get_current_time(self) -> str: """ Get the current time in a more human-readable format. :return: The current time. """ now = datetime.now() current_time = now.strftime("%I:%M:%S %p") # Using 12-hour format with AM/PM current_date = now.strftime( "%A, %B %d, %Y" ) # Full weekday, month name, day, and year return f"Current Date and Time = {current_date}, {current_time}" def calculator(self, equation: str) -> str: """ Calculate the result of an equation. :param equation: The equation to calculate. """ # Avoid using eval in production code # https://nedbatchelder.com/blog/201206/eval_really_is_dangerous.html try: result = eval(equation) return f"{equation} = {result}" except Exception as e: print(e) return "Invalid equation" def get_current_weather(self, city: str) -> str: """ Get the current weather for a given city. :param city: The name of the city to get the weather for. :return: The current weather information or an error message. """ api_key = os.getenv("OPENWEATHER_API_KEY") if not api_key: return ( "API key is not set in the environment variable 'OPENWEATHER_API_KEY'." ) base_url = "http://api.openweathermap.org/data/2.5/weather" params = { "q": city, "appid": api_key, "units": "metric", # Optional: Use 'imperial' for Fahrenheit } try: response = requests.get(base_url, params=params) response.raise_for_status() # Raise HTTPError for bad responses (4xx and 5xx) data = response.json() if data.get("cod") != 200: return f"Error fetching weather data: {data.get('message')}" weather_description = data["weather"][0]["description"] temperature = data["main"]["temp"] humidity = data["main"]["humidity"] wind_speed = data["wind"]["speed"] return f"Weather in {city}: {temperature}°C" except requests.RequestException as e: return f"Error fetching weather data: {str(e)}" `; const saveHandler = async () => { loading = true; dispatch('save', { id, name, meta, content }); }; const submitHandler = async () => { if (codeEditor) { content = _content; await tick(); const res = await codeEditor.formatPythonCodeHandler(); await tick(); content = _content; await tick(); if (res) { console.log('Code formatted successfully'); saveHandler(); } } }; </script> <div class=" flex flex-col justify-between w-full overflow-y-auto h-full"> <div class="mx-auto w-full md:px-0 h-full"> <form bind:this={formElement} class=" flex flex-col max-h-[100dvh] h-full" on:submit|preventDefault={() => { if (edit) { submitHandler(); } else { showConfirm = true; } }} > <div class="flex flex-col flex-1 overflow-auto h-0"> <div class="w-full mb-2 flex flex-col gap-0.5"> <div class="flex w-full items-center"> <div class=" flex-shrink-0 mr-2"> <Tooltip content={$i18n.t('Back')}> <button class="w-full text-left text-sm py-1.5 px-1 rounded-lg dark:text-gray-300 dark:hover:text-white hover:bg-black/5 dark:hover:bg-gray-850" on:click={() => { goto('/workspace/tools'); }} type="button" > <ChevronLeft strokeWidth="2.5" /> </button> </Tooltip> </div> <div class="flex-1"> <input class="w-full text-2xl font-medium bg-transparent outline-none" type="text" placeholder={$i18n.t('Toolkit Name (e.g. My ToolKit)')} bind:value={name} required /> </div> <div> <Badge type="muted" content={$i18n.t('Tool')} /> </div> </div> <div class=" flex gap-2 px-1"> {#if edit} <div class="text-sm text-gray-500 flex-shrink-0"> {id} </div> {:else} <input class="w-full text-sm disabled:text-gray-500 bg-transparent outline-none" type="text" placeholder={$i18n.t('Toolkit ID (e.g. my_toolkit)')} bind:value={id} required disabled={edit} /> {/if} <input class="w-full text-sm bg-transparent outline-none" type="text" placeholder={$i18n.t( 'Toolkit Description (e.g. A toolkit for performing various operations)' )} bind:value={meta.description} required /> </div> </div> <div class="mb-2 flex-1 overflow-auto h-0 rounded-lg"> <CodeEditor bind:this={codeEditor} value={content} {boilerplate} lang="python" on:change={(e) => { _content = e.detail.value; }} on:save={() => { if (formElement) { formElement.requestSubmit(); } }} /> </div> <div class="pb-3 flex justify-between"> <div class="flex-1 pr-3"> <div class="text-xs text-gray-500 line-clamp-2"> <span class=" font-semibold dark:text-gray-200">{$i18n.t('Warning:')}</span> {$i18n.t('Tools are a function calling system with arbitrary code execution')} <br />— <span class=" font-medium dark:text-gray-400" >{$i18n.t(`don't install random tools from sources you don't trust.`)}</span > </div> </div> <button class="px-3.5 py-1.5 text-sm font-medium bg-black hover:bg-gray-900 text-white dark:bg-white dark:text-black dark:hover:bg-gray-100 transition rounded-full" type="submit" > {$i18n.t('Save')} </button> </div> </div> </form> </div> </div> <ConfirmDialog bind:show={showConfirm} on:confirm={() => { submitHandler(); }} > <div class="text-sm text-gray-500"> <div class=" bg-yellow-500/20 text-yellow-700 dark:text-yellow-200 rounded-lg px-4 py-3"> <div>{$i18n.t('Please carefully review the following warnings:')}</div> <ul class=" mt-1 list-disc pl-4 text-xs"> <li> {$i18n.t('Tools have a function calling system that allows arbitrary code execution.')} </li> <li>{$i18n.t('Do not install tools from sources you do not fully trust.')}</li> </ul> </div> <div class="my-3"> {$i18n.t( 'I acknowledge that I have read and I understand the implications of my action. I am aware of the risks associated with executing arbitrary code and I have verified the trustworthiness of the source.' )} </div> </div> </ConfirmDialog>