Add getFunctionsAsTools() to easily turn functions into tool messages

Author: Ran-Mewo

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

@@ -138,6 +138,16 @@ public List<ChatFunction> getFunctions() {
         return new ArrayList<>(FUNCTIONS.values());
     }
 
+    public List<ChatTool> getFunctionsAsTools() {
+        return FUNCTIONS.values().stream()
+                .map(function -> {
+                    ChatTool tool = new ChatTool();
+                    tool.setFunction(function);
+                    return tool;
+                })
+                .collect(Collectors.toList());
+    }
+
     public void setFunctions(List<ChatFunction> functions) {
         this.FUNCTIONS.clear();
         functions.forEach(f -> this.FUNCTIONS.put(f.getName(), f));

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

@@ -308,8 +308,6 @@ void createChatCompletionWithToolFunctions() {
                 .executor(Weather.class, w -> new WeatherResponse(w.location, w.unit, 25, "sunny"))
                 .build());
         final FunctionExecutor functionExecutor = new FunctionExecutor(functions);
-        final ChatTool tool = new ChatTool();
-        tool.setFunction(functionExecutor.getFunctions().get(0));
         final List<ChatMessage> messages = new ArrayList<>();
         final ChatMessage systemMessage = new ChatMessage(ChatMessageRole.SYSTEM.value(), "You are a helpful assistant.");
         final ChatMessage userMessage = new ChatMessage(ChatMessageRole.USER.value(), "What is the weather in Monterrey, Nuevo León?");
@@ -320,7 +318,7 @@ void createChatCompletionWithToolFunctions() {
                 .builder()
                 .model("gpt-3.5-turbo-0613")
                 .messages(messages)
-                .tools(List.of(tool))
+                .tools(functionExecutor.getFunctionsAsTools())
                 .toolChoice("auto")
                 .n(1)
                 .maxTokens(100)
@@ -364,7 +362,7 @@ void createChatCompletionWithToolFunctions() {
                 .builder()
                 .model("gpt-3.5-turbo-0613")
                 .messages(messages)
-                .tools(List.of(tool))
+                .tools(functionExecutor.getFunctionsAsTools())
                 .toolChoice("auto")
                 .n(1)
                 .maxTokens(100)