Customize Your Memory Context String

1
import os
2
import uuid
3
import time
4
from dotenv import find_dotenv, load_dotenv
5
from zep_cloud.client import Zep
6
import json
7
8
load_dotenv(dotenv_path=find_dotenv())
9
API_KEY = os.environ.get("ZEP_API_KEY") or "YOUR_API_KEY"
10
11
client = Zep(api_key=API_KEY)
12
uuid_value = uuid.uuid4().hex
13
group_id = "MyGroup" + uuid_value
14
group = client.group.add(group_id=group_id)
15
16
car_inventory_description = {
17
    "entity": "Car inventory",
18
    "description": "This is a list of cars in the inventory. Each car has a maker, model, and gas mileage."
19
}
20
car_inventory_list = [    
21
    {
22
        "maker": "Toyota",
23
        "model": "Sedan-01",
24
        "gas_mileage": "25 mpg",
25
    },
26
    {
27
        "maker": "Ford",
28
        "model": "SUV-01",
29
        "gas_mileage": "20 mpg",
30
    },
31
    {
32
        "maker": "Ford",
33
        "model": "SUV-02",
34
        "gas_mileage": "22 mpg",
35
    },
36
    {
37
        "maker": "Chevrolet",
38
        "model": "Truck-01",
39
        "gas_mileage": "25 mpg",
40
    },
41
]
42
43
json_string = json.dumps(car_inventory_description)
44
client.graph.add(
45
    group_id=group_id,
46
    type="json",
47
    data=json_string,
48
)
49
for item in car_inventory_list:
50
    # This helps contextualize the car items in the graph
51
    item["description"] = "This is one of the cars in the car inventory"
52
    json_string = json.dumps(item)
53
    client.graph.add(
54
        group_id=group_id,
55
        type="json",
56
        data=json_string,
57
    )
58
59
# Wait a minute or two
60
time.sleep(120)

1
query = "What Ford cars do we have in stock and what are there gas mileages?"
2
node_search_results = client.graph.search(
3
    group_id=group_id,
4
    query=query,
5
    scope="nodes",
6
    limit=3,
7
    reranker="cross_encoder",
8
)
9
edge_search_results = client.graph.search(
10
    group_id=group_id,
11
    query=query,
12
    scope="edges",
13
    limit=3,
14
    reranker="cross_encoder",
15
)

1
custom_context_string = "Below are facts and entities related to what Ford cars we have in stock:\n"
2
3
custom_context_string += "<FACTS>\n"
4
for edge in edge_search_results.edges:
5
    custom_context_string += f"- {edge.fact}\n"
6
custom_context_string += "</FACTS>\n"
7
8
custom_context_string += "<ENTITIES>\n"
9
for node in node_search_results.nodes:
10
    custom_context_string += f"- {node.name}: {node.summary}\n"
11
custom_context_string += "</ENTITIES>\n"
12
13
print(custom_context_string)