Add publishing to pip (#460)

databricks · Jul 20, 2023 · 34db2e7 · 34db2e7
1 parent b762af7
commit 34db2e7
Show file tree

Hide file tree

Showing 124 changed files with 134 additions and 93 deletions.
diff --git a/.coveragerc b/.coveragerc
@@ -1,5 +1,5 @@
 [run]
-source = src
+source = lilac
 omit =
   *_test.py
   */__init__.py
diff --git a/.github/labeler.yml b/.github/labeler.yml
@@ -1,6 +1,6 @@
-# Add 'backend' label to any change within src/ but not web* folders
+# Add 'backend' label to any change within lilac/ but not web* folders
 'backend':
-  - any: ['src/**/*']
+  - any: ['lilac/**/*']
 
 # Add 'frontend' label to any change to blueprint folder
 'frontend':

diff --git a/.github/workflows/python.yml b/.github/workflows/python.yml
@@ -2,7 +2,7 @@ on:
   pull_request:
     types: [opened, reopened, synchronize]
     paths:
-      - 'src/**'
+      - 'lilac/**'
       - '**/*.py'
       - 'poetry.lock'
       - 'poetry.toml'

diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -63,14 +63,14 @@
   "git.confirmSync": false,
   "git.autofetch": true,
   "git.postCommitCommand": "sync",
-  "python.testing.pytestArgs": ["src"],
+  "python.testing.pytestArgs": ["lilac"],
   "python.testing.unittestEnabled": false,
   "python.testing.pytestEnabled": true,
   "python.analysis.importFormat": "relative",
   "python.analysis.indexing": true,
   "python.analysis.useLibraryCodeForTypes": true,
-  "python.analysis.extraPaths": ["src"],
-  "python.analysis.include": ["src"],
+  "python.analysis.extraPaths": ["lilac"],
+  "python.analysis.include": ["lilac"],
   "notebook.output.textLineLimit": 200,
   "notebook.output.scrolling": true,
   "notebook.lineNumbers": "on",

diff --git a/Dockerfile b/Dockerfile
@@ -20,13 +20,13 @@ COPY LICENSE .
 COPY /web/blueprint/build ./web/blueprint/build
 
 # Copy python files.
-COPY /src ./src/
+COPY /lilac ./lilac/
 
 # Copy the data files. We use glob so docker copy won't fail if the directory doesn't exist.
 COPY /dat[a] ./data/
 
 CMD [ \
-  "gunicorn", "src.server:app", \
+  "gunicorn", "lilac.server:app", \
   "--bind", "0.0.0.0:5432", \
   "-k", "uvicorn.workers.UvicornWorker" \
   ]
diff --git a/README.md b/README.md
@@ -100,6 +100,15 @@ key = ''.join(random.choices(string.ascii_uppercase + string.digits, k=64))
 print(f"LILAC_OAUTH_SECRET_KEY='{key}'")
 ```
 
+### Publishing on pip
+
+To authenticate, add the `PYPI_TOKEN` to your `.env.local` file. You can get the token from
+[pypi.org](https://pypi.org/manage/project/lilacai/settings/). To publish, run:
+
+```sh
+./scripts/publish_pip.sh
+```
+
 ### Configuration
 
 To use various API's, API keys need to be provided. Create a file named `.env.local` in the root, and add variables that are listed in `.env` with your own values.
@@ -129,7 +138,7 @@ Test JavaScript:
 Datasets can be ingested entirely from the UI, however if you prefer to use the CLI you can ingest data with the following command:
 
 ```sh
-poetry run python -m src.data_loader \
+poetry run python -m lilac.data_loader \
   --dataset_name=$DATASET \
   --output_dir=./data/ \
   --config_path=./datasets/the_movies_dataset.json

diff --git a/src/__init__.py → lilac/__init__.py b/src/__init__.py → lilac/__init__.py
diff --git a/src/auth.py → lilac/auth.py b/src/auth.py → lilac/auth.py
diff --git a/src/concepts/__init__.py → lilac/concepts/__init__.py b/src/concepts/__init__.py → lilac/concepts/__init__.py
diff --git a/src/concepts/concept.py → lilac/concepts/concept.py b/src/concepts/concept.py → lilac/concepts/concept.py
diff --git a/src/concepts/concept_test.py → lilac/concepts/concept_test.py b/src/concepts/concept_test.py → lilac/concepts/concept_test.py
diff --git a/src/concepts/db_concept.py → lilac/concepts/db_concept.py b/src/concepts/db_concept.py → lilac/concepts/db_concept.py
diff --git a/src/concepts/db_concept_test.py → lilac/concepts/db_concept_test.py b/src/concepts/db_concept_test.py → lilac/concepts/db_concept_test.py
diff --git a/src/config.py → lilac/config.py b/src/config.py → lilac/config.py
diff --git a/src/conftest.py → lilac/conftest.py b/src/conftest.py → lilac/conftest.py
diff --git a/src/data/__init__.py → lilac/data/__init__.py b/src/data/__init__.py → lilac/data/__init__.py
diff --git a/src/data/dataset.py → lilac/data/dataset.py b/src/data/dataset.py → lilac/data/dataset.py
diff --git a/...data/dataset_compute_signal_chain_test.py → ...data/dataset_compute_signal_chain_test.py b/...data/dataset_compute_signal_chain_test.py → ...data/dataset_compute_signal_chain_test.py
diff --git a/src/data/dataset_compute_signal_test.py → lilac/data/dataset_compute_signal_test.py b/src/data/dataset_compute_signal_test.py → lilac/data/dataset_compute_signal_test.py
diff --git a/src/data/dataset_duckdb.py → lilac/data/dataset_duckdb.py b/src/data/dataset_duckdb.py → lilac/data/dataset_duckdb.py
diff --git a/src/data/dataset_select_groups_test.py → lilac/data/dataset_select_groups_test.py b/src/data/dataset_select_groups_test.py → lilac/data/dataset_select_groups_test.py
diff --git a/src/data/dataset_select_rows_filter_test.py → ...c/data/dataset_select_rows_filter_test.py b/src/data/dataset_select_rows_filter_test.py → ...c/data/dataset_select_rows_filter_test.py
diff --git a/src/data/dataset_select_rows_schema_test.py → ...c/data/dataset_select_rows_schema_test.py b/src/data/dataset_select_rows_schema_test.py → ...c/data/dataset_select_rows_schema_test.py
diff --git a/src/data/dataset_select_rows_search_test.py → ...c/data/dataset_select_rows_search_test.py b/src/data/dataset_select_rows_search_test.py → ...c/data/dataset_select_rows_search_test.py
diff --git a/src/data/dataset_select_rows_sort_test.py → lilac/data/dataset_select_rows_sort_test.py b/src/data/dataset_select_rows_sort_test.py → lilac/data/dataset_select_rows_sort_test.py
diff --git a/src/data/dataset_select_rows_udf_test.py → lilac/data/dataset_select_rows_udf_test.py b/src/data/dataset_select_rows_udf_test.py → lilac/data/dataset_select_rows_udf_test.py
diff --git a/src/data/dataset_settings_test.py → lilac/data/dataset_settings_test.py b/src/data/dataset_settings_test.py → lilac/data/dataset_settings_test.py
diff --git a/src/data/dataset_stats_test.py → lilac/data/dataset_stats_test.py b/src/data/dataset_stats_test.py → lilac/data/dataset_stats_test.py
diff --git a/src/data/dataset_test.py → lilac/data/dataset_test.py b/src/data/dataset_test.py → lilac/data/dataset_test.py
diff --git a/src/data/dataset_test_utils.py → lilac/data/dataset_test_utils.py b/src/data/dataset_test_utils.py → lilac/data/dataset_test_utils.py
diff --git a/src/data/dataset_utils.py → lilac/data/dataset_utils.py b/src/data/dataset_utils.py → lilac/data/dataset_utils.py
diff --git a/src/data/dataset_utils_test.py → lilac/data/dataset_utils_test.py b/src/data/dataset_utils_test.py → lilac/data/dataset_utils_test.py
@@ -134,6 +134,7 @@ def func(xs: Iterable[int]) -> Iterable[int]:
   out = sparse_to_dense_compute(sparse_input, func)
   assert list(out) == [None, 2, 8, None, None, 4, None, 6, None, None]
 
+
 def test_sparse_to_dense_compute_batching() -> None:
   sparse_input = iter([None, 1, 7, None, None, 3, None, 5, None, None])
 

diff --git a/src/data/duckdb_utils.py → lilac/data/duckdb_utils.py b/src/data/duckdb_utils.py → lilac/data/duckdb_utils.py
diff --git a/src/data/sources/__init__.py → lilac/data/sources/__init__.py b/src/data/sources/__init__.py → lilac/data/sources/__init__.py
diff --git a/src/data/sources/csv_source.py → lilac/data/sources/csv_source.py b/src/data/sources/csv_source.py → lilac/data/sources/csv_source.py
diff --git a/src/data/sources/csv_source_test.py → lilac/data/sources/csv_source_test.py b/src/data/sources/csv_source_test.py → lilac/data/sources/csv_source_test.py
diff --git a/src/data/sources/default_sources.py → lilac/data/sources/default_sources.py b/src/data/sources/default_sources.py → lilac/data/sources/default_sources.py
diff --git a/src/data/sources/gmail_source.py → lilac/data/sources/gmail_source.py b/src/data/sources/gmail_source.py → lilac/data/sources/gmail_source.py
diff --git a/src/data/sources/huggingface_source.py → lilac/data/sources/huggingface_source.py b/src/data/sources/huggingface_source.py → lilac/data/sources/huggingface_source.py
diff --git a/src/data/sources/huggingface_source_test.py → ...c/data/sources/huggingface_source_test.py b/src/data/sources/huggingface_source_test.py → ...c/data/sources/huggingface_source_test.py
diff --git a/src/data/sources/json_source.py → lilac/data/sources/json_source.py b/src/data/sources/json_source.py → lilac/data/sources/json_source.py
diff --git a/src/data/sources/json_source_test.py → lilac/data/sources/json_source_test.py b/src/data/sources/json_source_test.py → lilac/data/sources/json_source_test.py
diff --git a/src/data/sources/pandas_source.py → lilac/data/sources/pandas_source.py b/src/data/sources/pandas_source.py → lilac/data/sources/pandas_source.py
diff --git a/src/data/sources/pandas_source_test.py → lilac/data/sources/pandas_source_test.py b/src/data/sources/pandas_source_test.py → lilac/data/sources/pandas_source_test.py
diff --git a/src/data/sources/reddit_hf_source.py → lilac/data/sources/reddit_hf_source.py b/src/data/sources/reddit_hf_source.py → lilac/data/sources/reddit_hf_source.py
diff --git a/src/data/sources/source.py → lilac/data/sources/source.py b/src/data/sources/source.py → lilac/data/sources/source.py
diff --git a/src/data/sources/source_registry.py → lilac/data/sources/source_registry.py b/src/data/sources/source_registry.py → lilac/data/sources/source_registry.py
diff --git a/src/data/sources/source_registry_test.py → lilac/data/sources/source_registry_test.py b/src/data/sources/source_registry_test.py → lilac/data/sources/source_registry_test.py
diff --git a/src/data_loader.py → lilac/data_loader.py b/src/data_loader.py → lilac/data_loader.py
@@ -2,7 +2,7 @@
 
 To run the source loader as a binary directly:
 
-poetry run python -m src.data_loader \
+poetry run python -m lilac.data_loader \
   --dataset_name=movies_dataset \
   --output_dir=./data/ \
   --config_path=./datasets/the_movies_dataset.json

diff --git a/src/data_loader_test.py → lilac/data_loader_test.py b/src/data_loader_test.py → lilac/data_loader_test.py
diff --git a/src/db_manager.py → lilac/db_manager.py b/src/db_manager.py → lilac/db_manager.py
diff --git a/src/embeddings/__init__.py → lilac/embeddings/__init__.py b/src/embeddings/__init__.py → lilac/embeddings/__init__.py
diff --git a/src/embeddings/cohere.py → lilac/embeddings/cohere.py b/src/embeddings/cohere.py → lilac/embeddings/cohere.py
diff --git a/src/embeddings/embedding.py → lilac/embeddings/embedding.py b/src/embeddings/embedding.py → lilac/embeddings/embedding.py
diff --git a/src/embeddings/embedding_test.py → lilac/embeddings/embedding_test.py b/src/embeddings/embedding_test.py → lilac/embeddings/embedding_test.py
diff --git a/src/embeddings/openai.py → lilac/embeddings/openai.py b/src/embeddings/openai.py → lilac/embeddings/openai.py
diff --git a/src/embeddings/palm.py → lilac/embeddings/palm.py b/src/embeddings/palm.py → lilac/embeddings/palm.py
diff --git a/src/embeddings/sbert.py → lilac/embeddings/sbert.py b/src/embeddings/sbert.py → lilac/embeddings/sbert.py
diff --git a/src/embeddings/vector_store.py → lilac/embeddings/vector_store.py b/src/embeddings/vector_store.py → lilac/embeddings/vector_store.py
diff --git a/src/embeddings/vector_store_numpy.py → lilac/embeddings/vector_store_numpy.py b/src/embeddings/vector_store_numpy.py → lilac/embeddings/vector_store_numpy.py
diff --git a/src/embeddings/vector_store_test.py → lilac/embeddings/vector_store_test.py b/src/embeddings/vector_store_test.py → lilac/embeddings/vector_store_test.py
diff --git a/src/make_openapi.py → lilac/make_openapi.py b/src/make_openapi.py → lilac/make_openapi.py
@@ -1,6 +1,6 @@
 """Writes the openapi.json file to the specified output.
 
-This is meant to run as a standalone script. It lives in src/ so that we can import the FastAPI app.
+This is meant to run as a standalone script. It lives in lilac/ so we can import the FastAPI app.
 """
 import json
 

diff --git a/src/parquet_writer.py → lilac/parquet_writer.py b/src/parquet_writer.py → lilac/parquet_writer.py
diff --git a/src/router_concept.py → lilac/router_concept.py b/src/router_concept.py → lilac/router_concept.py
diff --git a/src/router_data_loader.py → lilac/router_data_loader.py b/src/router_data_loader.py → lilac/router_data_loader.py
@@ -2,7 +2,7 @@
 
 To run the source loader as a binary directly:
 
-poetry run python -m src.datasets.loader \
+poetry run python -m lilac.datasets.loader \
   --dataset_name=$DATASET \
   --output_dir=./data/ \
   --config_path=./datasets/the_movies_dataset.json

diff --git a/src/router_dataset.py → lilac/router_dataset.py b/src/router_dataset.py → lilac/router_dataset.py
diff --git a/src/router_google_login.py → lilac/router_google_login.py b/src/router_google_login.py → lilac/router_google_login.py
diff --git a/src/router_signal.py → lilac/router_signal.py b/src/router_signal.py → lilac/router_signal.py
diff --git a/src/router_tasks.py → lilac/router_tasks.py b/src/router_tasks.py → lilac/router_tasks.py
diff --git a/src/router_utils.py → lilac/router_utils.py b/src/router_utils.py → lilac/router_utils.py
diff --git a/src/schema.py → lilac/schema.py b/src/schema.py → lilac/schema.py
diff --git a/src/schema_test.py → lilac/schema_test.py b/src/schema_test.py → lilac/schema_test.py
diff --git a/src/server.py → lilac/server.py b/src/server.py → lilac/server.py
diff --git a/src/server_concept_test.py → lilac/server_concept_test.py b/src/server_concept_test.py → lilac/server_concept_test.py
diff --git a/src/server_dataset_test.py → lilac/server_dataset_test.py b/src/server_dataset_test.py → lilac/server_dataset_test.py
diff --git a/src/signals/__init__.py → lilac/signals/__init__.py b/src/signals/__init__.py → lilac/signals/__init__.py
diff --git a/src/signals/concept_labels.py → lilac/signals/concept_labels.py b/src/signals/concept_labels.py → lilac/signals/concept_labels.py
diff --git a/src/signals/concept_labels_test.py → lilac/signals/concept_labels_test.py b/src/signals/concept_labels_test.py → lilac/signals/concept_labels_test.py
diff --git a/src/signals/concept_scorer.py → lilac/signals/concept_scorer.py b/src/signals/concept_scorer.py → lilac/signals/concept_scorer.py
diff --git a/src/signals/concept_scorer_test.py → lilac/signals/concept_scorer_test.py b/src/signals/concept_scorer_test.py → lilac/signals/concept_scorer_test.py
diff --git a/src/signals/default_signals.py → lilac/signals/default_signals.py b/src/signals/default_signals.py → lilac/signals/default_signals.py
diff --git a/src/signals/lang_detection.py → lilac/signals/lang_detection.py b/src/signals/lang_detection.py → lilac/signals/lang_detection.py
diff --git a/src/signals/lang_detection_test.py → lilac/signals/lang_detection_test.py b/src/signals/lang_detection_test.py → lilac/signals/lang_detection_test.py
diff --git a/src/signals/minhash_dup.py → lilac/signals/minhash_dup.py b/src/signals/minhash_dup.py → lilac/signals/minhash_dup.py
diff --git a/src/signals/near_dup.py → lilac/signals/near_dup.py b/src/signals/near_dup.py → lilac/signals/near_dup.py
diff --git a/src/signals/near_dup_test.py → lilac/signals/near_dup_test.py b/src/signals/near_dup_test.py → lilac/signals/near_dup_test.py
diff --git a/src/signals/ner.py → lilac/signals/ner.py b/src/signals/ner.py → lilac/signals/ner.py
diff --git a/src/signals/ner_test.py → lilac/signals/ner_test.py b/src/signals/ner_test.py → lilac/signals/ner_test.py
diff --git a/src/signals/pii.py → lilac/signals/pii.py b/src/signals/pii.py → lilac/signals/pii.py
diff --git a/src/signals/pii_ip_address.py → lilac/signals/pii_ip_address.py b/src/signals/pii_ip_address.py → lilac/signals/pii_ip_address.py
diff --git a/src/signals/pii_secrets.py → lilac/signals/pii_secrets.py b/src/signals/pii_secrets.py → lilac/signals/pii_secrets.py
diff --git a/src/signals/pii_test.py → lilac/signals/pii_test.py b/src/signals/pii_test.py → lilac/signals/pii_test.py
diff --git a/src/signals/semantic_similarity.py → lilac/signals/semantic_similarity.py b/src/signals/semantic_similarity.py → lilac/signals/semantic_similarity.py
diff --git a/src/signals/semantic_similarity_test.py → lilac/signals/semantic_similarity_test.py b/src/signals/semantic_similarity_test.py → lilac/signals/semantic_similarity_test.py
diff --git a/src/signals/signal.py → lilac/signals/signal.py b/src/signals/signal.py → lilac/signals/signal.py
diff --git a/src/signals/signal_test.py → lilac/signals/signal_test.py b/src/signals/signal_test.py → lilac/signals/signal_test.py
diff --git a/src/signals/splitters/__init__.py → lilac/signals/splitters/__init__.py b/src/signals/splitters/__init__.py → lilac/signals/splitters/__init__.py
diff --git a/src/signals/splitters/chunk_splitter.py → lilac/signals/splitters/chunk_splitter.py b/src/signals/splitters/chunk_splitter.py → lilac/signals/splitters/chunk_splitter.py
diff --git a/src/signals/splitters/chunk_splitter_test.py → .../signals/splitters/chunk_splitter_test.py b/src/signals/splitters/chunk_splitter_test.py → .../signals/splitters/chunk_splitter_test.py
diff --git a/src/signals/splitters/text_splitter_spacy.py → .../signals/splitters/text_splitter_spacy.py b/src/signals/splitters/text_splitter_spacy.py → .../signals/splitters/text_splitter_spacy.py
diff --git a/...als/splitters/text_splitter_spacy_test.py → ...als/splitters/text_splitter_spacy_test.py b/...als/splitters/text_splitter_spacy_test.py → ...als/splitters/text_splitter_spacy_test.py
diff --git a/...als/splitters/text_splitter_test_utils.py → ...als/splitters/text_splitter_test_utils.py b/...als/splitters/text_splitter_test_utils.py → ...als/splitters/text_splitter_test_utils.py
diff --git a/...plitters/text_splitter_test_utils_test.py → ...plitters/text_splitter_test_utils_test.py b/...plitters/text_splitter_test_utils_test.py → ...plitters/text_splitter_test_utils_test.py
diff --git a/src/signals/substring_search.py → lilac/signals/substring_search.py b/src/signals/substring_search.py → lilac/signals/substring_search.py
diff --git a/src/signals/substring_search_test.py → lilac/signals/substring_search_test.py b/src/signals/substring_search_test.py → lilac/signals/substring_search_test.py
diff --git a/src/signals/text_statistics.py → lilac/signals/text_statistics.py b/src/signals/text_statistics.py → lilac/signals/text_statistics.py
diff --git a/src/signals/text_statistics_test.py → lilac/signals/text_statistics_test.py b/src/signals/text_statistics_test.py → lilac/signals/text_statistics_test.py
diff --git a/src/tasks.py → lilac/tasks.py b/src/tasks.py → lilac/tasks.py
diff --git a/src/test_utils.py → lilac/test_utils.py b/src/test_utils.py → lilac/test_utils.py
diff --git a/src/utils.py → lilac/utils.py b/src/utils.py → lilac/utils.py
diff --git a/notebooks/API.ipynb b/notebooks/API.ipynb
@@ -12,28 +12,20 @@
    "execution_count": 1,
    "metadata": {},
    "outputs": [
-    {
-     "name": "stderr",
-     "output_type": "stream",
-     "text": [
-      "/Users/dsmilkov/code/lilac/.venv/lib/python3.9/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
-      "  from .autonotebook import tqdm as notebook_tqdm\n"
-     ]
-    },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Computing signal \"signal_name='concept_score' embedding='sbert' namespace='lilac' concept_name='legal-termination' draft='main' num_negative_examples=100\" took 0.280s.\n",
-      "Wrote signal manifest to ./data/datasets/local/legal-clauses/clause_text/sbert/embedding/lilac/legal-termination/v31/signal_manifest.json\n"
+      "Computing signal \"signal_name='concept_score' embedding='sbert' namespace='lilac' concept_name='legal-termination' draft='main' num_negative_examples=100\" took 0.339s.\n",
+      "Wrote signal manifest to ./data/datasets/local/legal-clauses/clause_text/sbert/embedding/lilac/legal-termination/v33/signal_manifest.json\n"
      ]
     }
    ],
    "source": [
-    "from src.db_manager import get_dataset, set_default_dataset_cls\n",
-    "from src.data.dataset_duckdb import DatasetDuckDB\n",
-    "from src.signals.concept_scorer import ConceptScoreSignal\n",
-    "from src.signals.default_signals import register_default_signals\n",
+    "from lilac.db_manager import get_dataset, set_default_dataset_cls\n",
+    "from lilac.data.dataset_duckdb import DatasetDuckDB\n",
+    "from lilac.signals.concept_scorer import ConceptScoreSignal\n",
+    "from lilac.signals.default_signals import register_default_signals\n",
     "\n",
     "register_default_signals()\n",
     "set_default_dataset_cls(DatasetDuckDB)\n",

diff --git a/notebooks/Sentiment.ipynb b/notebooks/Sentiment.ipynb
@@ -265,7 +265,7 @@
     }
    ],
    "source": [
-    "from src.embeddings.openai import OpenAI\n",
+    "from lilac.embeddings.openai import OpenAI\n",
     "\n",
     "signal = OpenAI()\n",
     "signal.setup()\n",
@@ -318,7 +318,7 @@
    "source": [
     "from sklearn.linear_model import LogisticRegression\n",
     "from sklearn.metrics import roc_auc_score, f1_score\n",
-    "from src.utils import DebugTimer\n",
+    "from lilac.utils import DebugTimer\n",
     "\n",
     "model = LogisticRegression(\n",
     "  class_weight='balanced', C=30, tol=1e-5, warm_start=True, max_iter=10_000)\n",
@@ -356,7 +356,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from src.concepts.concept import Concept, Example\n",
+    "from lilac.concepts.concept import Concept, Example\n",
     "\n",
     "\n",
     "def save_concept(positive_sentiment):\n",

diff --git a/notebooks/Toxicity.ipynb b/notebooks/Toxicity.ipynb
@@ -181,7 +181,7 @@
    "source": [
     "from sklearn.metrics import roc_auc_score\n",
     "\n",
-    "from src.utils import DebugTimer\n",
+    "from lilac.utils import DebugTimer\n",
     "\n",
     "SAMPLE_SIZE_PER_GROUP = 512\n",
     "LABEL_TYPE = 'obscene'\n",
@@ -214,7 +214,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from src.concepts.concept import Concept, Example\n",
+    "from lilac.concepts.concept import Concept, Example\n",
     "\n",
     "data = {}\n",
     "\n",

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,10 +1,12 @@
 [tool.poetry]
 authors = ["Lilac AI Inc. <[email protected]>"]
-description = "Index your dataset"
-name = "lilac"
-packages = [{include = "src"}]
+description = "Organize unstructured data"
+license = "Apache-2.0"
+name = "lilacai"
+repository = "https://github.com/lilacai/lilac"
+packages = [{include = "lilac"}]
 readme = "README.md"
-version = "0.0.1"
+version = "0.0.6"
 
 [tool.poetry.dependencies]
 
@@ -27,9 +29,9 @@ pillow = "^9.3.0" # Image processing.
 psutil = "^5.9.5"
 pyarrow = "^9.0.0"
 pydantic = "^1.10.11"
-python = "~3.9"
+python = ">=3.9,<4.0"
 python-dotenv = "^1.0.0"
-requests = "^2.28.1"
+requests = "^2"
 scikit-learn = "^1.3.0"
 tenacity = "^8.2.2"
 tqdm = "^4.65.0"
@@ -64,7 +66,6 @@ regex = "^2023.6.3"
 # For language detection.
 langdetect = {version = "^1.0.9", optional = true}
 
-
 [tool.poetry.extras]
 all = [
   "cohere",
@@ -137,11 +138,11 @@ exclude = [
   "**/node_modules",
   "**/__pycache__",
 ]
-include = ["src"]
+include = ["lilac"]
 
 [tool.ruff]
 line-length = 100
-src = ["src"]
+src = ["lilac"]
 
 # Enable Pyflakes `F`, pycodestyle "W" and `E`, "Q" quotes, "I" imports.
 fix = true

diff --git a/pytest.ini b/pytest.ini
@@ -1,7 +1,7 @@
 # Filter deprecation warnings from subpackages so they don't spam the console when testing.
 [pytest]
 testpaths =
-    src
+    lilac
 filterwarnings =
     ignore::DeprecationWarning:bokeh.*:
     ignore::DeprecationWarning:google.api_core.*:

diff --git a/run_server_dev.sh b/run_server_dev.sh
@@ -10,15 +10,15 @@ npm run dev --workspace web/blueprint -- --open &
 pid[2]=$!
 
 # Run the node server.
-poetry run uvicorn src.server:app --reload --port 5432 --host 0.0.0.0 \
-  --reload-dir src &
+poetry run uvicorn lilac.server:app --reload --port 5432 --host 0.0.0.0 \
+  --reload-dir lilac &
 pid[1]=$!
 
 poetry run watchmedo shell-command \
   --patterns="*.py" \
   --recursive \
   --command='poetry run python -m scripts.make_fastapi_client' \
-  ./src &
+  ./lilac &
 pid[0]=$!
 
 # When control+c is pressed, kill all process ids.