xiongzhu 2 years ago
parent
commit
14ff782a0b
1 changed files with 6 additions and 6 deletions
  1. 6 6
      src/paper/paper-gen/llm.ts

+ 6 - 6
src/paper/paper-gen/llm.ts

@@ -12,13 +12,13 @@ require('dotenv').config()
 export function createLLM(model = 'gpt-3.5-turbo-16k') : LLMTools {
     const usage = { completionTokens: 0, promptTokens: 0, totalTokens: 0 }
     const llm = new ChatOpenAI({
-        openAIApiKey: process.env.OPENAI_API_KEY,
-        modelName: model,
+        // openAIApiKey: process.env.OPENAI_API_KEY,
+        // modelName: model,
 
-        // azureOpenAIApiKey: process.env.AZURE_OPENAI_KEY,
-        // azureOpenAIApiVersion: process.env.AZURE_OPENAI_VERSION,
-        // azureOpenAIApiInstanceName: process.env.AZURE_OPENAI_INSTANCE,
-        // azureOpenAIApiDeploymentName: process.env.AZURE_OPENAI_DEPLOYMENT,
+        azureOpenAIApiKey: process.env.AZURE_OPENAI_KEY,
+        azureOpenAIApiVersion: process.env.AZURE_OPENAI_VERSION,
+        azureOpenAIApiInstanceName: process.env.AZURE_OPENAI_INSTANCE,
+        azureOpenAIApiDeploymentName: 'gpt-35-turbo-16k',
 
         timeout: 1000 * 60 * 5,
         configuration: {