gar/function-call-example.py at main · gulliantonio/gar · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import vertexai
from google.cloud import aiplatform
import vertexai.preview
from vertexai.generative_models import GenerativeModel
from vertexai.generative_models import (
    Content,
    FunctionDeclaration,
    GenerativeModel,
    Part,
    Tool,
)
import requests

get_exchange_rate_func = FunctionDeclaration(
    name="get_exchange_rate",
    description="Get the exchange rate for currencies between countries",
    parameters={
    "type": "object",
    "properties": {
        "currency_date": {
            "type": "string",
            "description": "A date that must always be in YYYY-MM-DD format or the value 'latest' if a time period is not specified"
        },
        "currency_from": {
            "type": "string",
            "description": "The currency to convert from in ISO 4217 format"
        },
        "currency_to": {
            "type": "string",
            "description": "The currency to convert to in ISO 4217 format"
        }
    },
         "required": [
            "currency_from",
            "currency_date",
      ]
  },
)

def init():
  vertexai.init(project='octo-gar')

def call_model():
  model = GenerativeModel("gemini-1.0-pro")

  exchange_rate_tool = Tool(
    function_declarations=[get_exchange_rate_func],
  )

  prompt = """What is the exchange rate from Australian dollars to Swedish krona?
  How much is 500 Australian dollars worth in Swedish krona?"""

  response = model.generate_content(
    prompt,
    tools=[exchange_rate_tool],
  )

  print(response.candidates[0].content)

  params = {}
  for key, value in response.candidates[0].content.parts[0].function_call.args.items():
    params[key[9:]] = value
  params

  url = f"https://api.frankfurter.app/{params['date']}"
  api_response = requests.get(url, params=params)
  print (api_response.text)


init()
call_model()