ekhatskevich commited on
Commit
eebcf01
·
1 Parent(s): b77bd3a

fix: address

Browse files
Files changed (2) hide show
  1. app.py +5 -7
  2. prompts.yaml +3 -3
app.py CHANGED
@@ -1,6 +1,5 @@
1
  from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
  import geoip2.database
3
- import requests
4
  import datetime
5
  import pytz
6
  import yaml
@@ -9,17 +8,16 @@ from tools.final_answer import FinalAnswerTool
9
  from Gradio_UI import GradioUI
10
 
11
  @tool
12
- def show_internet_location()-> str:
13
- """A tool that shows user's current location based on IP address.
 
 
14
  """
15
  try:
16
- response = requests.get("https://api64.ipify.org?format=json")
17
- public_ip = response.json()["ip"]
18
-
19
  reader = geoip2.database.Reader('geo_cities/GeoLite2-City.mmdb')
20
  response = reader.city(public_ip)
21
 
22
- internet_location = f"Based on your IP address {public_ip}, you are in {response.country.name}, {response.city.name}"
23
  return internet_location
24
  except Exception as e:
25
  return f"Error fetching the response for query internet location: {str(e)}"
 
1
  from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
  import geoip2.database
 
3
  import datetime
4
  import pytz
5
  import yaml
 
8
  from Gradio_UI import GradioUI
9
 
10
  @tool
11
+ def show_internet_location(public_ip)-> str:
12
+ """A tool that shows location based on IP address.
13
+ Args:
14
+ public_ip: IP address
15
  """
16
  try:
 
 
 
17
  reader = geoip2.database.Reader('geo_cities/GeoLite2-City.mmdb')
18
  response = reader.city(public_ip)
19
 
20
+ internet_location = f"IP address {public_ip}, is from {response.country.name}, {response.city.name}"
21
  return internet_location
22
  except Exception as e:
23
  return f"Error fetching the response for query internet location: {str(e)}"
prompts.yaml CHANGED
@@ -39,12 +39,12 @@
39
  ```<end_code>
40
 
41
  ---
42
- Task: "What is my current location?"
43
 
44
- Thought: I will use python code to retrieve user's current location based on IP address and then return the final answer using the `final_answer` tool
45
  Code:
46
  ```py
47
- result = show_internet_location()
48
  final_answer(result)
49
  ```<end_code>
50
 
 
39
  ```<end_code>
40
 
41
  ---
42
+ Task: "What is the location of the IP address: 8.8.8.8?"
43
 
44
+ Thought: I will use python code to retrieve the location based on IP address provided in the request and then return the final answer using the `final_answer` tool
45
  Code:
46
  ```py
47
+ result = show_internet_location(public_ip)
48
  final_answer(result)
49
  ```<end_code>
50