Skip to content

Commit e641282

Browse files
committed
[Assistants] Support retrieve assistant file
1 parent d3bf6e1 commit e641282

6 files changed

Lines changed: 51 additions & 4 deletions

File tree

docs/docs/reference/openai/assistants.md

Lines changed: 28 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,9 @@ AssistantsEntity entity = AssistantsEntity.builder()
2727
.model(CompletionModel.GPT_35_TURBO)
2828
.instructions("You are a personal math tutor. When asked a question, write and run Python code to answer the question.")
2929
.build();
30-
client.createAssistants(entity);
30+
client.
31+
32+
createAssistants(entity);
3133
```
3234

3335
Returns:
@@ -58,7 +60,7 @@ Returns:
5860
Create an assistant file by attaching a File to an assistant.
5961

6062
```java
61-
client.createAssistantsFile("file-jNuKdx61rNQ0FUhuPFpMNmGZ", "asst_xv9N9dNXstuV8OVLElLqgV7U")
63+
client.createAssistantsFile("file-jNuKdx61rNQ0FUhuPFpMNmGZ","asst_xv9N9dNXstuV8OVLElLqgV7U")
6264
```
6365

6466
Returns:
@@ -85,7 +87,9 @@ client.listAssistants(null);
8587
QueryEntity configure = QueryEntity.builder()
8688
.limit(2)
8789
.build();
88-
client.assistants(configure);
90+
client.
91+
92+
assistants(configure);
8993
```
9094

9195
Returns:
@@ -172,3 +176,24 @@ Returns:
172176
"metadata": {}
173177
}
174178
```
179+
180+
### Retrieve assistant file
181+
182+
---
183+
184+
Retrieves an assistant file.
185+
186+
```java
187+
client.retrieveAssistantFile("asst_xv9N9dNXstuV8OVLElLqgV7U","file-jNuKdx61rNQ0FUhuPFpMNmGZ");
188+
```
189+
190+
Returns:
191+
192+
```json
193+
{
194+
"id": "file-abc123",
195+
"object": "assistant.file",
196+
"created_at": 1699055364,
197+
"assistant_id": "asst_abc123"
198+
}
199+
```

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

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -273,4 +273,11 @@ Single<AssistantsFileEntity> fetchCreateAssistantFile(@Url String url,
273273
*/
274274
@GET
275275
Single<AssistantsEntity> fetchRetrieveAssistant(@Url String url);
276+
277+
/**
278+
* Retrieves an AssistantFile.
279+
* 检索助手文件。
280+
*/
281+
@GET
282+
Single<AssistantsFileEntity> fetchRetrieveAssistantFile(@Url String url);
276283
}

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

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -310,6 +310,13 @@ public AssistantsEntity retrieveAssistant(String assistantId)
310310
.blockingGet();
311311
}
312312

313+
public AssistantsFileEntity retrieveAssistantFile(String assistantId, String fileId)
314+
{
315+
String url = String.format(ProviderUtils.getUrl(provider, UrlModel.FETCH_RETRIEVE_ASSISTANT_FILE), assistantId, fileId);
316+
return this.api.fetchRetrieveAssistantFile(url)
317+
.blockingGet();
318+
}
319+
313320
private ObjectMapper createObjectMapper()
314321
{
315322
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
@@ -22,5 +22,6 @@ public enum UrlModel
2222
FETCH_FINE_TUNING_JOBS_CANCEL,
2323
FETCH_ASSISTANTS,
2424
FETCH_ASSISTANTS_FILES,
25-
FETCH_RETRIEVE_ASSISTANT
25+
FETCH_RETRIEVE_ASSISTANT,
26+
FETCH_RETRIEVE_ASSISTANT_FILE;
2627
}

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

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ public static String getUrl(ProviderModel provider, UrlModel model)
6060
DEFAULT_PROVIDER.put(UrlModel.FETCH_ASSISTANTS, "v1/assistants");
6161
DEFAULT_PROVIDER.put(UrlModel.FETCH_ASSISTANTS_FILES, "v1/assistants/%s/files");
6262
DEFAULT_PROVIDER.put(UrlModel.FETCH_RETRIEVE_ASSISTANT, "v1/assistants/%s");
63+
DEFAULT_PROVIDER.put(UrlModel.FETCH_RETRIEVE_ASSISTANT_FILE, "v1/assistants/%s/files/%s");
6364

6465
AZURE_PROVIDER.put(UrlModel.FETCH_COMPLETIONS, "completions");
6566
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
@@ -59,4 +59,10 @@ public void testRetrieveAssistants()
5959
{
6060
Assert.assertNotNull(client.retrieveAssistant("asst_xv9N9dNXstuV8OVLElLqgV7U"));
6161
}
62+
63+
@Test
64+
public void testRetrieveAssistantsFile()
65+
{
66+
Assert.assertNotNull(client.retrieveAssistantFile("asst_xv9N9dNXstuV8OVLElLqgV7U", "file-jNuKdx61rNQ0FUhuPFpMNmGZ"));
67+
}
6268
}

0 commit comments

Comments
 (0)