Skip to content

Commit

Permalink
docs(cache): add a standalone page in the documentation
Browse files Browse the repository at this point in the history
Ref: #44
Signed-off-by: Tomas Dvorak <toomas2d@gmail.com>
  • Loading branch information
Tomas2D committed Oct 5, 2024
1 parent 810a080 commit 9b60f8a
Show file tree
Hide file tree
Showing 12 changed files with 529 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ The Bee Agent Framework makes it easy to build scalable agent-based workflows wi
- 💾 **Memory**: Multiple [strategies](https://github.com/i-am-bee/bee-agent-framework/blob/main/docs/overview.md#memory) to optimize token spend.
- ⏸️ **Serialization** Handle complex agentic workflows and easily pause/resume them [without losing state](https://github.com/i-am-bee/bee-agent-framework/blob/main/docs/overview.md#serializer).
- 🔍 **Traceability**: Get full visibility of your agent’s inner workings, [log](https://github.com/i-am-bee/bee-agent-framework/blob/main/docs/overview.md#logger) all running events, and use our MLflow integration (coming soon) to debug performance.
- 🎛️ **Production-level** control with [caching](https://github.com/i-am-bee/bee-agent-framework/blob/main/docs/overview.md#cache) and [error handling](https://github.com/i-am-bee/bee-agent-framework/blob/main/docs/overview.md#errors).
- 🎛️ **Production-level** control with [caching](./docs/cache.md) and [error handling](https://github.com/i-am-bee/bee-agent-framework/blob/main/docs/overview.md#errors).
- 🚧 (Coming soon) **API**: Configure and deploy your agents with a production-hardened API.
- 🚧 (Coming soon) **Chat UI**: Serve your agent to users in a delightful GUI with built-in transparency, explainability, and user controls.
- ... more on our [Roadmap](#roadmap)
Expand Down Expand Up @@ -139,7 +139,7 @@ The source directory (`src`) provides numerous modules that one can use.
| **template** | Prompt Templating system based on `Mustache` with various improvements\_. |
| **memory** | Various types of memories to use with agent. |
| [**tools**](./docs/tools.md) | Tools that an agent can use. |
| **cache** | Preset of different caching approaches that can be used together with tools. |
| [**cache**](./docs/cache.md) | Preset of different caching approaches that can be used together with tools. |
| **errors** | Base framework error classes used by each module. |
| **adapters** | Concrete implementations of given modules for different environments. |
| **logger** | Core component for logging all actions within the framework. |
Expand Down
343 changes: 343 additions & 0 deletions docs/cache.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,343 @@
# Cache

> [!TIP]
>
> Location within the framework `bee-agent-framework/cache`.
Caching is a process used to temporarily store copies of data or computations in a cache (a storage location) to facilitate faster access upon future requests. The primary purpose of caching is to improve the efficiency and performance of systems by reducing the need to repeatedly fetch or compute the same data from a slower or more resource-intensive source.

## Usage

### Capabilities showcase

<!-- embedme examples/cache/unconstrainedCache.ts -->

```ts
import { UnconstrainedCache } from "bee-agent-framework/cache/unconstrainedCache";

const cache = new UnconstrainedCache();

// Save
await cache.set("a", 1);
await cache.set("b", 2);

// Read
const result = await cache.get("a");
console.log(result); // 1

// Meta
console.log(cache.enabled); // true
console.log(await cache.has("a")); // true
console.log(await cache.has("b")); // true
console.log(await cache.has("c")); // false
console.log(await cache.size()); // 2

// Delete
await cache.delete("a");
console.log(await cache.has("a")); // false

// Clear
await cache.clear();
console.log(await cache.size()); // 0
```

_Source: [examples/cache/unconstrainedCache.ts](/examples/cache/unconstrainedCache.ts)_

### Caching function output + intermediate steps

<!-- embedme examples/cache/unconstrainedCacheFunction.ts -->

```ts
import { UnconstrainedCache } from "bee-agent-framework/cache/unconstrainedCache";

const cache = new UnconstrainedCache<number>();

async function fibonacci(n: number): Promise<number> {
const cacheKey = n.toString();
const cached = await cache.get(cacheKey);
if (cached !== undefined) {
return cached;
}

const result = n < 1 ? 0 : n <= 2 ? 1 : (await fibonacci(n - 1)) + (await fibonacci(n - 2));
await cache.set(cacheKey, result);
return result;
}

console.info(await fibonacci(10)); // 55
console.info(await fibonacci(9)); // 34 (retrieved from cache)
console.info(`Cache size ${await cache.size()}`); // 10
```

_Source: [examples/cache/unconstrainedCacheFunction.ts](/examples/cache/unconstrainedCacheFunction.ts)_

### Usage with tools

<!-- embedme examples/cache/toolCache.ts -->

```ts
import { DuckDuckGoSearchTool } from "bee-agent-framework/tools/search/duckDuckGoSearch";
import { SlidingCache } from "bee-agent-framework/cache/slidingCache";

const ddg = new DuckDuckGoSearchTool({
cache: new SlidingCache({
size: 100, // max 100 entries
ttl: 5 * 60 * 1000, // 5 minutes lifespan
}),
});

const response = await ddg.run({
query: "the time of the fastest marathon run",
page: 1,
});
// upcoming requests with the EXACTLY same input will be retrieved from the cache
```

_Source: [examples/cache/toolCache.ts](/examples/cache/toolCache.ts)_

> [!IMPORTANT]
>
> Cache key is created by serializing function parameters (the order of keys in the object does not matter).
## Cache types

The framework provides multiple out-of-the-box cache implementations.

### UnconstrainedCache

Unlimited in size.

```ts
import { UnconstrainedCache } from "bee-agent-framework/cache/unconstrainedCache";
const cache = new UnconstrainedCache();

await cache.set("a", 1);
console.log(await cache.has("a")); // true
console.log(await cache.size()); // 1
```

### SlidingCache

Keeps last `k` entries in the memory. The oldest ones are deleted.

<!-- embedme examples/cache/slidingCache.ts -->

```ts
import { SlidingCache } from "bee-agent-framework/cache/slidingCache";

const cache = new SlidingCache<number>({
size: 3, // (required) number of items that can be live in the cache at a single moment
ttl: 1000, // (optional, default is Infinity) Time in miliseconds after the element is removed from a cache
});

await cache.set("a", 1);
await cache.set("b", 2);
await cache.set("c", 3);

await cache.set("d", 4); // overflow - cache internally removes the oldest entry (key "a")
console.log(await cache.has("a")); // false
console.log(await cache.size()); // 3
```

_Source: [examples/cache/slidingCache.ts](/examples/cache/slidingCache.ts)_

### FileCache

One may want to persist data to a file so that the data can be later loaded. In that case the `FileCache` is ideal candidate.
You have to provide a location where the cache is persisted.

<!-- embedme examples/cache/fileCache.ts -->

```ts
import { FileCache } from "bee-agent-framework/cache/fileCache";
import * as os from "node:os";

const cache = new FileCache({
fullPath: `${os.tmpdir()}/bee_file_cache.json`,
});
console.log(`Saving cache to "${cache.source}"`);
await cache.set("abc", { firstName: "John", lastName: "Doe" });
```

_Source: [examples/cache/fileCache.ts](/examples/cache/fileCache.ts)_

> [!NOTE]
>
> Provided location (`fullPath`) doesn't have to exist. It gets automatically created when needed.
> [!NOTE]
>
> Every modification to the cache (adding, deleting, clearing) immediately updates the target file.
#### Using a custom provider

<!-- embedme examples/cache/fileCacheCustomProvider.ts -->

```ts
import { FileCache } from "bee-agent-framework/cache/fileCache";
import { UnconstrainedCache } from "bee-agent-framework/cache/unconstrainedCache";
import os from "node:os";

const memoryCache = new UnconstrainedCache<number>();
await memoryCache.set("a", 1);

const fileCache = await FileCache.fromProvider(memoryCache, {
fullPath: `${os.tmpdir()}/bee_file_cache.json`,
});
console.log(`Saving cache to "${fileCache.source}"`);
console.log(await fileCache.get("a")); // 1
```

_Source: [examples/cache/fileCacheCustomProvider.ts](/examples/cache/fileCacheCustomProvider.ts)_

### NullCache

The special type of cache is `NullCache` which implements the `BaseCache` interface but does nothing.

The reason for implementing is to enable [Null object pattern](https://en.wikipedia.org/wiki/Null_object_pattern).

### @Cache (decorator cache)

<!-- embedme examples/cache/decoratorCache.ts -->

```ts
import { Cache } from "bee-agent-framework/cache/decoratorCache";

class Generator {
@Cache()
get(seed: number) {
return (Math.random() * 1000) / Math.max(seed, 1);
}
}

const generator = new Generator();
const a = generator.get(5);
const b = generator.get(5);
console.info(a === b); // true
console.info(a === generator.get(6)); // false
```

_Source: [examples/cache/decoratorCache.ts](/examples/cache/decoratorCache.ts)_

**Complex example**

<!-- embedme examples/cache/decoratorCacheComplex.ts -->

```ts
import { Cache, SingletonCacheKeyFn } from "bee-agent-framework/cache/decoratorCache";

class MyService {
@Cache({
cacheKey: SingletonCacheKeyFn,
ttl: 3600,
enumerable: true,
enabled: true,
})
get id() {
return Math.floor(Math.random() * 1000);
}

reset() {
Cache.getInstance(this, "id").clear();
}
}

const service = new MyService();
const a = service.id;
console.info(a === service.id); // true
service.reset();
console.info(a === service.id); // false
```

_Source: [examples/cache/decoratorCacheComplex.ts](/examples/cache/decoratorCacheComplex.ts)_

> [!NOTE]
>
> Default `cacheKey` function is `ObjectHashKeyFn`
> [!CAUTION]
>
> Calling an annotated method with the `@Cache` decorator with different parameters (despite the fact you are not using them) yields in cache bypass (different arguments = different cache key) generated.
> Be aware of that. If you want your method always to return the same response, use `SingletonCacheKeyFn`.
### CacheFn

Because previously mentioned `CacheDecorator` can be applied only to class methods/getter the framework
provides a way how to do caching on a function level.

<!-- embedme examples/cache/cacheFn.ts -->

```ts
import { CacheFn } from "bee-agent-framework/cache/decoratorCache";
import { setTimeout } from "node:timers/promises";

const getSecret = CacheFn.create(
async () => {
// instead of mocking response you would do a real fetch request
const response = await Promise.resolve({ secret: Math.random(), expiresIn: 100 });
getSecret.updateTTL(response.expiresIn);
return response.secret;
},
{}, // options object
);

const token = await getSecret();
console.info(token === (await getSecret())); // true
await setTimeout(150);
console.info(token === (await getSecret())); // false
```

_Source: [examples/cache/cacheFn.ts](/examples/cache/cacheFn.ts)_

> [!NOTE]
>
> Internally, the function is wrapped as a class; therefore, the same rules apply here as if it were a method annotated with the `@Cache` decorator.
## Creating a custom cache provider

To create your cache implementation, you must implement the `BaseCache` class.

<!-- embedme examples/cache/custom.ts -->

```ts
import { BaseCache } from "bee-agent-framework/cache/base";
import { NotImplementedError } from "bee-agent-framework/errors";

export class CustomCache<T> extends BaseCache<T> {
size(): Promise<number> {
throw new NotImplementedError();
}

set(key: string, value: T): Promise<void> {
throw new NotImplementedError();
}

get(key: string): Promise<T | undefined> {
throw new NotImplementedError();
}

has(key: string): Promise<boolean> {
throw new NotImplementedError();
}

delete(key: string): Promise<boolean> {
throw new NotImplementedError();
}

clear(): Promise<void> {
throw new NotImplementedError();
}

createSnapshot() {
throw new NotImplementedError();
}

loadSnapshot(snapshot: ReturnType<typeof this.createSnapshot>): void {
throw new NotImplementedError();
}
}
```

_Source: [examples/cache/custom.ts](/examples/cache/custom.ts)_

The simplest implementation is `UnconstrainedCache`, which can be found [here](/src/cache/unconstrainedCache.ts).
17 changes: 17 additions & 0 deletions examples/cache/cacheFn.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,17 @@
import { CacheFn } from "bee-agent-framework/cache/decoratorCache";
import { setTimeout } from "node:timers/promises";

const getSecret = CacheFn.create(
async () => {
// instead of mocking response you would do a real fetch request
const response = await Promise.resolve({ secret: Math.random(), expiresIn: 100 });
getSecret.updateTTL(response.expiresIn);
return response.secret;
},
{}, // options object
);

const token = await getSecret();
console.info(token === (await getSecret())); // true
await setTimeout(150);
console.info(token === (await getSecret())); // false
Loading

0 comments on commit 9b60f8a

Please sign in to comment.