Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -1,11 +1,14 @@
|
|
1 |
import gradio as gr
|
2 |
import json
|
3 |
-
from smolagents import CodeAgent, tool, Tool
|
4 |
-
from smolagents.models
|
5 |
-
import HfApiModel
|
6 |
|
7 |
-
|
8 |
-
model = HfApiModel(
|
|
|
|
|
|
|
|
|
|
|
9 |
|
10 |
|
11 |
class ProjectScopingTool(Tool):
|
@@ -28,7 +31,8 @@ class ProjectScopingTool(Tool):
|
|
28 |
self.project_details = {
|
29 |
"type": project_type,
|
30 |
"domain": domain,
|
31 |
-
"budget_range": {"min"
|
|
|
32 |
"timeline": {"months": timeline_months}
|
33 |
}
|
34 |
return json.dumps(self.project_details, indent=2)
|
@@ -121,7 +125,17 @@ tools = [
|
|
121 |
meeting_tool.generate_meeting_agenda
|
122 |
]
|
123 |
|
124 |
-
agent = CodeAgent(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
125 |
|
126 |
# Define Gradio Interface
|
127 |
with gr.Blocks() as demo:
|
|
|
1 |
import gradio as gr
|
2 |
import json
|
3 |
+
from smolagents import CodeAgent, tool, Tool, FinalAnswerTool, HfApiModel, load_tool
|
|
|
|
|
4 |
|
5 |
+
final_answer = FinalAnswerTool()
|
6 |
+
model = HfApiModel(
|
7 |
+
max_tokens=2096,
|
8 |
+
temperature=0.5,
|
9 |
+
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
|
10 |
+
custom_role_conversions=None,
|
11 |
+
)
|
12 |
|
13 |
|
14 |
class ProjectScopingTool(Tool):
|
|
|
31 |
self.project_details = {
|
32 |
"type": project_type,
|
33 |
"domain": domain,
|
34 |
+
"budget_range": {"min"
|
35 |
+
: budget_min, "max": budget_max},
|
36 |
"timeline": {"months": timeline_months}
|
37 |
}
|
38 |
return json.dumps(self.project_details, indent=2)
|
|
|
125 |
meeting_tool.generate_meeting_agenda
|
126 |
]
|
127 |
|
128 |
+
agent = CodeAgent(
|
129 |
+
model=model,
|
130 |
+
tools=[final_answer, tools], # add your tools here (don't remove final_answer)
|
131 |
+
max_steps=6,
|
132 |
+
verbosity_level=1,
|
133 |
+
grammar=None,
|
134 |
+
planning_interval=None,
|
135 |
+
name=None,
|
136 |
+
description=None,
|
137 |
+
prompt_templates=prompt_templates
|
138 |
+
)
|
139 |
|
140 |
# Define Gradio Interface
|
141 |
with gr.Blocks() as demo:
|