(AI) Function calling implementation question (PART 2)

[KB-WEB-DEVELOPMENT]

Hi

If you don't mind me asking:

I am carefully studying both pages:

https://docs.puter.com/AI/chat/#function-calling
https://platform.openai.com/docs/guides/function-calling#function-tool-example

Could you please show me some sample code how you would query my own SQL database table using

(1) the defined callable tool
(2) a certain number of functions which generate the SQL query and retrieve the data from my database table

Thanks in advance.

KBW

Originally posted by @kamibarut609-commits in #2107

[ProgrammerIn-wonderland]

we don't really know what API your SQL database uses so it's hard for us to really give an example unless you also provide which library your database would use.

Though, I would typically discourage you from letting AI write queries for SQL DB's

[KB-WEB-DEVELOPMENT]

@ProgrammerIn-wonderland - Thanks a lot.

Could you recommend a "database system" / "database management system" (puter.com function calling documentation uses the expression "external system") which I could call and query from within my own defined functions.

(A JSON schema is insufficient for what I am trying to build.)

Thanks.

KBW

[reynaldichernando]

hi @KB-WEB-DEVELOPMENT , thanks for opening an issue

from what i understand, you want to prompt the AI, and have it query your database via the function call
this is roughly how you can achieve that

1. create an actual function that queries your db, maybe say searchUserByName
2. specify this as the function call, plus including the logic to actually call the function
3. call puter.ai.chat with the options

i can give you an example snippet, but as @ProgrammerIn-wonderland we don't know what API your db uses
so you need to adjust it to your own use case

import { init } from "@heyputer/puter.js/src/init.cjs";
import { PrismaClient } from '@prisma/client';

const puter = init(process.env.puterAuthToken);
const prisma = new PrismaClient();

/**
 * Regular Function: Search User
 */
async function searchUserByName(name) {
    console.log(`Searching MySQL for: ${name}`);
    return await prisma.user.findMany({
        where: { name: { contains: name } },
    });
}

/**
 * Main AI Execution
 */
async function main() {
    const question = "Do we have any users named Alice?";

    // 1. Describe the tool to the AI
    const tools = [{
        type: "function",
        function: {
            name: "searchUserByName",
            description: "Search for a user in the database by name",
            parameters: {
                type: "object",
                properties: { name: { type: "string" } },
                required: ["name"]
            }
        }
    }];

    // 2. Initial Chat
    const response = await puter.ai.chat(question, { tools });
    const message = response.message;

    // 3. Handle the Function Call
    if (message.tool_calls) {
        const toolCall = message.tool_calls[0];
        
        if (toolCall.function.name === "searchUserByName") {
            // Parse arguments and call our regular function
            const { name } = JSON.parse(toolCall.function.arguments);
            const data = await searchUserByName(name);

            // 4. Submit result back for final answer
            const finalResponse = await puter.ai.chat([
                { role: "user", content: question },
                message,
                { role: "tool", tool_call_id: toolCall.id, content: JSON.stringify(data) }
            ]);

            console.log("AI Answer:", finalResponse.message.content);
        }
    } else {
        console.log("AI Answer:", message.content);
    }
}

main();

notes on above code

• it uses Prisma library for the example, its one of the famous ORM in js
• instead of letting the AI generate actual query, we only tell it specifically what it can do, which is the search user by name
• again as said by @ProgrammerIn-wonderland , we don't recommend letting AI query directly to your DB

on the "external system" mention:
you can let the AI interact with any systems, in this case the DB, because it's like with any code, you can do whatever you want with it, but having it defined as function call, the AI can now also access it

[KB-WEB-DEVELOPMENT]

@reynaldichernando - Thanks so much. Much appreciated.

That should get me going.

KBW