mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-08 15:48:29 +00:00
21 lines
457 B
Python
21 lines
457 B
Python
from typing import Optional
|
|
|
|
from pydantic import PrivateAttr
|
|
|
|
|
|
class CacheHandler:
|
|
"""Callback handler for tool usage."""
|
|
|
|
_cache: PrivateAttr = {}
|
|
|
|
def __init__(self):
|
|
self._cache = {}
|
|
|
|
def add(self, tool, input, output):
|
|
input = input.strip()
|
|
self._cache[f"{tool}-{input}"] = output
|
|
|
|
def read(self, tool, input) -> Optional[str]:
|
|
input = input.strip()
|
|
return self._cache.get(f"{tool}-{input}")
|