mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-09 08:08:32 +00:00
19 lines
416 B
Python
19 lines
416 B
Python
from typing import Optional
|
|
|
|
|
|
class CacheHandler:
|
|
"""Callback handler for tool usage."""
|
|
|
|
_cache: dict = {}
|
|
|
|
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}")
|