[Feature][Java] Add OpenAI chat model integration (#320)

Author: xli1996

e2e-test/flink-agents-end-to-end-tests-integration/pom.xml

@@ -64,6 +64,11 @@ under the License.
             <artifactId>flink-agents-integrations-chat-models-azureai</artifactId>
             <version>${project.version}</version>
         </dependency>
+        <dependency>
+            <groupId>org.apache.flink</groupId>
+            <artifactId>flink-agents-integrations-chat-models-openai</artifactId>
+            <version>${project.version}</version>
+        </dependency>
         <dependency>
             <groupId>org.apache.flink</groupId>
             <artifactId>flink-agents-integrations-chat-models-ollama</artifactId>
@@ -76,4 +81,4 @@ under the License.
         </dependency>
     </dependencies>
 
-</project>
+</project>

e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ChatModelIntegrationAgent.java

@@ -37,6 +37,8 @@
 import org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelSetup;
 import org.apache.flink.agents.integrations.chatmodels.ollama.OllamaChatModelConnection;
 import org.apache.flink.agents.integrations.chatmodels.ollama.OllamaChatModelSetup;
+import org.apache.flink.agents.integrations.chatmodels.openai.OpenAIChatModelConnection;
+import org.apache.flink.agents.integrations.chatmodels.openai.OpenAIChatModelSetup;
 
 import java.util.Collections;
 import java.util.List;
@@ -80,6 +82,11 @@ public static ResourceDescriptor chatModelConnection() {
                     .addInitialArgument("endpoint", endpoint)
                     .addInitialArgument("apiKey", apiKey)
                     .build();
+        } else if (provider.equals("OPENAI")) {
+            String apiKey = System.getenv().get("OPENAI_API_KEY");
+            return ResourceDescriptor.Builder.newBuilder(OpenAIChatModelConnection.class.getName())
+                    .addInitialArgument("api_key", apiKey)
+                    .build();
         } else {
             throw new RuntimeException(String.format("Unknown model provider %s", provider));
         }
@@ -105,6 +112,14 @@ public static ResourceDescriptor chatModel() {
                             "tools",
                             List.of("calculateBMI", "convertTemperature", "createRandomNumber"))
                     .build();
+        } else if (provider.equals("OPENAI")) {
+            return ResourceDescriptor.Builder.newBuilder(OpenAIChatModelSetup.class.getName())
+                    .addInitialArgument("connection", "chatModelConnection")
+                    .addInitialArgument("model", "gpt-4o-mini")
+                    .addInitialArgument(
+                            "tools",
+                            List.of("calculateBMI", "convertTemperature", "createRandomNumber"))
+                    .build();
         } else {
             throw new RuntimeException(String.format("Unknown model provider %s", provider));
         }

e2e-test/flink-agents-end-to-end-tests-integration/src/test/java/org/apache/flink/agents/integration/test/ChatModelIntegrationTest.java

@@ -52,7 +52,7 @@ public ChatModelIntegrationTest() throws IOException {
     }
 
     @ParameterizedTest()
-    @ValueSource(strings = {"OLLAMA", "AZURE"})
+    @ValueSource(strings = {"OLLAMA", "AZURE", "OPENAI"})
     public void testChatModeIntegration(String provider) throws Exception {
         Assumptions.assumeTrue(
                 (OLLAMA.equals(provider) && ollamaReady)

integrations/chat-models/openai/pom.xml

@@ -0,0 +1,54 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<!--
+Licensed to the Apache Software Foundation (ASF) under one
+or more contributor license agreements.  See the NOTICE file
+distributed with this work for additional information
+regarding copyright ownership.  The ASF licenses this file
+to you under the Apache License, Version 2.0 (the
+"License"); you may not use this file except in compliance
+with the License.  You may obtain a copy of the License at
+  http://www.apache.org/licenses/LICENSE-2.0
+Unless required by applicable law or agreed to in writing,
+software distributed under the License is distributed on an
+"AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+KIND, either express or implied.  See the License for the
+specific language governing permissions and limitations
+under the License.
+-->
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <parent>
+        <groupId>org.apache.flink</groupId>
+        <artifactId>flink-agents-integrations-chat-models</artifactId>
+        <version>0.2-SNAPSHOT</version>
+        <relativePath>../pom.xml</relativePath>
+    </parent>
+
+    <artifactId>flink-agents-integrations-chat-models-openai</artifactId>
+    <name>Flink Agents : Integrations: Chat Models: OpenAI</name>
+    <packaging>jar</packaging>
+
+    <dependencies>
+        <dependency>
+            <groupId>org.apache.flink</groupId>
+            <artifactId>flink-agents-api</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+        <dependency>
+            <groupId>org.apache.flink</groupId>
+            <artifactId>flink-agents-plan</artifactId>
+            <version>${project.version}</version>
+        </dependency>
+
+        <dependency>
+            <groupId>com.openai</groupId>
+            <artifactId>openai-java</artifactId>
+            <version>${openai.version}</version>
+        </dependency>
+    </dependencies>
+
+</project>
+