From 57daf25db4451bfe80f94796089ad7e7fbae0d11 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Tue, 5 Dec 2023 11:58:17 -0800
Subject: [PATCH] [EXAMPLES FIX]

---
 README.md                      | 7 +++++--
 example.py                     | 1 +
 sequential_workflow_example.py | 6 ++++--
 3 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/README.md b/README.md
index c9441685..3ce706e6 100644
--- a/README.md
+++ b/README.md
@@ -50,7 +50,9 @@ api_key = os.environ.get("OPENAI_API_KEY")
 # Initialize the language model
 llm = OpenAIChat(
     temperature=0.5,
+    model_name="gpt-4",
     openai_api_key=api_key,
+    max_tokens=4000
 )
 
 
@@ -86,9 +88,10 @@ api_key = os.getenv("OPENAI_API_KEY")
 
 # Initialize the language agent
 llm = OpenAIChat(
-    openai_api_key=api_key,
     temperature=0.5,
-    max_tokens=3000,
+    model_name="gpt-4",
+    openai_api_key=api_key,
+    max_tokens=4000
 )
 
 
diff --git a/example.py b/example.py
index 9fc89485..6ed2cbab 100644
--- a/example.py
+++ b/example.py
@@ -15,6 +15,7 @@ api_key = os.environ.get("OPENAI_API_KEY")
 # Initialize the language model
 llm = OpenAIChat(
     temperature=0.5,
+    model_name="gpt-4",
     openai_api_key=api_key,
 )
 
diff --git a/sequential_workflow_example.py b/sequential_workflow_example.py
index 1742f49c..38cf5559 100644
--- a/sequential_workflow_example.py
+++ b/sequential_workflow_example.py
@@ -11,10 +11,12 @@ api_key = os.getenv("OPENAI_API_KEY")
 
 
 # Initialize the language agent
+# Initialize the language model
 llm = OpenAIChat(
-    openai_api_key=api_key,
     temperature=0.5,
-    max_tokens=2000,
+    model_name="gpt-4",
+    openai_api_key=api_key,
+    max_tokens=4000
 )