Merge pull request #29 from winniesi/dev

Allow users to customize the model

.env.local.example

@@ -2,6 +2,10 @@
 # If you need to set a BASE URL, uncomment and set the following:
 # OPENAI_API_BASE=
 
+# Used to set the model for OpenAI API requests.
+# If not set, the default is gpt-4-turbo.
+# OPENAI_API_MODEL='gpt-4-turbo'
+
 # OpenAI API key retrieved here: https://platform.openai.com/api-keys
 OPENAI_API_KEY=
 

README.md

@@ -50,6 +50,10 @@ Your .env.local file should look like this:
 # If you need to set a BASE URL, uncomment and set the following:
 # OPENAI_API_BASE=
 
+# Used to set the model for OpenAI API requests.
+# If not set, the default is gpt-4-turbo.
+# OPENAI_API_MODEL='gpt-4-turbo'
+
 # OpenAI API key retrieved here: https://platform.openai.com/api-keys
 OPENAI_API_KEY=[YOUR_OPENAI_API_KEY]
 

lib/agents/inquire.tsx

@@ -18,7 +18,7 @@ export async function inquire(
 
   let finalInquiry: PartialInquiry = {}
   await experimental_streamObject({
-    model: openai.chat('gpt-4-turbo'),
+    model: openai.chat(process.env.OPENAI_API_MODEL || 'gpt-4-turbo'),
     system: `As a professional web researcher, your role is to deepen your understanding of the user's input by conducting further inquiries when necessary.
     After receiving an initial response from the user, carefully assess whether additional questions are absolutely essential to provide a comprehensive and accurate answer. Only proceed with further inquiries if the available information is insufficient or ambiguous.
 

lib/agents/query-suggestor.tsx

@@ -22,7 +22,7 @@ export async function querySuggestor(
   )
 
   await experimental_streamObject({
-    model: openai.chat('gpt-4-turbo'),
+    model: openai.chat(process.env.OPENAI_API_MODEL || 'gpt-4-turbo'),
     system: `As a professional web researcher, your task is to generate a set of three queries that explore the subject matter more deeply, building upon the initial query and the information uncovered in its search results.
 
     For instance, if the original query was "Starship's third test flight key milestones", your output should follow this format:

lib/agents/researcher.tsx

@@ -38,7 +38,7 @@ export async function researcher(
   )
 
   const result = await experimental_streamText({
-    model: openai.chat('gpt-4-turbo'),
+    model: openai.chat(process.env.OPENAI_API_MODEL || 'gpt-4-turbo'),
     maxTokens: 2500,
     system: `As a professional search expert, you possess the ability to search for any information on the web. 
     For each user query, utilize the search results to their fullest potential to provide additional information and assistance in your response.

lib/agents/task-manager.tsx

@@ -11,7 +11,7 @@ export async function taskManager(messages: ExperimentalMessage[]) {
   })
 
   const result = await experimental_generateObject({
-    model: openai.chat('gpt-3.5-turbo'),
+    model: openai.chat(process.env.OPENAI_API_MODEL || 'gpt-4-turbo'),
     system: `As a professional web researcher, your primary objective is to fully comprehend the user's query, conduct thorough web searches to gather the necessary information, and provide an appropriate response.
     To achieve this, you must first analyze the user's input and determine the optimal course of action. You have two options at your disposal:
     1. "proceed": If the provided information is sufficient to address the query effectively, choose this option to proceed with the research and formulate a response.