@@ -13,6 +13,7 @@ CONTAINER_DB_PATH ?= /.llama/data/distributions/ansible-chatbot
1313RAG_CONTENT_IMAGE ?= quay.io/ansible/aap-rag-content:1.0.1753876830
1414LIGHTSPEED_STACK_CONFIG ?= lightspeed-stack.yaml
1515LLAMA_STACK_RUN_CONFIG ?= ansible-chatbot-run.yaml
16+ SYSTEM_PROMPT ?= ansible-chatbot-system-prompt.txt
1617# Colors for terminal output
1718RED := \033[0;31m
1819NC := \033[0m # No Color
@@ -132,12 +133,15 @@ run: check-env-run
132133 -v ./vector_db/aap_faiss_store.db:$(CONTAINER_DB_PATH ) /aap_faiss_store.db \
133134 -v ./$(LIGHTSPEED_STACK_CONFIG ) :/.llama/distributions/ansible-chatbot/config/lightspeed-stack.yaml \
134135 -v ./$(LLAMA_STACK_RUN_CONFIG ) :/.llama/distributions/llama-stack/config/ansible-chatbot-run.yaml \
135- -v ./ansible-chatbot-system-prompt.txt :/.llama/distributions/ansible-chatbot/system-prompts/default.txt \
136+ -v ./$( SYSTEM_PROMPT ) :/.llama/distributions/ansible-chatbot/system-prompts/default.txt \
136137 --env VLLM_URL=$(ANSIBLE_CHATBOT_VLLM_URL ) \
137138 --env VLLM_API_TOKEN=$(ANSIBLE_CHATBOT_VLLM_API_TOKEN ) \
138139 --env INFERENCE_MODEL=$(ANSIBLE_CHATBOT_INFERENCE_MODEL ) \
139140 --env INFERENCE_MODEL_FILTER=$(ANSIBLE_CHATBOT_INFERENCE_MODEL_FILTER ) \
140141 --env GEMINI_API_KEY=$(GEMINI_API_KEY ) \
142+ --env OPENAI_INFERENCE_MODEL=$(OPENAI_INFERENCE_MODEL ) \
143+ --env OPENAI_API_KEY=$(OPENAI_API_KEY ) \
144+ --env OPENAI_BASE_URL=$(OPENAI_BASE_URL ) \
141145 $(IMAGE_PREFIX ) ansible-chatbot-stack:$(ANSIBLE_CHATBOT_VERSION )
142146
143147run-test :
@@ -171,12 +175,15 @@ run-local-db: check-env-run-local-db
171175 -v ./vector_db/aap_faiss_store.db:$(CONTAINER_DB_PATH ) /aap_faiss_store.db \
172176 -v ./$(LIGHTSPEED_STACK_CONFIG ) :/.llama/distributions/ansible-chatbot/config/lightspeed-stack.yaml \
173177 -v ./$(LLAMA_STACK_RUN_CONFIG ) :/.llama/distributions/llama-stack/config/ansible-chatbot-run.yaml \
174- -v ./ansible-chatbot-system-prompt.txt :/.llama/distributions/ansible-chatbot/system-prompts/default.txt \
178+ -v ./$( SYSTEM_PROMPT ) :/.llama/distributions/ansible-chatbot/system-prompts/default.txt \
175179 --env VLLM_URL=$(ANSIBLE_CHATBOT_VLLM_URL ) \
176180 --env VLLM_API_TOKEN=$(ANSIBLE_CHATBOT_VLLM_API_TOKEN ) \
177181 --env INFERENCE_MODEL=$(ANSIBLE_CHATBOT_INFERENCE_MODEL ) \
178182 --env INFERENCE_MODEL_FILTER=$(ANSIBLE_CHATBOT_INFERENCE_MODEL_FILTER ) \
179183 --env GEMINI_API_KEY=$(GEMINI_API_KEY ) \
184+ --env OPENAI_INFERENCE_MODEL=$(OPENAI_INFERENCE_MODEL ) \
185+ --env OPENAI_API_KEY=$(OPENAI_API_KEY ) \
186+ --env OPENAI_BASE_URL=$(OPENAI_BASE_URL ) \
180187 $(IMAGE_PREFIX ) ansible-chatbot-stack:$(ANSIBLE_CHATBOT_VERSION )
181188
182189clean :
0 commit comments