Phoenix21 commited on
Commit
31cd36e
·
verified ·
1 Parent(s): 6c91d6e

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +4 -2
pipeline.py CHANGED
@@ -166,14 +166,16 @@ def classify_query(query: str) -> str:
166
  query_input = QueryInput(query=query)
167
 
168
  # Quick pattern-based approach for 'Wellness'
169
- wellness_keywords = ["box breathing", "meditation", "yoga", "mindfulness", "breathing exercises"]
 
170
  if any(keyword in query_input.query.lower() for keyword in wellness_keywords):
171
  return "Wellness"
172
 
173
  # Use chain for everything else
174
  class_result = classification_chain.invoke({"query": query_input.query})
175
  print(class_result)
176
- classification = class_result.get("text", "").strip()
 
177
 
178
  return classification if classification != "" else "OutOfScope"
179
  except ValidationError as e:
 
166
  query_input = QueryInput(query=query)
167
 
168
  # Quick pattern-based approach for 'Wellness'
169
+ # wellness_keywords = ["box breathing", "meditation", "yoga", "mindfulness", "breathing exercises"]
170
+ wellness_keywords=[]
171
  if any(keyword in query_input.query.lower() for keyword in wellness_keywords):
172
  return "Wellness"
173
 
174
  # Use chain for everything else
175
  class_result = classification_chain.invoke({"query": query_input.query})
176
  print(class_result)
177
+ # classification = class_result.get("text", "").strip()
178
+ classification=class_result
179
 
180
  return classification if classification != "" else "OutOfScope"
181
  except ValidationError as e: