File size: 2,287 Bytes
c79af1e
9b5b26a
 
c79af1e
4a0a932
9b5b26a
4a0a932
c79af1e
4a0a932
 
9b5b26a
4a0a932
 
9b5b26a
4a0a932
9b5b26a
4a0a932
 
abce7c5
 
 
 
 
4a0a932
 
c79af1e
9b5b26a
 
 
 
 
 
 
 
 
 
 
 
 
 
8c01ffb
c79af1e
4a0a932
 
 
 
13d500a
8c01ffb
c79af1e
4a0a932
c79af1e
 
8c01ffb
c79af1e
8c01ffb
8fe992b
c79af1e
4a0a932
 
8fe992b
 
77eb7a8
 
 
 
 
 
abce7c5
 
77eb7a8
 
e20383d
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
72
from smolagents import CodeAgent, tool, OpenAIServerModel, FinalAnswerTool
import datetime
import pytz
import yfinance as yf
import os
from Gradio_UI import GradioUI

# Fetch the API key from the environment variable (if needed)
API_KEY = os.getenv("KEY")

@tool
def get_currency_rate(currency_pair: str) -> str:
    """Fetches the current exchange rate for a given currency pair (e.g., 'USD-EUR').
    Args:
        currency_pair: The currency pair to query, formatted as 'BASE-CURRENCY'.
    """
    try:
        base, target = currency_pair.split("-")
        # Use yfinance to get the exchange rate
        ticker = f"{base}{target}=X"  # Format for yfinance (e.g., "USDEUR=X")
        data = yf.Ticker(ticker)
        rate = data.history(period="1d")["Close"].iloc[-1]
        return f"Current {currency_pair} rate: {rate}"
    except Exception as e:
        return f"Error fetching rate: {str(e)}"

@tool
def get_current_time_in_timezone(timezone: str) -> str:
    """A tool that fetches the current local time in a specified timezone.
    Args:
        timezone: A string representing a valid timezone (e.g., 'America/New_York').
    """
    try:
        # Create timezone object
        tz = pytz.timezone(timezone)
        # Get current time in that timezone
        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
        return f"The current local time in {timezone} is: {local_time}"
    except Exception as e:
        return f"Error fetching time for timezone '{timezone}': {str(e)}"

# Set up the model
model = OpenAIServerModel(
    model_id="deepseek-chat",
    api_base="https://api.deepseek.com",
    api_key=API_KEY
)

# Load tools
final_answer = FinalAnswerTool()
currency_tool = get_currency_rate
time_tool = get_current_time_in_timezone

# Create the agent
agent = CodeAgent(
    model=model,
    tools=[currency_tool, time_tool, final_answer],  # Include your tools here
    max_steps=3,
    verbosity_level=1
)

# Define example usage instructions
example_usage = """
### Example Usage
1. **Currency Rate**: Ask for the exchange rate of a currency pair.
   - Example: "What is the current exchange rate for USD to JPY?"
   - Example: "Get the EUR to GBP rate.
"""


# Launch the Gradio UI with the example usage
CustomGradioUI(agent).launch()