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)
|