polish openAIKey

rajadilipkolli · Jan 3, 2025 · b670292 · b670292
1 parent dbd65ad
commit b670292
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -1,5 +1,12 @@
 {
     "configurations": [
+        {
+            "type": "java",
+            "name": "LangChain4JAIServicesDemo",
+            "request": "launch",
+            "mainClass": "com.learning.openai.LangChain4JAIServicesDemo",
+            "projectName": "playground-langchain4j"
+        },
         {
             "type": "java",
             "name": "LangChain4JObservabilityDemo",

diff --git a/playground-langchain4j/src/main/java/com/learning/openai/LangChain4JAIServicesDemo.java b/playground-langchain4j/src/main/java/com/learning/openai/LangChain4JAIServicesDemo.java
@@ -8,15 +8,16 @@
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
+import java.util.Optional;
+
 public class LangChain4JAIServicesDemo {
 
     private static final Logger LOGGER = LoggerFactory.getLogger(LangChain4JAIServicesDemo.class);
 
     public static void main(String[] args) {
-        //        String openAIKey = System.getenv("OPEN_AI_KEY");
-        //        OpenAiChatModel openAiChatModel = OpenAiChatModel.withApiKey("demo");
+        String openAIKey = System.getenv("OPEN_AI_KEY");
         OpenAiChatModel openAiChatModel = OpenAiChatModel.builder()
-                .apiKey("demo")
+                .apiKey(Optional.ofNullable(openAIKey).orElse("demo"))
                 .modelName(OpenAiChatModelName.GPT_4_O_MINI)
                 .logRequests(false)
                 .logResponses(false)