Spaces:
Sleeping
Sleeping
File size: 710 Bytes
b814c5a 7b414ac b814c5a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 | import ollama
def classify_intent(question):
prompt = f"""
Classify the user message.
Possible labels:
- chat
- dataset_question
Rules:
chat:
- greetings
- small talk
- thanks
- introductions
- general conversation
dataset_question:
- any question requiring dataset analysis
- statistics
- charts
- correlations
- rows
- columns
- sales
- business insights
Message:
{question}
Return ONLY one label.
"""
response = ollama.chat(
model="qwen2.5:3b",
messages=[
{
"role": "user",
"content": prompt
}
]
)
return response["message"]["content"].strip().lower() |