Superbrida commited on
Commit
db8cd81
·
verified ·
1 Parent(s): bb81171

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +47 -47
app.py CHANGED
@@ -1,69 +1,69 @@
1
- from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
2
  import datetime
3
  import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
- from Gradio_UI import GradioUI
8
 
 
9
 
10
- # Di seguito un esempio di uno strumento che non fa nulla. Stupiscici con la tua creatività!
11
- @attrezzo
12
- def my_custom_tool ( arg1: str , arg2: int )-> str : importante specificare il tipo di ritorno
13
- #Mantieni questo formato per la descrizione / argomenti / descrizione argomenti ma sentiti libero di modificare lo strumento
14
- """Uno strumento che non fa ancora nulla
15
- Argomenti:
16
- arg1: il primo argomento
17
- arg2: il secondo argomento
18
  """
19
- ritorna "Quale magia costruirai?"
20
 
21
- @attrezzo
22
- def get_current_time_in_timezone ( fuso orario: str ) -> str :
23
- """Uno strumento che recupera l'ora locale corrente in un fuso orario specificato.
24
- Argomenti:
25
- fuso orario: una stringa che rappresenta un fuso orario valido (ad esempio, 'America/New_York').
26
  """
27
- Tentativo :
28
- # Crea oggetto fuso orario
29
- tz = pytz.timezone(fuso orario)
30
- # Ottieni l'ora corrente in quel fuso orario
31
- ora_locale = datetime.datetime.now(tz).strftime( "%Y-%m-%d %H:%M:%S" )
32
- restituisci f"L'ora locale corrente in {timezone} è: {local_time} "
33
- eccetto Eccezione come e:
34
- restituisci f"Errore durante il recupero dell'ora per il fuso orario ' {timezone} ': { str (e)} "
35
 
36
 
37
- risposta_finale = StrumentoRispostaFinale()
38
 
39
- # Se l'agente non risponde, il modello è sovraccarico, utilizzare un altro modello o il seguente Hugging Face Endpoint che contenga anche il codificatore qwen2.5:
40
- # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
41
 
42
- modello = HfApiModel(
43
- numero massimo di token = 2096 ,
44
- temperatura= 0,5 ,
45
- model_id= 'Qwen/Qwen2.5-Coder-32B-Instruct' , # è possibile che questo modello sia sovraccarico
46
- custom_role_conversions= Nessuno ,
47
  )
48
 
49
 
50
- # Strumento di importazione dall'hub
51
- image_generation_tool = load_tool( "agenti-corso/testo-in-immagine" , trust_remote_code= True )
52
 
53
- con open ( "prompts.yaml" , 'r' ) come flusso:
54
- prompt_templates = yaml.safe_load(flusso)
55
 
56
- agente = CodeAgent(
57
- modello=modello,
58
- tools=[final_answer], ## aggiungi i tuoi strumenti qui (non rimuovere la risposta finale)
59
- max_passi= 6 ,
60
- livello_di_verbosità= 1 ,
61
- grammatica= Nessuna ,
62
- planning_interval= Nessuno ,
63
- nome= Nessuno ,
64
- descrizione= Nessuno ,
65
- prompt_templates=template_prompt
66
  )
67
 
68
 
69
- GradioUI(agente).launch()
 
1
+ from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
  import datetime
3
  import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
 
7
 
8
+ from Gradio_UI import GradioUI
9
 
10
+ # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
+ @tool
12
+ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
13
+ #Keep this format for the description / args / args description but feel free to modify the tool
14
+ """A tool that does nothing yet
15
+ Args:
16
+ arg1: the first argument
17
+ arg2: the second argument
18
  """
19
+ return "What magic will you build ?"
20
 
21
+ @tool
22
+ def get_current_time_in_timezone(timezone: str) -> str:
23
+ """A tool that fetches the current local time in a specified timezone.
24
+ Args:
25
+ timezone: A string representing a valid timezone (e.g., 'America/New_York').
26
  """
27
+ try:
28
+ # Create timezone object
29
+ tz = pytz.timezone(timezone)
30
+ # Get current time in that timezone
31
+ local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
32
+ return f"The current local time in {timezone} is: {local_time}"
33
+ except Exception as e:
34
+ return f"Error fetching time for timezone '{timezone}': {str(e)}"
35
 
36
 
37
+ final_answer = FinalAnswerTool()
38
 
39
+ # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
40
+ # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
41
 
42
+ model = HfApiModel(
43
+ max_tokens=2096,
44
+ temperature=0.5,
45
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
46
+ custom_role_conversions=None,
47
  )
48
 
49
 
50
+ # Import tool from Hub
51
+ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
52
 
53
+ with open("prompts.yaml", 'r') as stream:
54
+ prompt_templates = yaml.safe_load(stream)
55
 
56
+ agent = CodeAgent(
57
+ model=model,
58
+ tools=[final_answer], ## add your tools here (don't remove final answer)
59
+ max_steps=6,
60
+ verbosity_level=1,
61
+ grammar=None,
62
+ planning_interval=None,
63
+ name=None,
64
+ description=None,
65
+ prompt_templates=prompt_templates
66
  )
67
 
68
 
69
+ GradioUI(agent).launch()