Spaces:
Sleeping
Sleeping
File size: 1,950 Bytes
686a009 |
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 |
import os
from typing import List
from dotenv import load_dotenv
from langchain.tools import tool
from logger.decorators import log_entry
from utils.currency_converter import CurrencyConverter
class CurrencyConverterTool:
"""
A wrapper class that defines a LangChain-compatible tool for converting currencies
using a custom CurrencyConverter service.
Attributes:
api_key (str): API key for the currency conversion service.
currency_service (CurrencyConverter): Instance of the currency conversion service.
currency_converter_tool_list (List): List of registered LangChain tools.
"""
def __init__(self):
"""
Initialize the CurrencyConverterTool with API credentials and tool setup.
"""
load_dotenv()
self.api_key = os.environ.get("EXCHANGE_RATE_API_KEY")
self.currency_service = CurrencyConverter(self.api_key)
self.currency_converter_tool_list = self._setup_tools()
def _setup_tools(self) -> List:
"""
Define and register the currency conversion tool.
Returns:
List: A list containing the currency conversion tool function.
"""
@tool
@log_entry
def convert_currency(amount: float, from_currency: str, to_currency: str) -> float:
"""
Convert an amount from one currency to another using real-time exchange rates.
Args:
amount (float): The amount of money to convert.
from_currency (str): The currency code to convert from (e.g., "USD").
to_currency (str): The currency code to convert to (e.g., "EUR").
Returns:
float: The equivalent amount in the target currency.
"""
print('Entered into convert_currency().')
return self.currency_service.convert(amount, from_currency, to_currency)
return [convert_currency]
|