From 2413f56980d32a2423ff2c6b1a3215178516f780 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil.hafeez@gmail.com>
Date: Thu, 16 Jan 2025 14:33:59 -0800
Subject: [PATCH] pending changes

---
 tests/archgw/arch_config.yaml | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/tests/archgw/arch_config.yaml b/tests/archgw/arch_config.yaml
index e9afa5c1..3f450717 100644
--- a/tests/archgw/arch_config.yaml
+++ b/tests/archgw/arch_config.yaml
@@ -12,13 +12,22 @@ endpoints:
     connect_timeout: 0.005s
 
 llm_providers:
-
-  - name: local-llm
-    provider: local-llm
-    endpoint: host.docker.internal:51002
-    model: test-local-model
+  - name: gpt-4o-mini
+    access_key: $OPENAI_API_KEY
+    provider: openai
+    model: gpt-4o-mini
     default: true
 
+  - name: gpt-3.5-turbo-0125
+    access_key: $OPENAI_API_KEY
+    provider: openai
+    model: gpt-3.5-turbo-0125
+
+  - name: gpt-4o
+    access_key: $OPENAI_API_KEY
+    provider: openai
+    model: gpt-4o
+
 system_prompt: |
   You are a helpful assistant.