Skip to content

Commit 6258ddf

Browse files
committed
example: cleanup pyproject, rename
1 parent b764061 commit 6258ddf

File tree

7 files changed

+16
-59
lines changed

7 files changed

+16
-59
lines changed

examples/postgres_embedding/pyproject.toml

Lines changed: 0 additions & 52 deletions
This file was deleted.
File renamed without changes.
File renamed without changes.

examples/postgres_embedding/README.md renamed to examples/postgres_source/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# PostgreSQL Source Embedding Example 🗄️
1+
# PostgreSQL Source Example 🗄️
22

33
[![GitHub](https://img.shields.io/github/stars/cocoindex-io/cocoindex?color=5B5BD6)](https://github.com/cocoindex-io/cocoindex)
44

examples/postgres_embedding/main.py renamed to examples/postgres_source/main.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def postgres_message_indexing_flow(
2525
)
2626
)
2727

28-
message_embeddings = data_scope.add_collector()
28+
indexed_messages = data_scope.add_collector()
2929
with data_scope["messages"].row() as message_row:
3030
# Use the indexing column for embedding generation
3131
message_row["embedding"] = message_row["message"].transform(
@@ -34,14 +34,14 @@ def postgres_message_indexing_flow(
3434
)
3535
)
3636
# Collect the data - include key columns and content
37-
message_embeddings.collect(
37+
indexed_messages.collect(
3838
id=message_row["id"],
3939
author=message_row["author"],
4040
message=message_row["message"],
4141
embedding=message_row["embedding"],
4242
)
4343

44-
message_embeddings.export(
44+
indexed_messages.export(
4545
"output",
4646
cocoindex.targets.Postgres(),
4747
primary_key_fields=["id"],
@@ -93,7 +93,7 @@ def postgres_product_indexing_flow(
9393
)
9494
)
9595

96-
product_embeddings = data_scope.add_collector()
96+
indexed_product = data_scope.add_collector()
9797
with data_scope["products"].row() as product:
9898
product["full_description"] = flow_builder.transform(
9999
make_full_description,
@@ -111,7 +111,7 @@ def postgres_product_indexing_flow(
111111
model="sentence-transformers/all-MiniLM-L6-v2"
112112
)
113113
)
114-
product_embeddings.collect(
114+
indexed_product.collect(
115115
product_category=product["_key"]["product_category"],
116116
product_name=product["_key"]["product_name"],
117117
description=product["description"],
@@ -121,7 +121,7 @@ def postgres_product_indexing_flow(
121121
embedding=product["embedding"],
122122
)
123123

124-
product_embeddings.export(
124+
indexed_product.export(
125125
"output",
126126
cocoindex.targets.Postgres(),
127127
primary_key_fields=["product_category", "product_name"],
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
[project]
2+
name = "postgres-source"
3+
version = "0.1.0"
4+
description = "Demonstrate how to use Postgres tables as the source for CocoIndex."
5+
requires-python = ">=3.11"
6+
dependencies = ["cocoindex[embeddings]>=0.1.82"]
7+
8+
[tool.setuptools]
9+
packages = []

0 commit comments

Comments
 (0)