Phoenix21 commited on
Commit
f1772af
·
verified ·
1 Parent(s): 7cf22bd

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +7 -7
pipeline.py CHANGED
@@ -36,13 +36,13 @@ from chain.tailor_chain_wellnessBrand import get_tailor_chain_wellnessBrand
36
 
37
  from mistralai import Mistral
38
 
39
- from smolagents import (
40
- CodeAgent,
41
- DuckDuckGoSearchTool,
42
- HfApiModel,
43
- ToolCallingAgent,
44
- VisitWebpageTool,
45
- )
46
 
47
  from chain.prompts import selfharm_prompt, frustration_prompt, ethical_conflict_prompt, classification_prompt, refusal_prompt, tailor_prompt, cleaner_prompt
48
 
 
36
 
37
  from mistralai import Mistral
38
 
39
+ # from smolagents import (
40
+ # CodeAgent,
41
+ # DuckDuckGoSearchTool,
42
+ # HfApiModel,
43
+ # ToolCallingAgent,
44
+ # VisitWebpageTool,
45
+ # )
46
 
47
  from chain.prompts import selfharm_prompt, frustration_prompt, ethical_conflict_prompt, classification_prompt, refusal_prompt, tailor_prompt, cleaner_prompt
48