function_calling_one_function.py

def function_calling_one_function(API_KEY, query):
    # Step 1: send the conversation and available functions to GPT
    functions = [
        {
            "name": "get_temperature",
            "description": "Get the temperature at a given location and date",
            "parameters": {
                "type": "object",
                "properties": {
                    "location": {
                        "type": "string",
                        "description": "Return the location cited in the query"
                    },
                    "date": {
                        "type": "string",
                        "description": "Return the date in in 'YYYY-MM-DD' format."
                    }
                 },
                },
                "required": ["location", "date"],
            }
    ]

    openai.api_key = API_KEY
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo-0613",
        messages=[{"role": "system", "content": query}],
        functions=functions,
        function_call="auto",  # auto is default, but we'll be explicit
    )
    response_message = response["choices"][0]["message"]
    return response_message