jamesbright commited on
Commit
4163334
·
verified ·
1 Parent(s): 874666c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -6
app.py CHANGED
@@ -3,7 +3,8 @@ 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',
@@ -26,15 +27,15 @@ class ProjectScopingTool(Tool):
26
  "description": "The industry or field related to the project (e.g., 'finance', 'healthcare').",
27
  },
28
  "budget_min": {
29
- "type": "number",
30
  "description": "The minimum budget allocated for the project.",
31
  },
32
  "budget_max": {
33
- "type": "number",
34
  "description": "The maximum budget allocated for the project.",
35
  },
36
  "timeline_months": {
37
- "type": "number",
38
  "description": "The expected duration of the project in months.",
39
  },
40
  }
@@ -93,11 +94,11 @@ class CostEstimationTool(Tool):
93
 
94
  inputs = {
95
  "architecture_size": {
96
- "type": "number",
97
  "description": "The estimated complexity of the architecture on a scale from 1 to 10.",
98
  },
99
  "timeline_months": {
100
- "type": "number",
101
  "description": "The project duration in months.",
102
  }
103
  }
 
3
  from smolagents import CodeAgent, tool, Tool, FinalAnswerTool, HfApiModel, load_tool
4
 
5
  final_answer = FinalAnswerTool()
6
+ model = HfApiModel
7
+ (
8
  max_tokens=2096,
9
  temperature=0.5,
10
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 
27
  "description": "The industry or field related to the project (e.g., 'finance', 'healthcare').",
28
  },
29
  "budget_min": {
30
+ "type": "Integer",
31
  "description": "The minimum budget allocated for the project.",
32
  },
33
  "budget_max": {
34
+ "type": "Integer",
35
  "description": "The maximum budget allocated for the project.",
36
  },
37
  "timeline_months": {
38
+ "type": "Integer",
39
  "description": "The expected duration of the project in months.",
40
  },
41
  }
 
94
 
95
  inputs = {
96
  "architecture_size": {
97
+ "type": "Integer",
98
  "description": "The estimated complexity of the architecture on a scale from 1 to 10.",
99
  },
100
  "timeline_months": {
101
+ "type": "Integer",
102
  "description": "The project duration in months.",
103
  }
104
  }