File size: 1,647 Bytes
c5eab2c 3ad9743 c5eab2c 992a47c c5eab2c 992a47c c5eab2c 3ad9743 | 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 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 | '''
TODO tools
- web_search
Facultative
- wiki_search
- arxiv_search
'''
from smolagents import CodeAgent, HfApiModel, tool, DuckDuckGoSearchTool, MLXModel
@tool
def add(a:int, b:int) -> int:
"""
This tool returns the sum of two numbers.
Args:
a: first number
b: second number
"""
return a+b
@tool
def subtract(a:int, b:int) -> int:
"""
This tool returns the difference between two numbers.
Args:
a: first number
b: second number
"""
return a-b
@tool
def multiply(a:int, b:int) -> int:
"""
This tool multiplies two numbers.
Args:
a: first number
b: second number
"""
return a*b
@tool
def divide(a:int, b:int) -> float:
"""
This tool divides two numbers.
Args:
a: first number
b: second number
"""
if b==0: raise ValueError('Cannot divide by zero')
return a/b
@tool
def modulus(a:int, b:int) -> int:
"""
This tool returns the modulus of two numbers.
Args:
a: first number
b: second number
"""
return a%b
@tool
def rounder(a:float, n:int) -> float:
"""
This tool return a number rounded to a certain number of decimals.
Args:
a: number to be rounded
n: number of decimals to use when rounding the number
"""
return round(a,n)
def get_agent() -> CodeAgent:
search_tool = DuckDuckGoSearchTool()
model = MLXModel(
"mlx-community/Qwen2.5-Coder-32B-Instruct-4bit",
{
"temperature": 0.7,
"top_k": 20,
"top_p": 0.8,
"min_p": 0.05,
"num_ctx": 32768,
},
)
return CodeAgent(tools=[add, subtract, multiply, divide, modulus, rounder, search_tool], model=model) |