Mady1 commited on
Commit
29d384c
·
verified ·
1 Parent(s): 8c5c24b

Rename app.py to app_2.py

Browse files
Files changed (2) hide show
  1. app.py +0 -69
  2. app_2.py +79 -0
app.py DELETED
@@ -1,69 +0,0 @@
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()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
app_2.py ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # app.py
2
+
3
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
4
+ from tools.final_answer import FinalAnswerTool
5
+ import datetime
6
+ import pytz
7
+ import yaml
8
+ import gradio as gr
9
+
10
+ # ✅✅✅ ---- Natural language time tool ----
11
+ @tool
12
+ def get_current_time_in_timezone(timezone: str) -> str:
13
+ """
14
+ A tool that fetches the current local time in a specified timezone.
15
+ Args:
16
+ timezone: A string representing a valid timezone (e.g., 'Europe/London').
17
+ """
18
+ try:
19
+ tz = pytz.timezone(timezone)
20
+ local_time = datetime.datetime.now(tz).strftime("%A, %B %d, %Y %I:%M %p")
21
+ return f"The current local time in {timezone} is {local_time}."
22
+ except Exception as e:
23
+ return f"Sorry, I couldn’t find the time for {timezone}: {str(e)}"
24
+
25
+ # ✅✅✅ ---- Final Answer Tool ----
26
+ final_answer = FinalAnswerTool()
27
+
28
+ # ✅✅✅ ---- Hugging Face Model ----
29
+ model = HfApiModel(
30
+ max_tokens=2048,
31
+ temperature=0.5,
32
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
33
+ )
34
+
35
+ # ✅✅✅ ---- Load optional tool from Hub ----
36
+ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
37
+
38
+ # ✅✅✅ ---- System prompt template ----
39
+ # This will ensure the agent keeps full sentences!
40
+ prompt_templates = {
41
+ "system": """You are a helpful agent that thinks step by step.
42
+ You must always return your final answer as a clear, complete sentence in natural language.
43
+ If you call tools, use their output exactly and do not strip details.
44
+ When you have the final information, pass it using the FinalAnswerTool.
45
+ """
46
+ }
47
+
48
+ # ✅✅✅ ---- Build the CodeAgent ----
49
+ agent = CodeAgent(
50
+ model=model,
51
+ tools=[
52
+ get_current_time_in_timezone,
53
+ DuckDuckGoSearchTool(),
54
+ image_generation_tool,
55
+ final_answer
56
+ ],
57
+ max_steps=6,
58
+ verbosity_level=1,
59
+ prompt_templates=prompt_templates
60
+ )
61
+
62
+ # ✅✅✅ ---- Gradio UI ----
63
+ def agentic_chat(user_input):
64
+ result = agent.run(user_input)
65
+ if hasattr(result, 'final_answer'):
66
+ return result.final_answer
67
+ else:
68
+ return str(result)
69
+
70
+ iface = gr.Interface(
71
+ fn=agentic_chat,
72
+ inputs="text",
73
+ outputs="text",
74
+ title="Agentic AI Example",
75
+ description="Ask me anything! I will reason step-by-step and provide clear final answers."
76
+ )
77
+
78
+ if __name__ == "__main__":
79
+ iface.launch()