|
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool |
|
import datetime |
|
import requests |
|
import pytz |
|
import yaml |
|
from tools.final_answer import FinalAnswerTool |
|
|
|
from Gradio_UI import GradioUI |
|
|
|
|
|
@tool |
|
def blackjack_strategy(player_cards: list[str], dealer_card:str)-> str: |
|
"""A tool that determines optimal Blackjack strategy based on player's cards and dealer's face-up card. |
|
Args: |
|
player_cards: A list of strings representing the player's cards. |
|
Each card is represented as two character string (ex: 'AH' for Ace of hearts, '10S' for 10 of Spades). |
|
dealer_card: A string representing delaler's face-up card |
|
|
|
Returns: Recommended action ("Hit", "Stand", or "Double Down"). |
|
""" |
|
def card_value(card: str) -> int: |
|
"""Converts a card string into its numeric value for Blackjack. |
|
Args: |
|
card: A string representing a playing card, where the first part is the rank |
|
(2-10, J, Q, K, A) and the last character is the suit (H, D, C, S). |
|
|
|
Returns: |
|
int: The numeric value of the card: |
|
- A (Ace): 11 |
|
- J, Q, K (Face cards): 10 |
|
- Numeric cards: Their respective integer value |
|
""" |
|
if card[0] == 'A': return 11 |
|
elif card[0] in 'KQJ': return 10 |
|
else: return int(card[:-1]) |
|
|
|
player_values = [card_value(card) for card in player_cards] |
|
player_total = sum(player_values) |
|
dealer_total = card_value(dealer_card) |
|
|
|
if 11 in player_values and player_total > 21: |
|
player_total -= 10 |
|
|
|
if player_total >= 17: |
|
return 'Stand' |
|
elif player_total >= 13 and player_total <= 16: |
|
if dealer_total >= 7: |
|
return 'Hit' |
|
else: |
|
return 'Stand' |
|
elif player_total == 12: |
|
if dealer_total >= 4 and dealer_total <= 6: |
|
return 'Stand' |
|
else: |
|
return 'Hit' |
|
elif player_total <= 11: |
|
|
|
if len(player_cards) == 2 and (player_total in [9,10,11]): |
|
if dealer_total >= 2 and dealer_total <= 9: |
|
return 'Double Down' |
|
else: |
|
return 'Hit' |
|
else: |
|
return 'Hit' |
|
|
|
@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: |
|
|
|
tz = pytz.timezone(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)}" |
|
|
|
|
|
final_answer = FinalAnswerTool() |
|
|
|
|
|
|
|
|
|
model = HfApiModel( |
|
max_tokens=2096, |
|
temperature=0.5, |
|
model_id='Qwen/Qwen2.5-Coder-32B-Instruct', |
|
custom_role_conversions=None, |
|
) |
|
|
|
|
|
|
|
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True) |
|
|
|
with open("prompts.yaml", 'r') as stream: |
|
prompt_templates = yaml.safe_load(stream) |
|
|
|
agent = CodeAgent( |
|
model=model, |
|
tools=[final_answer, blackjack_strategy], |
|
max_steps=6, |
|
verbosity_level=1, |
|
grammar=None, |
|
planning_interval=None, |
|
name=None, |
|
description=None, |
|
prompt_templates=prompt_templates |
|
) |
|
|
|
|
|
GradioUI(agent).launch() |