Update core_creator/intent_parser.py
Browse files
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 |
-
|
34 |
-
|
35 |
-
|
36 |
|
37 |
-
|
38 |
-
|
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)
|