juanmaguitar commited on
Commit
fca9c3d
·
1 Parent(s): 0ad5e93

minimal AI agent confg

Browse files
Files changed (1) hide show
  1. app.py +8 -9
app.py CHANGED
@@ -133,16 +133,15 @@ web_search = DuckDuckGoSearchTool()
133
  # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
134
 
135
  model = HfApiModel(
136
- max_tokens=1024,
137
  temperature=0.5,
138
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
139
- # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
140
- model_id='mistralai/Mistral-7B-Instruct',
141
- token=os.getenv("HUGGINGFACEHUB_API_TOKEN"), # Ensure this is set
142
  custom_role_conversions=None,
143
- top_p=0.9,
144
- frequency_penalty=0.0,
145
- presence_penalty=0.0,
146
  )
147
 
148
  # Add error handling wrapper for the model
@@ -174,10 +173,10 @@ agent = CodeAgent(
174
  model=model,
175
  tools=[
176
  final_answer,
177
- get_current_time_in_timezone,
178
  image_generation_tool,
179
  quick_research,
180
- get_weather,
181
  wordpress_post,
182
  blog_generator,
183
  visit_webpage,
 
133
  # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
134
 
135
  model = HfApiModel(
136
+ max_tokens=2096,
137
  temperature=0.5,
138
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
139
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
140
+ # token=os.getenv("HUGGINGFACEHUB_API_TOKEN"), # Ensure this is set
 
141
  custom_role_conversions=None,
142
+ # top_p=0.9,
143
+ # frequency_penalty=0.0,
144
+ # presence_penalty=0.0,
145
  )
146
 
147
  # Add error handling wrapper for the model
 
173
  model=model,
174
  tools=[
175
  final_answer,
176
+ # get_current_time_in_timezone,
177
  image_generation_tool,
178
  quick_research,
179
+ # get_weather,
180
  wordpress_post,
181
  blog_generator,
182
  visit_webpage,