|
|
|
|
|
from utils.knowledge_base import KnowledgeBase |
|
|
from pydantic import PrivateAttr |
|
|
from crewai.tools import BaseTool |
|
|
|
|
|
|
|
|
|
|
|
class SearchKnowledgeTool(BaseTool): |
|
|
name: str = "search_knowledge" |
|
|
description: str = "Search self-help or spiritual wisdom." |
|
|
model_config = {"arbitrary_types_allowed": True} |
|
|
_kp: KnowledgeBase = PrivateAttr() |
|
|
def __init__(self, config=None): |
|
|
super().__init__() |
|
|
self._kb = KnowledgeBase(config) |
|
|
def _run(self, query: str, k: int = 5): |
|
|
return self.kb.search(query, k=k) if self.kb.is_initialized() else \ |
|
|
[{"text": "General wisdom", "score": 1.0}] |
|
|
|
|
|
class ExtractWisdomTool(BaseTool): |
|
|
name: str = "extract_wisdom" |
|
|
description: str = "Extract most relevant wisdom for a given query." |
|
|
model_config = {"arbitrary_types_allowed": True} |
|
|
def __init__(self, config=None): |
|
|
super().__init__() |
|
|
def _run(self, search_results: list, user_context: dict): |
|
|
return search_results[:3] |
|
|
|
|
|
class SuggestPracticesTool(BaseTool): |
|
|
name: str = "suggest_practices" |
|
|
description: str = "Recommend meditations or self-care practices." |
|
|
model_config = {"arbitrary_types_allowed": True} |
|
|
def __init__(self, config=None): |
|
|
super().__init__() |
|
|
def _run(self, emotional_state: str, cultural_context: str = None): |
|
|
return {"name": "Mindful Breathing", "description": "Focus on your breath to calm the mind."} |
|
|
|
|
|
class KnowledgeTools: |
|
|
def __init__(self, config=None): |
|
|
self.search_knowledge = SearchKnowledgeTool(config) |
|
|
self.extract_wisdom = ExtractWisdomTool(config) |
|
|
self.suggest_practices = SuggestPracticesTool(config) |
