Upsert

1#!/usr/bin/env python3
2# /// script
3# requires-python = ">=3.10"
4# dependencies = [
5#   "pinecone==9.1.0",
6#   "typer>=0.15.0",
7# ]
8# ///
9
10import os
11import typer
12from pinecone import Pinecone
13
14app = typer.Typer()
15
16@app.command()
17def main(
18    index: str = typer.Option(..., "--index", help="Name of the Pinecone index to upsert into"),
19    namespace: str = typer.Option("example-namespace", "--namespace", help="Namespace to upsert into"),
20):
21    api_key = os.environ.get("PINECONE_API_KEY")
22    if not api_key:
23        typer.echo("Error: PINECONE_API_KEY environment variable not set", err=True)
24        raise typer.Exit(1)
25
26    pc = Pinecone(api_key=api_key, source_tag="pinecone_skills:upsert")
27
28    records = [
29        # Health / feeling unwell
30        {"_id": "rec1", "chunk_text": "I've been sneezing all day and my nose won't stop running.", "category": "health"},
31        {"_id": "rec2", "chunk_text": "She stayed home with a pounding headache and a low-grade fever.", "category": "health"},
32        {"_id": "rec3", "chunk_text": "He felt completely drained after waking up with a sore throat and chills.", "category": "health"},
33        # Productivity / work
34        {"_id": "rec4", "chunk_text": "She blocked off two hours in the morning to focus without interruptions.", "category": "productivity"},
35        {"_id": "rec5", "chunk_text": "He finished all his tasks ahead of schedule by prioritizing the hardest ones first.", "category": "productivity"},
36        {"_id": "rec6", "chunk_text": "Turning off notifications helped her get into a deep flow state.", "category": "productivity"},
37        # Outdoors / nature
38        {"_id": "rec7", "chunk_text": "A red fox darted across the trail and disappeared into the underbrush.", "category": "nature"},
39        {"_id": "rec8", "chunk_text": "The hikers paused to watch a bald eagle circle lazily over the valley.", "category": "nature"},
40        {"_id": "rec9", "chunk_text": "Fireflies lit up the meadow as the sun dipped below the treeline.", "category": "nature"},
41    ]
42
43    idx = pc.Index(index)
44    idx.upsert_records(namespace=namespace, records=records)
45    typer.echo(f"Upserted {len(records)} records into '{index}' (namespace: '{namespace}')")
46
47if __name__ == "__main__":
48    app()