Skip to content

Commit 04974e7

Browse files
authored
add openai support (#94)
Add openai support Tested to work with core chatbot and with mcp tested models to work: "gpt-4o-mini" "gpt-3.5-turbo" best tested model: "gpt-4o-mini" gpt-5 models does not work , no compatibility with llama-stack at this moment Signed-off-by: Djebran Lezzoum <[email protected]>
1 parent c588b84 commit 04974e7

File tree

3 files changed

+27
-2
lines changed

3 files changed

+27
-2
lines changed

Makefile

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ CONTAINER_DB_PATH ?= /.llama/data/distributions/ansible-chatbot
1313
RAG_CONTENT_IMAGE ?= quay.io/ansible/aap-rag-content:1.0.1753876830
1414
LIGHTSPEED_STACK_CONFIG ?= lightspeed-stack.yaml
1515
LLAMA_STACK_RUN_CONFIG ?= ansible-chatbot-run.yaml
16+
SYSTEM_PROMPT ?= ansible-chatbot-system-prompt.txt
1617
# Colors for terminal output
1718
RED := \033[0;31m
1819
NC := \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

143147
run-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

182189
clean:

ansible-chatbot-mcp-run.yaml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,11 @@ providers:
2222
provider_type: remote::gemini
2323
config:
2424
api_key: ${env.GEMINI_API_KEY:=fake}
25+
- provider_id: openai
26+
provider_type: remote::openai
27+
config:
28+
api_key: ${env.OPENAI_API_KEY:=}
29+
base_url: ${env.OPENAI_BASE_URL:=https://api.openai.com/v1}
2530
- provider_id: inline_sentence-transformer
2631
provider_type: inline::sentence-transformers
2732
config: {}
@@ -94,6 +99,10 @@ models:
9499
provider_id: gemini
95100
provider_model_id: gemini/gemini-2.5-flash
96101
model_type: llm
102+
- metadata: {}
103+
model_id: ${env.OPENAI_INFERENCE_MODEL:=gpt-4o-mini}
104+
provider_id: openai
105+
provider_model_id: null
97106
shields: []
98107
vector_dbs:
99108
- metadata: {}

ansible-chatbot-run.yaml

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,11 @@ providers:
2222
provider_type: remote::gemini
2323
config:
2424
api_key: ${env.GEMINI_API_KEY:=fake}
25+
- provider_id: openai
26+
provider_type: remote::openai
27+
config:
28+
api_key: ${env.OPENAI_API_KEY:=}
29+
base_url: ${env.OPENAI_BASE_URL:=https://api.openai.com/v1}
2530
- provider_id: inline_sentence-transformer
2631
provider_type: inline::sentence-transformers
2732
config: {}
@@ -94,6 +99,10 @@ models:
9499
provider_id: gemini
95100
provider_model_id: gemini/gemini-2.5-flash
96101
model_type: llm
102+
- metadata: {}
103+
model_id: ${env.OPENAI_INFERENCE_MODEL:=gpt-4o-mini}
104+
provider_id: openai
105+
provider_model_id: null
97106
shields: []
98107
vector_dbs:
99108
- metadata: {}

0 commit comments

Comments
 (0)