yetessam commited on
Commit
74773e5
·
verified ·
1 Parent(s): cf8dfff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -8,7 +8,7 @@ import pytz # Had to give it permission in Code agent
8
 
9
  from ui.contentagentui import ContentAgentUI
10
 
11
- from smolagents import CodeAgent, InferenceClientModel, PromptTemplates
12
 
13
  from checks.status_check import resolve_endpoint, is_endpoint_healthy
14
 
@@ -18,9 +18,7 @@ from prompts.prompts import load_prompts
18
 
19
  def initialize_agent(endpoint_uri: str):
20
 
21
- # Load prompts
22
-
23
- prompt_templates = PromptTemplates( system_prompt= load_prompts())
24
  combined_tools = load_tools() ## from the tools folder (don't remove final answer)
25
 
26
  """Initialize and configure the CodeAgent"""
@@ -37,6 +35,7 @@ def initialize_agent(endpoint_uri: str):
37
  max_steps=6,
38
  verbosity_level=5,
39
  name="content_agent",
 
40
  description="Evaluates whether text is polite or impolite.",
41
  prompt_templates=prompt_templates, # override prompts
42
  additional_authorized_imports=["pytz"],
 
8
 
9
  from ui.contentagentui import ContentAgentUI
10
 
11
+ from smolagents import CodeAgent, InferenceClientModel
12
 
13
  from checks.status_check import resolve_endpoint, is_endpoint_healthy
14
 
 
18
 
19
  def initialize_agent(endpoint_uri: str):
20
 
21
+
 
 
22
  combined_tools = load_tools() ## from the tools folder (don't remove final answer)
23
 
24
  """Initialize and configure the CodeAgent"""
 
35
  max_steps=6,
36
  verbosity_level=5,
37
  name="content_agent",
38
+ system_prompt= load_prompts()
39
  description="Evaluates whether text is polite or impolite.",
40
  prompt_templates=prompt_templates, # override prompts
41
  additional_authorized_imports=["pytz"],