Merge pull request #19 from didalgolab/assistants-api

Assistants API

README.md

@@ -25,7 +25,7 @@ Full support for all of the OpenAI services, including the latest changes announ
 * Embeddings
 * Fine tuning
 
-NOTE: Beta services are not included yet.
+NOTE: Includes the Beta Assistants API.
 
 ![Services](media/supported_services.png)
 

pom.xml

@@ -6,7 +6,7 @@
 
   <groupId>io.github.sashirestela</groupId>
   <artifactId>simple-openai</artifactId>
-  <version>1.1.0</version>
+  <version>1.2.0</version>
   <packaging>jar</packaging>
 
   <name>simple-openai</name>
@@ -52,7 +52,7 @@
     <maven.compiler.release>11</maven.compiler.release>
     <!-- Dependencies Versions -->
     <slf4j.version>[2.0.9,3.0.0)</slf4j.version>
-    <cleverclient.version>0.9.0</cleverclient.version>
+    <cleverclient.version>0.10.0</cleverclient.version>
     <lombok.version>[1.18.30,2.0.0)</lombok.version>
     <jackson.version>[2.15.2,3.0.0)</jackson.version>
     <json.schema.version>[4.31.1,5.0.0)</json.schema.version>

src/demo/java/io/github/sashirestela/openai/demo/AssistantServiceDemo.java

@@ -0,0 +1,144 @@
+package io.github.sashirestela.openai.demo;
+
+import io.github.sashirestela.openai.domain.assistant.AssistantRequest;
+import io.github.sashirestela.openai.domain.assistant.AssistantTool;
+import io.github.sashirestela.openai.domain.assistant.ImageFileContent;
+import io.github.sashirestela.openai.domain.assistant.ThreadMessage;
+import io.github.sashirestela.openai.domain.assistant.ThreadMessageRequest;
+import io.github.sashirestela.openai.domain.assistant.ThreadRequest;
+import io.github.sashirestela.openai.domain.assistant.ThreadRun;
+import io.github.sashirestela.openai.domain.file.FileRequest;
+import io.github.sashirestela.openai.domain.file.PurposeType;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+import java.util.List;
+import java.util.concurrent.atomic.AtomicInteger;
+
+public class AssistantServiceDemo extends AbstractDemo {
+
+    String assistantId;
+    String fileId;
+    String threadId;
+    String runId;
+
+    public void demoCreateAssistant() {
+        var assistantRequest = AssistantRequest.builder()
+                .model("gpt-3.5-turbo-1106")
+                .build();
+
+        var assistant = openAI.assistants().create(assistantRequest).join();
+        System.out.println(assistant);
+        assistantId = assistant.getId();
+    }
+
+    public void demoRetrieveAndModifyAssistant() {
+        var assistant = openAI.assistants().getOne(assistantId).join();
+        var assistantRequest = assistant.mutate()
+                .name("Math Expert")
+                .instructions("You are a personal math expert. When asked a question, write and run Python code to answer the question.")
+                .tool(AssistantTool.CODE_INTERPRETER)
+                .build();
+
+        assistant = openAI.assistants().modify(assistant.getId(), assistantRequest).join();
+        System.out.println(assistant);
+    }
+
+    public void demoListAssistants() {
+        AtomicInteger count = new AtomicInteger();
+        openAI.assistants().getList()
+                .join()
+                .forEach(r -> System.out.println("\n#"+count.incrementAndGet()+"\n" + r));
+    }
+
+    public void demoUploadAssistantFile() {
+        var fileRequest = FileRequest.builder()
+                .file(Paths.get("src/demo/resources/code_interpreter_file.txt"))
+                .purpose(PurposeType.ASSISTANTS)
+                .build();
+        var file = openAI.files().create(fileRequest).join();
+        var assistantFile = openAI.assistants().createFile(assistantId, file.getId()).join();
+        System.out.println(assistantFile);
+        fileId = file.getId();
+    }
+
+    public void demoCreateThread() {
+        var threadRequest = ThreadRequest.builder()
+                .message(ThreadMessageRequest.builder()
+                        .role("user")
+                        .content("Inspect the content of the attached text file. After that plot graph of the formula requested in it.")
+                        .build())
+                .build();
+
+        var thread = openAI.threads().create(threadRequest).join();
+        System.out.println(thread);
+        threadId = thread.getId();
+    }
+
+    public void demoRunThreadAndWaitUntilComplete() {
+        var run = openAI.threads().createRun(threadId, assistantId).join();
+        runId = run.getId();
+
+        while (!run.getStatus().equals(ThreadRun.Status.COMPLETED)) {
+            sleep(1);
+            run = openAI.threads().getRun(run.getThreadId(), run.getId()).join();
+        }
+        System.out.println(run);
+
+        var messages = openAI.threads().getMessageList(threadId).join();
+        System.out.println(messages);
+    }
+
+    public void demoGetAssistantMessages() {
+        List<ThreadMessage> messages = openAI.threads().getMessageList(threadId).join();
+        ThreadMessage assistant = messages.get(0);
+        ImageFileContent assistantImageContent = assistant.getContent().stream()
+                .filter(ImageFileContent.class::isInstance)
+                .map(ImageFileContent.class::cast)
+                .findFirst().orElse(null);
+
+        System.out.println("All messages:");
+        System.out.println("=============");
+        System.out.println(messages);
+
+        if (assistantImageContent != null) {
+            System.out.println("\nAssistant answer contains an image. Downloading it now...");
+            try (var in = openAI.files().getContentInputStream(assistantImageContent.getImageFile().getFileId()).join()) {
+                Path tempFile = Files.createTempFile("code_interpreter", ".png");
+                Files.write(tempFile, in.readAllBytes());
+                System.out.println("Image file downloaded to: " + tempFile.toUri());
+            } catch (IOException e) {
+                throw new RuntimeException(e);
+            }
+        }
+    }
+
+    public void demoDeleteAssistant() {
+        openAI.assistants().delete(assistantId).join();
+    }
+
+    private static void sleep(int seconds) {
+        try {
+            java.lang.Thread.sleep(1000L * seconds);
+        } catch (InterruptedException e) {
+            throw new RuntimeException(e);
+        }
+    }
+
+    public static void main(String[] args) {
+        var demo = new AssistantServiceDemo();
+
+        demo.addTitleAction("Demo Call Assistant Create", demo::demoCreateAssistant);
+        demo.addTitleAction("Demo Call Assistant Retrieve and Modify", demo::demoRetrieveAndModifyAssistant);
+        demo.addTitleAction("Demo Call Assistant List", demo::demoListAssistants);
+        demo.addTitleAction("Demo Call Assistant File Upload", demo::demoUploadAssistantFile);
+        demo.addTitleAction("Demo Call Assistant Thread Create", demo::demoCreateThread);
+        demo.addTitleAction("Demo Call Assistant Thread Run", demo::demoRunThreadAndWaitUntilComplete);
+        demo.addTitleAction("Demo Call Assistant Messages Get", demo::demoGetAssistantMessages);
+        demo.addTitleAction("Demo Call Assistant Delete", demo::demoDeleteAssistant);
+
+        demo.run();
+    }
+}

src/demo/resources/code_interpreter_file.txt

@@ -0,0 +1 @@
+y := sin(x)