|
|
""" |
|
|
Knowledge Base Tools for RAG |
|
|
""" |
|
|
|
|
|
from crewai.tools import BaseTool |
|
|
from utils.knowledge_base import KnowledgeBase |
|
|
from typing import List, Dict |
|
|
|
|
|
class SearchKnowledgeTool(BaseTool): |
|
|
name: str = "search_spiritual_texts" |
|
|
description: str = "Search spiritual and self-help texts for relevant wisdom" |
|
|
|
|
|
def __init__(self): |
|
|
super().__init__() |
|
|
self.kb = KnowledgeBase() |
|
|
|
|
|
def _run(self, query: str, k: int = 5) -> List[Dict]: |
|
|
"""Search knowledge base""" |
|
|
if not self.kb.is_initialized(): |
|
|
return [{ |
|
|
"text": "Wisdom comes from understanding ourselves.", |
|
|
"source": "General Wisdom", |
|
|
"score": 1.0 |
|
|
}] |
|
|
|
|
|
results = self.kb.search(query, k=k) |
|
|
return results |
|
|
|
|
|
class ExtractWisdomTool(BaseTool): |
|
|
name: str = "extract_relevant_wisdom" |
|
|
description: str = "Extract most relevant wisdom for user's situation" |
|
|
|
|
|
def _run(self, search_results: List[Dict], user_context: Dict) -> List[Dict]: |
|
|
"""Filter and rank wisdom based on relevance""" |
|
|
emotion = user_context.get("primary_emotion", "neutral") |
|
|
concerns = user_context.get("concerns", []) |
|
|
|
|
|
|
|
|
scored_results = [] |
|
|
for result in search_results: |
|
|
score = result["score"] |
|
|
|
|
|
|
|
|
if emotion.lower() in result["text"].lower(): |
|
|
score *= 1.5 |
|
|
|
|
|
|
|
|
for concern in concerns: |
|
|
if concern.lower() in result["text"].lower(): |
|
|
score *= 1.3 |
|
|
|
|
|
result["relevance_score"] = score |
|
|
scored_results.append(result) |
|
|
|
|
|
|
|
|
scored_results.sort(key=lambda x: x["relevance_score"], reverse=True) |
|
|
|
|
|
return scored_results[:3] |
|
|
|
|
|
class SuggestPracticeTool(BaseTool): |
|
|
name: str = "suggest_meditation_practice" |
|
|
description: str = "Suggest appropriate meditation or practice" |
|
|
|
|
|
def _run(self, emotional_state: str, cultural_context: str = None) -> Dict: |
|
|
"""Suggest practice based on emotional state""" |
|
|
practices = { |
|
|
"anxiety": { |
|
|
"name": "Box Breathing Technique", |
|
|
"description": "A powerful technique used by Navy SEALs to calm anxiety", |
|
|
"steps": [ |
|
|
"Sit comfortably with back straight", |
|
|
"Exhale all air from your lungs", |
|
|
"Inhale through nose for 4 counts", |
|
|
"Hold breath for 4 counts", |
|
|
"Exhale through mouth for 4 counts", |
|
|
"Hold empty for 4 counts", |
|
|
"Repeat 4-8 times" |
|
|
], |
|
|
"benefits": "Activates parasympathetic nervous system, reduces cortisol", |
|
|
"duration": "5-10 minutes", |
|
|
"origin": "Modern breathwork" |
|
|
}, |
|
|
"sadness": { |
|
|
"name": "Metta (Loving-Kindness) Meditation", |
|
|
"description": "Ancient Buddhist practice to cultivate compassion", |
|
|
"steps": [ |
|
|
"Sit comfortably, close your eyes", |
|
|
"Place hand on heart", |
|
|
"Begin with self: 'May I be happy, may I be peaceful'", |
|
|
"Extend to loved ones", |
|
|
"Include neutral people", |
|
|
"Embrace difficult people", |
|
|
"Radiate to all beings" |
|
|
], |
|
|
"benefits": "Increases self-compassion, reduces depression", |
|
|
"duration": "15-20 minutes", |
|
|
"origin": "Buddhist tradition" |
|
|
}, |
|
|
"stress": { |
|
|
"name": "Progressive Muscle Relaxation", |
|
|
"description": "Systematic tension and release technique", |
|
|
"steps": [ |
|
|
"Lie down comfortably", |
|
|
"Start with toes - tense for 5 seconds", |
|
|
"Release suddenly, notice relaxation", |
|
|
"Move up through each muscle group", |
|
|
"Face and scalp last", |
|
|
"Rest in full body relaxation" |
|
|
], |
|
|
"benefits": "Reduces physical tension, improves sleep", |
|
|
"duration": "15-20 minutes", |
|
|
"origin": "Dr. Edmund Jacobson, 1920s" |
|
|
} |
|
|
} |
|
|
|
|
|
default = { |
|
|
"name": "Mindful Breathing", |
|
|
"description": "Foundation of all meditation practices", |
|
|
"steps": [ |
|
|
"Sit comfortably", |
|
|
"Follow natural breath", |
|
|
"Count breaths 1-10", |
|
|
"Start again when distracted", |
|
|
"No judgment, just awareness" |
|
|
], |
|
|
"benefits": "Calms mind, improves focus", |
|
|
"duration": "5-15 minutes", |
|
|
"origin": "Universal practice" |
|
|
} |
|
|
|
|
|
return practices.get(emotional_state.lower(), default) |
