Fix openai model cookbook

spiceai · Jan 21, 2025 · d4e7202 · d4e7202
1 parent b32dd4a
commit d4e7202
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git 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
@@ -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.