Function Calling (Tools)

from openai import OpenAI
import json
import os

client = OpenAI(
    base_url="https://api.neosantara.xyz/v1",
    api_key=os.environ["NEOSANTARA_API_KEY"]
)

tools = [
    {
        "type": "function",
        "function": {
            "name": "get_stock_price",
            "description": "Fetch real-time stock price by ticker symbol.",
            "parameters": {
                "type": "object",
                "properties": {
                    "ticker": {"type": "string", "description": "Stock ticker (e.g. BBCA)"}
                },
                "required": ["ticker"]
            }
        }
    }
]

# Step 1: Model evaluates request
response = client.chat.completions.create(
    model="deepseek-v4.1-flash",
    messages=[{"role": "user", "content": "What is the stock price of BBCA today?"}],
    tools=tools,
    tool_choice="auto"
)

message = response.choices[0].message

# Step 2: Check if model generated tool calls
if message.tool_calls:
    for tool_call in message.tool_calls:
        print(f"Function: {tool_call.function.name}")
        print(f"Arguments: {tool_call.function.arguments}")
        
        # Step 3: Execute local function
        tool_result = json.dumps({"ticker": "BBCA", "price": 9850, "currency": "IDR"})
        
        # Step 4: Supply result back to model
        final_response = client.chat.completions.create(
            model="deepseek-v4.1-flash",
            messages=[
                {"role": "user", "content": "What is the stock price of BBCA today?"},
                message,
                {
                    "role": "tool",
                    "tool_call_id": tool_call.id,
                    "content": tool_result
                }
            ]
        )
        print("Final Answer:", final_response.choices[0].message.content)

curl -X POST https://api.neosantara.xyz/v1/chat/completions \
  -H "Authorization: Bearer $NEOSANTARA_API_KEY" \
  -H "Content-Type: application/json" \
  -d '{
    "model": "deepseek-v4.1-flash",
    "messages": [{"role": "user", "content": "Check BBCA price"}],
    "tools": [
      {
        "type": "function",
        "function": {
          "name": "get_stock_price",
          "description": "Fetch stock price",
          "parameters": {
            "type": "object",
            "properties": {
              "ticker": {"type": "string"}
            },
            "required": ["ticker"]
          }
        }
      }
    ]
  }'