Skip to content

Commit d3bf6e1

Browse files
committed
[Assistants] Support retrieve assistant
1 parent 260f7c0 commit d3bf6e1

6 files changed

Lines changed: 53 additions & 0 deletions

File tree

docs/docs/reference/openai/assistants.md

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,3 +141,34 @@ Returns:
141141
"has_more": false
142142
}
143143
```
144+
145+
### Retrieve assistant
146+
147+
---
148+
149+
Retrieves an assistant.
150+
151+
```java
152+
client.retrieveAssistant("asst_xv9N9dNXstuV8OVLElLqgV7U");
153+
```
154+
155+
Returns:
156+
157+
```json
158+
{
159+
"id": "asst_abc123",
160+
"object": "assistant",
161+
"created_at": 1698984975,
162+
"name": "Math Tutor",
163+
"description": null,
164+
"model": "gpt-4",
165+
"instructions": "You are a personal math tutor. When asked a question, write and run Python code to answer the question.",
166+
"tools": [
167+
{
168+
"type": "code_interpreter"
169+
}
170+
],
171+
"file_ids": [],
172+
"metadata": {}
173+
}
174+
```

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

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -266,4 +266,11 @@ Single<AssistantsFileEntity> fetchCreateAssistantFile(@Url String url,
266266
*/
267267
@GET
268268
Single<AssistantsFileResponse> fetchAssistantFiles(@Url String url);
269+
270+
/**
271+
* Retrieves an assistant.
272+
* 检索助手。
273+
*/
274+
@GET
275+
Single<AssistantsEntity> fetchRetrieveAssistant(@Url String url);
269276
}

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

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -303,6 +303,13 @@ public AssistantsFileResponse assistantsFiles(String assistantId)
303303
.blockingGet();
304304
}
305305

306+
public AssistantsEntity retrieveAssistant(String assistantId)
307+
{
308+
String url = String.format(ProviderUtils.getUrl(provider, UrlModel.FETCH_RETRIEVE_ASSISTANT), assistantId);
309+
return this.api.fetchRetrieveAssistant(url)
310+
.blockingGet();
311+
}
312+
306313
private ObjectMapper createObjectMapper()
307314
{
308315
ObjectMapper objectMapper = new ObjectMapper();

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,4 +22,5 @@ public enum UrlModel
2222
FETCH_FINE_TUNING_JOBS_CANCEL,
2323
FETCH_ASSISTANTS,
2424
FETCH_ASSISTANTS_FILES,
25+
FETCH_RETRIEVE_ASSISTANT
2526
}

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,7 @@ public static String getUrl(ProviderModel provider, UrlModel model)
5959
DEFAULT_PROVIDER.put(UrlModel.FETCH_FINE_TUNING_JOBS_CANCEL, "v1/fine_tuning/jobs/%s/cancel");
6060
DEFAULT_PROVIDER.put(UrlModel.FETCH_ASSISTANTS, "v1/assistants");
6161
DEFAULT_PROVIDER.put(UrlModel.FETCH_ASSISTANTS_FILES, "v1/assistants/%s/files");
62+
DEFAULT_PROVIDER.put(UrlModel.FETCH_RETRIEVE_ASSISTANT, "v1/assistants/%s");
6263

6364
AZURE_PROVIDER.put(UrlModel.FETCH_COMPLETIONS, "completions");
6465
AZURE_PROVIDER.put(UrlModel.FETCH_CHAT_COMPLETIONS, "chat/completions");

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

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,4 +53,10 @@ public void testAssistantFiles()
5353
{
5454
Assert.assertNotNull(client.assistantsFiles("asst_xv9N9dNXstuV8OVLElLqgV7U"));
5555
}
56+
57+
@Test
58+
public void testRetrieveAssistants()
59+
{
60+
Assert.assertNotNull(client.retrieveAssistant("asst_xv9N9dNXstuV8OVLElLqgV7U"));
61+
}
5662
}

0 commit comments

Comments
 (0)