From 6274622a8d8df7f7e8bf708a7656c720835b87a9 Mon Sep 17 00:00:00 2001
From: David Chen <darkwei233@gmail.com>
Date: Tue, 28 Oct 2025 16:22:49 -0700
Subject: [PATCH] custom codex prompt dir

---
 zsh/functions/co.zsh | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/zsh/functions/co.zsh b/zsh/functions/co.zsh
index c923a99..f296d25 100644
--- a/zsh/functions/co.zsh
+++ b/zsh/functions/co.zsh
@@ -113,9 +113,16 @@ co() {
       fi
     fi
   done
-  if [ -d "$PWD/codex-prompts" ]; then
+  local project_prompts_dir=""
+  if [ -d "$PWD/.agent-prompts" ]; then
+    project_prompts_dir="$PWD/.agent-prompts"
+  elif [ -d "$PWD/codex-prompts" ]; then
+    project_prompts_dir="$PWD/codex-prompts"
+  fi
+
+  if [ -n "$project_prompts_dir" ]; then
     local copied_any=0
-    for f in "$PWD/codex-prompts"/*.md; do
+    for f in "$project_prompts_dir"/*.md; do
       [ -f "$f" ] || continue
       copied_any=1
       if ! cp -f "$f" "$tmp_home/prompts/" >/dev/null 2>&1; then
@@ -126,7 +133,7 @@ co() {
       fi
     done
     if (( copied_any )); then
-      print -u2 "co: added project prompts from $PWD/codex-prompts (overriding base on conflicts)"
+      print -u2 "co: added project prompts from $project_prompts_dir (overriding base on conflicts)"
     fi
   fi