Code Embedding (v1) 🧑‍💻

This example extracts code chunks from local Python files, stores the code and their vector embeddings in Postgres (pgvector), and provides a simple semantic search demo for code.

Prerequisites

A running Postgres with the pgvector extension. If you don't have one, start a local instance with the compose file in this repo:
sh
```
docker compose -f ../../dev/postgres.yaml up -d
```

POSTGRES_URL set, e.g.

export POSTGRES_URL="postgres://cocoindex:cocoindex@localhost/cocoindex"

Run

Install deps:

pip install -e .

Build/update the index (writes rows into Postgres). Pick one of the two modes:

Catch-up run — scan sources, sync changes, exit:
sh
```
cocoindex update main
```
Live run — catch up, then keep watching for file changes (the source declares live=True in main.py):
sh
```
cocoindex update -L main
```

Query:

python main.py "embedding"