Explained local setup

README.md

@@ -45,6 +45,26 @@ uv sync
 cp .env.example .env
 ```
 
+## Local Setup Note
+
+For now, `persona-data` and `persona-vectors` need to be checked out in the parent directory of `persona-ui`.
+
+Example:
+
+```bash
+git clone <persona-data-url> ../persona-data
+git clone <persona-vectors-url> ../persona-vectors
+```
+
+Expected layout:
+
+```text
+parent/
+├── persona-ui
+├── persona-data
+└── persona-vectors
+```
+
 ## Quickstart
 
 ```bash

pyproject.toml

@@ -7,21 +7,16 @@ requires-python = ">=3.10"
 dependencies = [
     "persona-vectors",
     "persona-data",
-    "nnterp>=1.3.0",
     "streamlit>=1.44.0",
     "plotly>=6.6.0",
-    "kaleido>=1.0.0",
     "python-dotenv>=1.2.2",
-    "torch>=2.10.0",
-    "transformers>=5.2.0",
 ]
 
 [tool.uv.sources]
-# NOTE: Switch to git sources after pushing the new package structure
-persona-vectors = { git = "ssh://git@github.com/implicit-personalization/persona-vectors.git" }
-persona-data = { git = "ssh://git@github.com/implicit-personalization/persona-data.git" }
-# persona-vectors = { path = "../persona-vectors", editable = true }
-# persona-data = { path = "../persona-data", editable = true }
+persona-vectors = { path = "../persona-vectors", editable = true }
+persona-data = { path = "../persona-data", editable = true }
+# persona-vectors = { git = "ssh://git@github.com/implicit-personalization/persona-vectors.git" }  # use for release
+# persona-data = { git = "ssh://git@github.com/implicit-personalization/persona-data.git" }  # use for release
 
 # [build-system]
 # requires = ["uv_build>=0.11.3,<0.12"]

utils/chat.py

@@ -10,6 +10,7 @@ logger = logging.getLogger(__name__)
 
 from persona_data.synth_persona import PersonaData
 from persona_data.prompts import (
+    format_empty_persona_prompt,
     format_biography_prompt,
     format_templated_prompt,
     normalize_messages,
@@ -17,8 +18,6 @@ from persona_data.prompts import (
 
 SystemPromptMode = Literal["empty", "templated", "biography", "custom"]
 
-_CUSTOM_PROMPT_DEFAULT = "You are a helpful assistant."
-
 
 @dataclass
 class ChatReply:
@@ -45,12 +44,14 @@ def resolve_system_prompt(
     if persona is None:
         return ""
 
+    if mode == "empty":
+        return ""
     if mode == "templated":
         return format_templated_prompt(persona.templated_prompt)
     if mode == "biography":
         return format_biography_prompt(persona.biography_md)
     if mode == "custom":
-        return _CUSTOM_PROMPT_DEFAULT
+        return format_empty_persona_prompt()
     return ""
 
 

uv.lock

@@ -1561,7 +1561,7 @@ wheels = [
 [[package]]
 name = "persona-data"
 version = "0.1.0"
-source = { git = "ssh://git@github.com/implicit-personalization/persona-data.git#d1a40e44a09090567bbe1a0d300f2d272b517851" }
+source = { editable = "../persona-data" }
 dependencies = [
     { name = "huggingface-hub" },
     { name = "numpy", version = "2.2.6", source = { registry = "https://pypi.org/simple" }, marker = "python_full_version < '3.11'" },
@@ -1570,41 +1570,40 @@ dependencies = [
     { name = "torch" },
 ]
 
+[package.metadata]
+requires-dist = [
+    { name = "huggingface-hub", specifier = ">=0.30.0" },
+    { name = "numpy", specifier = ">=1.24.0" },
+    { name = "python-dotenv", specifier = ">=1.0.0" },
+    { name = "torch", specifier = ">=2.0.0" },
+]
+
 [[package]]
 name = "persona-ui"
 version = "0.1.0"
 source = { virtual = "." }
 dependencies = [
-    { name = "kaleido" },
-    { name = "nnterp" },
     { name = "persona-data" },
     { name = "persona-vectors" },
     { name = "plotly" },
     { name = "python-dotenv" },
     { name = "streamlit" },
-    { name = "torch" },
-    { name = "transformers" },
 ]
 
 [package.metadata]
 requires-dist = [
-    { name = "kaleido", specifier = ">=1.0.0" },
-    { name = "nnterp", specifier = ">=1.3.0" },
-    { name = "persona-data", git = "ssh://git@github.com/implicit-personalization/persona-data.git" },
-    { name = "persona-vectors", git = "ssh://git@github.com/implicit-personalization/persona-vectors.git" },
+    { name = "persona-data", editable = "../persona-data" },
+    { name = "persona-vectors", editable = "../persona-vectors" },
     { name = "plotly", specifier = ">=6.6.0" },
     { name = "python-dotenv", specifier = ">=1.2.2" },
     { name = "streamlit", specifier = ">=1.44.0" },
-    { name = "torch", specifier = ">=2.10.0" },
-    { name = "transformers", specifier = ">=5.2.0" },
 ]
 
 [[package]]
 name = "persona-vectors"
 version = "0.1.0"
-source = { git = "ssh://git@github.com/implicit-personalization/persona-vectors.git#27288eba98127491d84db8a27b17c2da177e9624" }
+source = { editable = "../persona-vectors" }
 dependencies = [
-    { name = "huggingface-hub" },
     { name = "kaleido" },
     { name = "nnsight" },
     { name = "nnterp" },
@@ -1621,6 +1620,23 @@ dependencies = [
     { name = "umap-learn" },
 ]
 
+[package.metadata]
+requires-dist = [
+    { name = "kaleido", specifier = ">=1.0.0" },
+    { name = "nnsight", specifier = ">=0.6.1" },
+    { name = "nnterp", specifier = ">=1.3.0" },
+    { name = "persona-data", editable = "../persona-data" },
+    { name = "plotly", specifier = ">=6.6.0" },
+    { name = "python-dotenv", specifier = ">=1.2.2" },
+    { name = "safetensors", specifier = ">=0.7.0" },
+    { name = "scikit-learn", specifier = ">=1.6.0" },
+    { name = "torch", specifier = ">=2.10.0" },
+    { name = "torchvision", specifier = ">=0.26.0" },
+    { name = "tqdm", specifier = ">=4.67.3" },
+    { name = "transformers", specifier = ">=5.2.0" },
+    { name = "umap-learn", specifier = ">=0.5.7" },
+]
+
 [[package]]
 name = "pexpect"
 version = "4.9.0"