Adding Episodes

1
await graphiti.add_episode(
2
    name="tech_innovation_article",
3
    episode_body=(
4
        "MIT researchers have unveiled 'ClimateNet', an AI system capable of predicting "
5
        "climate patterns with unprecedented accuracy. Early tests show it can forecast "
6
        "major weather events up to three weeks in advance, potentially revolutionizing "
7
        "disaster preparedness and agricultural planning."
8
    ),
9
    source=EpisodeType.text,
10
    # A description of the source (e.g., "podcast", "news article")
11
    source_description="Technology magazine article",
12
    # The timestamp for when this episode occurred or was created
13
    reference_time=datetime(2023, 11, 15, 9, 30),
14
)

1
await graphiti.add_episode(
2
    name="Customer_Support_Interaction_1",
3
    episode_body=(
4
        "Customer: Hi, I'm having trouble with my Allbirds shoes. "
5
        "The sole is coming off after only 2 months of use.\n"
6
        "Support: I'm sorry to hear that. Can you please provide your order number?"
7
    ),
8
    source=EpisodeType.message,
9
    source_description="Customer support chat",
10
    reference_time=datetime(2024, 3, 15, 14, 45),
11
)

1
import json
2
3
product_data = {
4
    "id": "PROD001",
5
    "name": "Men's SuperLight Wool Runners",
6
    "color": "Dark Grey",
7
    "sole_color": "Medium Grey",
8
    "material": "Wool",
9
    "technology": "SuperLight Foam",
10
    "price": 125.00,
11
    "in_stock": True,
12
    "last_updated": "2024-03-15T10:30:00Z"
13
}
14
15
# Add the episode to the graph
16
await graphiti.add_episode(
17
    name="Product Update - PROD001",
18
    episode_body=json.dumps(product_data),  # episode_body must be a JSON string, not a dict
19
    source=EpisodeType.json,
20
    source_description="Allbirds product catalog update",
21
    reference_time=datetime.now(),
22
)

1
product_data = [
2
    {
3
        "id": "PROD001",
4
        "name": "Men's SuperLight Wool Runners",
5
        "color": "Dark Grey",
6
        "sole_color": "Medium Grey",
7
        "material": "Wool",
8
        "technology": "SuperLight Foam",
9
        "price": 125.00,
10
        "in_stock": true,
11
        "last_updated": "2024-03-15T10:30:00Z"
12
    },
13
    ...
14
    {
15
        "id": "PROD0100",
16
        "name": "Kids Wool Runner-up Mizzles",
17
        "color": "Natural Grey",
18
        "sole_color": "Orange",
19
        "material": "Wool",
20
        "technology": "Water-repellent",
21
        "price": 80.00,
22
        "in_stock": true,
23
        "last_updated": "2024-03-17T14:45:00Z"
24
    }
25
]
26
27
# Prepare the episodes for bulk loading
28
29
bulk_episodes = [
30
RawEpisode(
31
name=f"Product Update - {product['id']}",
32
content=json.dumps(product),
33
source=EpisodeType.json,
34
source_description="Allbirds product catalog update",
35
reference_time=datetime.now()
36
)
37
for product in product_data
38
]
39
40
await graphiti.add_episode_bulk(bulk_episodes)