Merge pull request #7 from muverystrong/feat/put-get-level

feat: task put protocol add getQueueLevel method

Author: nil4u

api/pom.xml

@@ -6,7 +6,7 @@
     <parent>
         <groupId>top.bella</groupId>
         <artifactId>openai-java</artifactId>
-        <version>0.23.44</version>
+        <version>0.23.45</version>
     </parent>
     <packaging>jar</packaging>
     <artifactId>openai-api</artifactId>

api/src/main/java/com/theokanning/openai/queue/Put.java

@@ -8,6 +8,7 @@
 import lombok.NoArgsConstructor;
 
 import java.util.Map;
+import java.util.Optional;
 
 /**
  * Represents a queue put operation for task queuing
@@ -63,4 +64,36 @@ public class Put {
     public String getFullQueueName() {
         return String.format("%s:%d", queue, level);
     }
+
+    /**
+     * Determines the effective queue level based on the response mode and level
+     *
+     * @return the effective queue level
+     */
+    @JsonIgnore
+    public Integer getQueueLevel() {
+        if("blocking".equals(responseMode) || "streaming".equals(responseMode)) {
+            return 0;
+        } else if("callback".equals(responseMode) || "batch".equals(responseMode)) {
+            return getLevel() != null && getLevel() >= 0 ? getLevel() : 1;
+        } else {
+            throw new IllegalArgumentException("Unsupported response mode: " + responseMode);
+        }
+    }
+
+    /**
+     * Determines the task timeout based on the response mode and specified timeout
+     *
+     * @return the task timeout in seconds
+     */
+    @JsonIgnore
+    public int getTaskTimeout() {
+        if("blocking".equals(responseMode) || "streaming".equals(responseMode)) {
+            return timeout > 0 ? Math.min(timeout, 300) : 300;
+        } else if("callback".equals(responseMode) || "batch".equals(responseMode)) {
+            return timeout > 0 ? timeout : 24 * 60 * 60;
+        } else {
+            throw new IllegalArgumentException("Unsupported response mode: " + responseMode);
+        }
+    }
 }

client/pom.xml

@@ -6,7 +6,7 @@
     <parent>
         <groupId>top.bella</groupId>
         <artifactId>openai-java</artifactId>
-        <version>0.23.44</version>
+        <version>0.23.45</version>
     </parent>
     <packaging>jar</packaging>
 

pom.xml

@@ -5,7 +5,7 @@
 
     <groupId>top.bella</groupId>
     <artifactId>openai-java</artifactId>
-    <version>0.23.44</version>
+    <version>0.23.45</version>
     <packaging>pom</packaging>
     <description>openai java 版本</description>
     <name>openai-java</name>

service/pom.xml

@@ -6,7 +6,7 @@
     <parent>
         <groupId>top.bella</groupId>
         <artifactId>openai-java</artifactId>
-        <version>0.23.44</version>
+        <version>0.23.45</version>
     </parent>
     <packaging>jar</packaging>