Add prompt name as an env var option

app.py

@@ -210,6 +210,7 @@ def format_response(neighbors: list[Work], distances: list[float]) -> str:
 def main():
     # TODO: figure out some better defaults?
     model_name = get_env_var("MODEL_NAME", default="all-MiniLM-L6-v2")
+    prompt_name = get_env_var("PROMPT_NAME")
     trust_remote_code = get_env_var("TRUST_REMOTE_CODE", bool, default=False)
     fp16 = get_env_var("FP16", bool, default=False)
     dir = get_env_var("DIR", Path, default=Path("index"))
@@ -229,7 +230,7 @@ def main():
 
     # function signature: (expanded tuple of input batches) -> tuple of output batches
     def search(query: list[str]) -> tuple[list[str]]:
-        query_embedding = model.encode(query)
+        query_embedding = model.encode(query, prompt_name)
         distances, faiss_ids = index.search_batch("embeddings", query_embedding, k)
 
         faiss_ids_flat = list(chain(*faiss_ids))