Tool Calling

import openai
import json

client = openai.OpenAI(
    base_url="https://api.deepinfra.com/v1/openai",
    api_key="$DEEPINFRA_TOKEN",
)

import OpenAI from "openai";

const client = new OpenAI({
    baseURL: "https://api.deepinfra.com/v1/openai",
    apiKey: "$DEEPINFRA_TOKEN",
});

def get_current_weather(location):
    """Get the current weather in a given location"""
    if "tokyo" in location.lower():
        return json.dumps({"location": "Tokyo", "temperature": "75"})
    elif "san francisco" in location.lower():
        return json.dumps({"location": "San Francisco", "temperature": "60"})
    elif "paris" in location.lower():
        return json.dumps({"location": "Paris", "temperature": "70"})
    else:
        return json.dumps({"location": location, "temperature": "unknown"})

async function get_current_weather(location) {
  if (location.toLowerCase().includes("tokyo")) {
    return JSON.stringify({"location": "Tokyo", "temperature": "75"});
  } else if (location.toLowerCase().includes("san francisco")) {
    return JSON.stringify({"location": "San Francisco", "temperature": "60"});
  } else if (location.toLowerCase().includes("paris")) {
    return JSON.stringify({"location": "Paris", "temperature": "70"});
  } else {
    return JSON.stringify({"location": location, "temperature": "unknown"});
  }
}

tools = [{
    "type": "function",
    "function": {
        "name": "get_current_weather",
        "description": "Get the current weather in a given location",
        "parameters": {
            "type": "object",
            "properties": {
                "location": {
                    "type": "string",
                    "description": "The city and state, e.g. San Francisco, CA"
                }
            },
            "required": ["location"]
        },
    }
}]

messages = [{"role": "user", "content": "What is the weather in San Francisco?"}]

response = client.chat.completions.create(
    model="deepseek-ai/DeepSeek-V3",
    messages=messages,
    tools=tools,
    tool_choice="auto",
)
tool_calls = response.choices[0].message.tool_calls
for tool_call in tool_calls:
    print(tool_call.model_dump())

const tools = [{
  "type": "function",
  "function": {
    "name": "get_current_weather",
    "description": "Get the current weather in a given location",
    "parameters": {
      "type": "object",
      "properties": {
        "location": {
          "type": "string",
          "description": "The city and state, e.g. San Francisco, CA"
        }
      },
      "required": ["location"]
    },
  }
}];

const messages = [{"role": "user", "content": "What is the weather in San Francisco?"}];

const response = await client.chat.completions.create({
  model: "deepseek-ai/DeepSeek-V3",
  messages: messages,
  tools: tools,
  tool_choice: "auto",
});

const tool_calls = response.choices[0].message.tool_calls;
for (const tool_call of tool_calls) {
  console.log(tool_call);
}

# Extend conversation with assistant's reply
messages.append(response.choices[0].message)

for tool_call in tool_calls:
    function_name = tool_call.function.name
    if function_name == "get_current_weather":
        function_args = json.loads(tool_call.function.arguments)
        function_response = get_current_weather(
            location=function_args.get("location")
        )

    messages.append({
        "tool_call_id": tool_call.id,
        "role": "tool",
        "content": function_response,
    })

# Get a new response from the model with function results
second_response = client.chat.completions.create(
    model="deepseek-ai/DeepSeek-V3",
    messages=messages,
    tools=tools,
    tool_choice="auto",
)

print(second_response.choices[0].message.content)

// Extend conversation with assistant's reply
messages.push(response.choices[0].message);

for (const tool_call of tool_calls) {
  const function_name = tool_call.function.name;

  if (function_name == "get_current_weather") {
    const function_args = JSON.parse(tool_call.function.arguments);
    const function_response = await get_current_weather(function_args.location);

    messages.push({
      "tool_call_id": tool_call.id,
      "role": "tool",
      "content": function_response,
    });
  }
}

const second_response = await client.chat.completions.create({
  model: "deepseek-ai/DeepSeek-V3",
  messages: messages,
  tools: tools,
  tool_choice: "auto",
});

console.log(second_response.choices[0].message.content);