Skip to content

Commit 311a818

Browse files
committed
[Thread] Support retrieve thread
1 parent 32de9df commit 311a818

5 files changed

Lines changed: 24 additions & 1 deletion

File tree

src/main/java/org/devlive/sdk/openai/DefaultApi.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -311,4 +311,11 @@ Single<AssistantsEntity> fetchUpdateAssistant(@Url String url,
311311
@POST
312312
Single<ThreadEntity> fetchCreateThread(@Url String url,
313313
@Body ThreadEntity configure);
314+
315+
/**
316+
* Retrieves a thread.
317+
* 检索线程。
318+
*/
319+
@GET
320+
Single<ThreadEntity> fetchRetrieveThread(@Url String url);
314321
}

src/main/java/org/devlive/sdk/openai/DefaultClient.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -346,6 +346,13 @@ public ThreadEntity createThread(ThreadEntity configure)
346346
.blockingGet();
347347
}
348348

349+
public ThreadEntity retrieveThread(String threadId)
350+
{
351+
String url = String.format(ProviderUtils.getUrl(provider, UrlModel.FETCH_RETRIEVE_THREADS), threadId);
352+
return this.api.fetchRetrieveThread(url)
353+
.blockingGet();
354+
}
355+
349356
private ObjectMapper createObjectMapper()
350357
{
351358
ObjectMapper objectMapper = new ObjectMapper();

src/main/java/org/devlive/sdk/openai/model/UrlModel.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,5 +24,6 @@ public enum UrlModel
2424
FETCH_ASSISTANTS_FILES,
2525
FETCH_RETRIEVE_ASSISTANT,
2626
FETCH_RETRIEVE_ASSISTANT_FILE,
27-
FETCH_THREADS;
27+
FETCH_THREADS,
28+
FETCH_RETRIEVE_THREADS;
2829
}

src/main/java/org/devlive/sdk/openai/utils/ProviderUtils.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ public static String getUrl(ProviderModel provider, UrlModel model)
6262
DEFAULT_PROVIDER.put(UrlModel.FETCH_RETRIEVE_ASSISTANT, "v1/assistants/%s");
6363
DEFAULT_PROVIDER.put(UrlModel.FETCH_RETRIEVE_ASSISTANT_FILE, "v1/assistants/%s/files/%s");
6464
DEFAULT_PROVIDER.put(UrlModel.FETCH_THREADS, "v1/threads");
65+
DEFAULT_PROVIDER.put(UrlModel.FETCH_RETRIEVE_THREADS, "v1/threads/%s");
6566

6667
AZURE_PROVIDER.put(UrlModel.FETCH_COMPLETIONS, "completions");
6768
AZURE_PROVIDER.put(UrlModel.FETCH_CHAT_COMPLETIONS, "chat/completions");

src/test/java/org/devlive/sdk/openai/ThreadTest.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,4 +25,11 @@ public void testCreateThread()
2525
.build();
2626
Assert.assertNotNull(client.createThread(configure));
2727
}
28+
29+
@Test
30+
public void testRetrieveThread()
31+
{
32+
String threadId = "thread_lsfBRIATCECds5WYLVXnVcpU";
33+
Assert.assertNotNull(client.retrieveThread(threadId));
34+
}
2835
}

0 commit comments

Comments
 (0)