From d4e7202dc916233d0350d6e76906e5b4663e7462 Mon Sep 17 00:00:00 2001
From: Sevenannn <qianqliu@uw.edu>
Date: Tue, 21 Jan 2025 14:19:33 -0800
Subject: [PATCH] Fix openai model cookbook

---
 models/openai/README.md     | 13 +++----------
 models/openai/spicepod.yaml |  3 +++
 2 files changed, 6 insertions(+), 10 deletions(-)

diff --git a/models/openai/README.md b/models/openai/README.md
index d4c14eb..34bea49 100644
--- a/models/openai/README.md
+++ b/models/openai/README.md
@@ -40,11 +40,14 @@ datasets:
 embeddings:
   - from: openai:text-embedding-3-small
     name: embeddings-model
+    params:
+      openai_api_key: ${secrets:OPENAI_API_KEY}
 
 models:
   - from: openai:gpt-4o
     name: chat-model
     params:
+      openai_api_key: ${secrets:OPENAI_API_KEY}
       tools: auto
       system_prompt: |
         You are a helpful Spice.ai Docs assistant.
@@ -170,16 +173,6 @@ I have access to the following dataset:
 This dataset contains documentation related to the Spice.ai project.
 ```
 
-```shell
-chat> how many records in taxi trips dataset
-There are a total of 2,964,624 records in the taxi trips dataset.
-```
-
-```shell
-chat> what is the longest taxi trip distance recorded
-The longest taxi trip distance recorded is approximately 312,722.3 meters.
-```
-
 ```shell
 chat> What are release criterias?
 The release criteria for Spice.ai components, such as models, data accelerators, and catalog connectors, are divided into stages, including Release Candidate (RC) and Stable release criteria. Here are the details for RC Criteria:
diff --git a/models/openai/spicepod.yaml b/models/openai/spicepod.yaml
index 89b34f7..d24f717 100644
--- a/models/openai/spicepod.yaml
+++ b/models/openai/spicepod.yaml
@@ -26,11 +26,14 @@ datasets:
 embeddings:
   - from: openai:text-embedding-3-small
     name: embeddings-model
+    params:
+      openai_api_key: ${secrets:OPENAI_API_KEY}
 
 models:
   - from: openai:gpt-4o
     name: chat-model
     params:
+      openai_api_key: ${secrets:OPENAI_API_KEY}
       tools: auto
       system_prompt: |
         You are a helpful Spice.ai Docs assistant.