mgbam commited on
Commit
6607bd7
·
verified ·
1 Parent(s): d787996

Update core_creator/intent_parser.py

Browse files
Files changed (1) hide show
  1. core_creator/intent_parser.py +6 -6
core_creator/intent_parser.py CHANGED
@@ -30,13 +30,13 @@ else:
30
 
31
  def classify_robot_idea(user_input: str) -> str:
32
  prompt = f"""
33
- Classify this user idea into one of the following categories:
34
- {', '.join(INTENT_CATEGORIES)}.
35
- Only return the category word. If none fits, return 'creative'.
36
 
37
- Idea: {user_input}
38
- Category:
39
- """
40
 
41
  if USE_GEMINI:
42
  response = gemini_model.generate_content(prompt)
 
30
 
31
  def classify_robot_idea(user_input: str) -> str:
32
  prompt = f"""
33
+ Classify this user idea into one of the following categories:
34
+ {', '.join(INTENT_CATEGORIES)}.
35
+ Only return the category word. If none fits, return 'creative'.
36
 
37
+ Idea: {user_input}
38
+ Category:
39
+ """
40
 
41
  if USE_GEMINI:
42
  response = gemini_model.generate_content(prompt)