Skip to content

Commit e40552f

Browse files
muverystrongfengyizhu
authored andcommitted
feat:add query queue task detail method
1 parent 23f234d commit e40552f

File tree

8 files changed

+47
-6
lines changed

8 files changed

+47
-6
lines changed

api/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
<parent>
77
<groupId>top.bella</groupId>
88
<artifactId>openai-java</artifactId>
9-
<version>0.24.3</version>
9+
<version>0.24.4</version>
1010
</parent>
1111
<packaging>jar</packaging>
1212
<artifactId>openai-api</artifactId>

client/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
<parent>
77
<groupId>top.bella</groupId>
88
<artifactId>openai-java</artifactId>
9-
<version>0.24.3</version>
9+
<version>0.24.4</version>
1010
</parent>
1111
<packaging>jar</packaging>
1212

client/src/main/java/com/theokanning/openai/client/OpenAiApi.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -412,6 +412,9 @@ public interface OpenAiApi {
412412
@POST("queue/{task_id}/complete")
413413
Single<String> completeTask(@Path("task_id") String taskId, @Body Map<String, Object> data);
414414

415+
@GET("queue/{task_id}")
416+
Single<Task> getTask(@Path("task_id") String taskId);
417+
415418
@POST("web/search")
416419
Single<WebSearchResponse> webSearch(@Body WebSearchRequest request);
417420

example/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66

77
<groupId>io.github.ke</groupId>
88
<artifactId>example</artifactId>
9-
<version>0.23.93</version>
9+
<version>0.24.4</version>
1010
<name>example</name>
1111

1212
<properties>

pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
<groupId>top.bella</groupId>
77
<artifactId>openai-java</artifactId>
8-
<version>0.24.3</version>
8+
<version>0.24.4</version>
99
<packaging>pom</packaging>
1010
<description>openai java 版本</description>
1111
<name>openai-java</name>

service/pom.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
<parent>
77
<groupId>top.bella</groupId>
88
<artifactId>openai-java</artifactId>
9-
<version>0.24.3</version>
9+
<version>0.24.4</version>
1010
</parent>
1111
<packaging>jar</packaging>
1212

service/src/main/java/com/theokanning/openai/service/OpenAiService.java

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -930,6 +930,10 @@ public String completeTask(String taskId, Map<String, Object> data) {
930930
return execute(api.completeTask(taskId, data));
931931
}
932932

933+
public Task getTask(String taskId) {
934+
return execute(api.getTask(taskId));
935+
}
936+
933937
public static OpenAiApi buildApi(String token, Duration timeout, String baseUrl) {
934938
OkHttpClient client = defaultClient(token, timeout);
935939
Retrofit retrofit = defaultRetrofit(client, mapper, baseUrl);

service/src/test/java/com/theokanning/openai/service/QueueTest.java

Lines changed: 35 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,19 @@
44
import com.theokanning.openai.queue.Register;
55
import com.theokanning.openai.queue.Take;
66
import com.theokanning.openai.queue.Task;
7+
import org.junit.jupiter.api.BeforeEach;
78
import org.junit.jupiter.api.Disabled;
89
import org.junit.jupiter.api.Test;
10+
import org.mockito.Mock;
11+
import org.mockito.MockitoAnnotations;
912

1013
import java.util.Arrays;
1114
import java.util.HashMap;
1215
import java.util.List;
1316
import java.util.Map;
1417

1518
import static org.junit.jupiter.api.Assertions.*;
19+
import static org.mockito.Mockito.when;
1620

1721
/**
1822
* Queue operation tests Note: These tests are disabled as they require a custom queue service backend that implements the /v1/queue/* endpoints,
@@ -21,7 +25,13 @@
2125
@Disabled("Queue endpoints require custom backend service")
2226
public class QueueTest {
2327

24-
com.theokanning.openai.service.OpenAiService service = new OpenAiService();
28+
@Mock
29+
private OpenAiService service;
30+
31+
@BeforeEach
32+
void setUp() {
33+
MockitoAnnotations.openMocks(this);
34+
}
2535

2636
@Test
2737
void registerQueue() {
@@ -90,4 +100,28 @@ void completeTask() {
90100

91101
assertNotNull(result);
92102
}
103+
104+
@Test
105+
void getTask() {
106+
String taskId = "test-task-id-123";
107+
108+
// Create a mock Task object
109+
Task expectedTask = Task.builder()
110+
.taskId(taskId)
111+
.queue("test-queue")
112+
.status("pending")
113+
.build();
114+
115+
// Mock the service call
116+
when(service.getTask(taskId)).thenReturn(expectedTask);
117+
118+
// Execute the test
119+
Task result = service.getTask(taskId);
120+
121+
// Verify the results
122+
assertNotNull(result);
123+
assertEquals(taskId, result.getTaskId());
124+
assertEquals("test-queue", result.getQueue());
125+
assertEquals("pending", result.getStatus());
126+
}
93127
}

0 commit comments

Comments
 (0)