Check Data Ingestion Status

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
7
load_dotenv(dotenv_path=find_dotenv())
8
9
client = Zep(api_key=os.environ.get("ZEP_API_KEY"))
10
uuid_value = uuid.uuid4().hex[:4]
11
user_id = "-" + uuid_value
12
client.user.add(
13
    user_id=user_id,
14
    first_name = "John",
15
    last_name = "Doe",
16
    email="[email protected]"
17
)

1
episode = client.graph.add(
2
    user_id=user_id,
3
    type="text", 
4
    data="The user is an avid fan of Eric Clapton"
5
)
6
7
search_results = client.graph.search(
8
    user_id=user_id,
9
    query="Eric Clapton",
10
    scope="nodes",
11
    limit=1,
12
    reranker="cross_encoder",
13
)
14
15
print(search_results.nodes)

1
while True:
2
    episode = client.graph.episode.get(
3
        uuid_=episode.uuid_,
4
    )
5
    if episode.processed:
6
        print("Episode processed successfully")
7
        break
8
    print("Waiting for episode to process...")
9
    time.sleep(1)

1
search_results = client.graph.search(
2
    user_id=user_id,
3
    query="Eric Clapton",
4
    scope="nodes",
5
    limit=1,
6
    reranker="cross_encoder",
7
)
8
9
print(search_results.nodes)