added support for functions

Author: deep-learning-dynamo

src/main/java/dev/ai4j/openai4j/chat/Function.java

@@ -90,24 +90,24 @@ public Builder parameters(Parameters parameters) {
         }
 
         @Experimental
-        public Builder addParameter(String name, Constraint... constraints) {
-            addOptionalParameter(name, constraints);
+        public Builder addParameter(String name, JsonSchemaProperty... jsonSchemaProperties) {
+            addOptionalParameter(name, jsonSchemaProperties);
             this.parameters.required().add(name);
             return this;
         }
 
         @Experimental
-        public Builder addOptionalParameter(String name, Constraint... constraints) {
+        public Builder addOptionalParameter(String name, JsonSchemaProperty... jsonSchemaProperties) {
             if (this.parameters == null) {
                 this.parameters = Parameters.builder().build();
             }
 
-            Map<String, Object> constraintMap = new HashMap<>();
-            for (Constraint constraint : constraints) {
-                constraintMap.put(constraint.key(), constraint.value());
+            Map<String, Object> jsonSchemaPropertiesMap = new HashMap<>();
+            for (JsonSchemaProperty jsonSchemaProperty : jsonSchemaProperties) {
+                jsonSchemaPropertiesMap.put(jsonSchemaProperty.key(), jsonSchemaProperty.value());
             }
 
-            this.parameters.properties().put(name, constraintMap);
+            this.parameters.properties().put(name, jsonSchemaPropertiesMap);
             return this;
         }
 

src/main/java/dev/ai4j/openai4j/chat/Constraint.java renamed to src/main/java/dev/ai4j/openai4j/chat/JsonSchemaProperty.java

@@ -5,19 +5,19 @@
 import java.util.Objects;
 
 @Experimental
-public class Constraint {
+public class JsonSchemaProperty {
 
-    public static final Constraint STRING = type("string");
-    public static final Constraint NUMBER = type("number");
-    public static final Constraint OBJECT = type("object");
-    public static final Constraint ARRAY = type("array");
-    public static final Constraint BOOLEAN = type("boolean");
-    public static final Constraint NULL = type("null");
+    public static final JsonSchemaProperty STRING = type("string");
+    public static final JsonSchemaProperty NUMBER = type("number");
+    public static final JsonSchemaProperty OBJECT = type("object");
+    public static final JsonSchemaProperty ARRAY = type("array");
+    public static final JsonSchemaProperty BOOLEAN = type("boolean");
+    public static final JsonSchemaProperty NULL = type("null");
 
     private final String key;
     private final Object value;
 
-    public Constraint(String key, Object value) {
+    public JsonSchemaProperty(String key, Object value) {
         this.key = key;
         this.value = value;
     }
@@ -33,11 +33,11 @@ public Object value() {
     @Override
     public boolean equals(Object another) {
         if (this == another) return true;
-        return another instanceof Constraint
-                && equalTo((Constraint) another);
+        return another instanceof JsonSchemaProperty
+                && equalTo((JsonSchemaProperty) another);
     }
 
-    private boolean equalTo(Constraint another) {
+    private boolean equalTo(JsonSchemaProperty another) {
         return Objects.equals(key, another.key)
                 && Objects.equals(value, another.value);
     }
@@ -52,39 +52,39 @@ public int hashCode() {
 
     @Override
     public String toString() {
-        return "Constraint{"
+        return "JsonSchemaProperty{"
                 + "key=" + key
                 + ", value=" + value
                 + "}";
     }
 
     @Experimental
-    public static Constraint from(String key, Object value) {
-        return new Constraint(key, value);
+    public static JsonSchemaProperty from(String key, Object value) {
+        return new JsonSchemaProperty(key, value);
     }
 
     @Experimental
-    public static Constraint constraint(String key, Object value) {
+    public static JsonSchemaProperty property(String key, Object value) {
         return from(key, value);
     }
 
     @Experimental
-    public static Constraint type(String value) {
+    public static JsonSchemaProperty type(String value) {
         return from("type", value);
     }
 
     @Experimental
-    public static Constraint description(String value) {
+    public static JsonSchemaProperty description(String value) {
         return from("description", value);
     }
 
     @Experimental
-    public static Constraint enums(String... enumValues) { // TODO names
+    public static JsonSchemaProperty enums(String... enumValues) {
         return from("enum", enumValues);
     }
 
     @Experimental
-    public static Constraint enums(Object... enumValues) { // TODO names
+    public static JsonSchemaProperty enums(Object... enumValues) {
         for (Object enumValue : enumValues) {
             if (!enumValue.getClass().isEnum()) {
                 throw new RuntimeException("Value " + enumValue.getClass().getName() + " should be enum");
@@ -95,7 +95,7 @@ public static Constraint enums(Object... enumValues) { // TODO names
     }
 
     @Experimental
-    public static Constraint enums(Class<?> enumClass) { // TODO names
+    public static JsonSchemaProperty enums(Class<?> enumClass) {
         if (!enumClass.isEnum()) {
             throw new RuntimeException("Class " + enumClass.getName() + " should be enum");
         }

src/test/java/dev/ai4j/openai4j/chat/ChatCompletionTest.java

@@ -10,7 +10,7 @@
 import java.util.Map;
 import java.util.stream.Stream;
 
-import static dev.ai4j.openai4j.chat.Constraint.*;
+import static dev.ai4j.openai4j.chat.JsonSchemaProperty.*;
 import static dev.ai4j.openai4j.chat.Message.functionMessage;
 import static dev.ai4j.openai4j.chat.Message.userMessage;
 import static dev.ai4j.openai4j.chat.Role.ASSISTANT;
@@ -116,7 +116,7 @@ void testFunctions() {
         assertThat(secondResponse.content()).contains("22");
     }
 
-    public static String getCurrentWeather(String location, Unit unit) { // TODO return POJO
+    public static String getCurrentWeather(String location, Unit unit) {
         System.out.println(location);
         System.out.println(unit);
         return "{ \"temperature\": 22, \"unit\": \"celsius\", \"description\": \"Sunny\" }";