diff --git a/API/src/main/java/me/zacharias/chat/api/APITool.java b/API/src/main/java/me/zacharias/chat/api/APITool.java
index 2417f4b..480097e 100644
--- a/API/src/main/java/me/zacharias/chat/api/APITool.java
+++ b/API/src/main/java/me/zacharias/chat/api/APITool.java
@@ -8,6 +8,8 @@ import me.zacharias.chat.ollama.OllamaFunctionTool;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
+import org.jetbrains.annotations.NotNull;
+import org.jspecify.annotations.NonNull;
import org.springframework.http.ResponseEntity;
import org.springframework.web.client.RestTemplate;
import org.springframework.web.util.UriComponentsBuilder;
@@ -45,7 +47,7 @@ public class APITool extends OllamaFunctionTool {
}
@Override
- public String name() {
+ public @NonNull String name() {
return name;
}
@@ -55,7 +57,7 @@ public class APITool extends OllamaFunctionTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
OllamaPerameter.OllamaPerameterBuilder parameter = OllamaPerameter.builder();
for (ToolArgument argument : arguments) {
parameter.addProperty(argument.getName(), argument.getType(), argument.getDescription(), argument.isRequired());
diff --git a/Core/src/main/java/me/zacharias/chat/core/files/tools/ReadFileTool.java b/Core/src/main/java/me/zacharias/chat/core/files/tools/ReadFileTool.java
index 9b46c37..ab58576 100644
--- a/Core/src/main/java/me/zacharias/chat/core/files/tools/ReadFileTool.java
+++ b/Core/src/main/java/me/zacharias/chat/core/files/tools/ReadFileTool.java
@@ -6,6 +6,7 @@ import me.zacharias.chat.ollama.OllamaFunctionTool;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
+import org.jetbrains.annotations.NotNull;
import java.io.*;
import java.nio.file.Path;
@@ -13,7 +14,7 @@ import java.nio.file.Path;
public class ReadFileTool extends OllamaFunctionTool {
FileHandler fs = FileHandler.getInstance();
@Override
- public String name() {
+ public @NotNull String name() {
return "read_file";
}
@@ -23,7 +24,7 @@ public class ReadFileTool extends OllamaFunctionTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("file_path", OllamaPerameter.OllamaPerameterBuilder.Type.STRING, "The path to the file to be read")
.build();
diff --git a/Core/src/main/java/me/zacharias/chat/core/memory/AddMemoryFunction.java b/Core/src/main/java/me/zacharias/chat/core/memory/AddMemoryFunction.java
index 5e33657..66562c8 100644
--- a/Core/src/main/java/me/zacharias/chat/core/memory/AddMemoryFunction.java
+++ b/Core/src/main/java/me/zacharias/chat/core/memory/AddMemoryFunction.java
@@ -5,6 +5,7 @@ import me.zacharias.chat.ollama.OllamaFunctionTool;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
+import org.jetbrains.annotations.NotNull;
/**
* Provides the add_memory function.
@@ -17,7 +18,7 @@ public class AddMemoryFunction extends OllamaFunctionTool {
CoreMemory memory = CoreMemory.getInstance();
@Override
- public String name() {
+ public @NotNull String name() {
return "add_memory";
}
@@ -27,7 +28,7 @@ public class AddMemoryFunction extends OllamaFunctionTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("memory", OllamaPerameter.OllamaPerameterBuilder.Type.STRING, "The memory to remember", true)
.addProperty("identity", OllamaPerameter.OllamaPerameterBuilder.Type.STRING, "The identity of the memory to remember", true)
diff --git a/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoriesFunction.java b/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoriesFunction.java
index 804bdf1..7d0bda1 100644
--- a/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoriesFunction.java
+++ b/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoriesFunction.java
@@ -4,10 +4,11 @@ import me.zacharias.chat.ollama.OllamaFunctionArgument;
import me.zacharias.chat.ollama.OllamaFunctionTool;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
+import org.jetbrains.annotations.NotNull;
public class GetMemoriesFunction extends OllamaFunctionTool {
@Override
- public String name() {
+ public @NotNull String name() {
return "get_memories";
}
@@ -17,7 +18,7 @@ public class GetMemoriesFunction extends OllamaFunctionTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return null;
}
diff --git a/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoryFunction.java b/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoryFunction.java
index b118aea..6253259 100644
--- a/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoryFunction.java
+++ b/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoryFunction.java
@@ -4,8 +4,7 @@ import me.zacharias.chat.ollama.OllamaFunctionArgument;
import me.zacharias.chat.ollama.OllamaFunctionTool;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
-import org.json.JSONArray;
-import org.json.JSONObject;
+import org.jetbrains.annotations.NotNull;
/**
* Provides the get_memory function.
@@ -18,7 +17,7 @@ public class GetMemoryFunction extends OllamaFunctionTool {
CoreMemory memory = CoreMemory.getInstance();
@Override
- public String name() {
+ public @NotNull String name() {
return "get_memory";
}
@@ -28,7 +27,7 @@ public class GetMemoryFunction extends OllamaFunctionTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("identity", OllamaPerameter.OllamaPerameterBuilder.Type.STRING, "The identity of the memory to retrieve", true)
.build();
diff --git a/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoryIdentitiesFunction.java b/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoryIdentitiesFunction.java
index 50eae58..0c11a97 100644
--- a/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoryIdentitiesFunction.java
+++ b/Core/src/main/java/me/zacharias/chat/core/memory/GetMemoryIdentitiesFunction.java
@@ -4,13 +4,14 @@ import me.zacharias.chat.ollama.OllamaFunctionArgument;
import me.zacharias.chat.ollama.OllamaFunctionTool;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
+import org.jetbrains.annotations.NotNull;
import org.json.JSONArray;
public class GetMemoryIdentitiesFunction extends OllamaFunctionTool {
CoreMemory memory = CoreMemory.getInstance();
@Override
- public String name() {
+ public @NotNull String name() {
return "get_memory_identities";
}
@@ -20,7 +21,7 @@ public class GetMemoryIdentitiesFunction extends OllamaFunctionTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return null;
}
diff --git a/Core/src/main/java/me/zacharias/chat/core/memory/RemoveMemoryFunction.java b/Core/src/main/java/me/zacharias/chat/core/memory/RemoveMemoryFunction.java
index 6d6d61b..bf31984 100644
--- a/Core/src/main/java/me/zacharias/chat/core/memory/RemoveMemoryFunction.java
+++ b/Core/src/main/java/me/zacharias/chat/core/memory/RemoveMemoryFunction.java
@@ -5,6 +5,7 @@ import me.zacharias.chat.ollama.OllamaFunctionTool;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
+import org.jetbrains.annotations.NotNull;
/**
* Provides the remove_memory function.
@@ -17,7 +18,7 @@ public class RemoveMemoryFunction extends OllamaFunctionTool {
CoreMemory memory = CoreMemory.getInstance();
@Override
- public String name() {
+ public @NotNull String name() {
return "remove_memory";
}
@@ -27,7 +28,7 @@ public class RemoveMemoryFunction extends OllamaFunctionTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("identity", OllamaPerameter.OllamaPerameterBuilder.Type.STRING, "The identity of the memory to forget", true)
.build();
diff --git a/Core/src/main/java/me/zacharias/chat/ollama/OllamaFunctionTool.java b/Core/src/main/java/me/zacharias/chat/ollama/OllamaFunctionTool.java
index 6cd7c0f..e096c32 100644
--- a/Core/src/main/java/me/zacharias/chat/ollama/OllamaFunctionTool.java
+++ b/Core/src/main/java/me/zacharias/chat/ollama/OllamaFunctionTool.java
@@ -2,6 +2,7 @@ package me.zacharias.chat.ollama;
import me.zacharias.chat.core.Core;
import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
+import org.jetbrains.annotations.NotNull;
import org.json.JSONObject;
/**
@@ -9,16 +10,22 @@ import org.json.JSONObject;
*/
public abstract class OllamaFunctionTool implements OllamaTool {
+ /**
+ * This field is set via Reflection injection from {@link OllamaObject#addTool(OllamaTool, String)}.
+ * As is, it will be over written. Do not set it yourself.
+ */
protected String source = "";
@Override
public String toString() {
JSONObject ret = new JSONObject();
- ret.put("tool", "function");
+ ret.put("type", "function");
JSONObject function = new JSONObject();
function.put("name", name()+(source != null ? source : ""));
- function.put("description", description());
+ if(description() != null) {
+ function.put("description", description());
+ }
function.put("parameters", (parameters() == null?
new JSONObject() : new JSONObject(parameters().toString())));
@@ -32,6 +39,7 @@ public abstract class OllamaFunctionTool implements OllamaTool {
* This is used by Ollama to know what the tool is
* @return The name of the tool
*/
+ @NotNull
abstract public String name();
/**
@@ -39,14 +47,18 @@ public abstract class OllamaFunctionTool implements OllamaTool {
* This is used by Ollama to know what the tool does
* @return The description of the tool
*/
- abstract public String description();
+ public String description(){
+ return null;
+ }
/**
* The parameters of the tool
* This is used by Ollama to know what parameters the tool takes
* If null, the tool does not take any parameters
+ *
* @return The parameters of the tool or null if the tool does not take any parameters
*/
+ @NotNull
abstract public OllamaPerameter parameters();
/**
diff --git a/Display/src/main/java/me/zacharias/chat/display/PythonRunner.java b/Display/src/main/java/me/zacharias/chat/display/PythonRunner.java
index cbeef69..ce96498 100644
--- a/Display/src/main/java/me/zacharias/chat/display/PythonRunner.java
+++ b/Display/src/main/java/me/zacharias/chat/display/PythonRunner.java
@@ -1,9 +1,6 @@
package me.zacharias.chat.display;
import com.github.dockerjava.api.DockerClient;
-import com.github.dockerjava.api.async.ResultCallback;
-import com.github.dockerjava.api.command.*;
-import com.github.dockerjava.api.model.Frame;
import com.github.dockerjava.api.model.HostConfig;
import com.github.dockerjava.core.DefaultDockerClientConfig;
import com.github.dockerjava.core.DockerClientImpl;
@@ -14,8 +11,10 @@ import me.zacharias.chat.core.Pair;
import me.zacharias.chat.ollama.*;
import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
import org.apache.commons.io.IOUtils;
+import org.jetbrains.annotations.NotNull;
import org.json.JSONArray;
import org.json.JSONObject;
+import org.jspecify.annotations.NonNull;
import java.io.*;
import java.net.ServerSocket;
@@ -28,7 +27,6 @@ import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
-import java.util.logging.Logger;
import static me.zacharias.chat.core.Core.writeLog;
@@ -266,7 +264,7 @@ public class PythonRunner extends OllamaFunctionTool {
}
@Override
- public String name() {
+ public @NonNull String name() {
return "python_runner";
}
@@ -276,7 +274,7 @@ public class PythonRunner extends OllamaFunctionTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("code", OllamaPerameter.OllamaPerameterBuilder.Type.STRING, "The code to be executed, optional if a file with the same name exists.")
.addProperty("name", OllamaPerameter.OllamaPerameterBuilder.Type.STRING, "The name of the python code")
diff --git a/Display/src/main/java/me/zacharias/chat/display/TimeTool.java b/Display/src/main/java/me/zacharias/chat/display/TimeTool.java
index 8d365a3..0c19253 100644
--- a/Display/src/main/java/me/zacharias/chat/display/TimeTool.java
+++ b/Display/src/main/java/me/zacharias/chat/display/TimeTool.java
@@ -4,6 +4,8 @@ import me.zacharias.chat.ollama.OllamaFunctionArgument;
import me.zacharias.chat.ollama.OllamaFunctionTool;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
+import org.jetbrains.annotations.NotNull;
+import org.jspecify.annotations.NonNull;
import java.util.Date;
@@ -20,12 +22,12 @@ public class TimeTool extends OllamaFunctionTool {
}
@Override
- public String name() {
+ public @NonNull String name() {
return "get_current_date";
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return null;
}
diff --git a/GeniusAPI/src/main/java/me/zacharias/neuro/dock/genius/endpoints/FindSong.java b/GeniusAPI/src/main/java/me/zacharias/neuro/dock/genius/endpoints/FindSong.java
index 443ca20..6c10ece 100644
--- a/GeniusAPI/src/main/java/me/zacharias/neuro/dock/genius/endpoints/FindSong.java
+++ b/GeniusAPI/src/main/java/me/zacharias/neuro/dock/genius/endpoints/FindSong.java
@@ -7,6 +7,7 @@ import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
import me.zacharias.neuro.dock.genius.GeniusEndpoint;
import me.zacharias.neuro.dock.genius.GeniusEndpointTool;
import me.zacharias.neuro.dock.genius.GeniusTools;
+import org.jetbrains.annotations.NotNull;
import org.json.JSONArray;
import org.json.JSONObject;
@@ -21,7 +22,7 @@ public class FindSong extends GeniusEndpointTool {
}
@Override
- public String name() {
+ public @NotNull String name() {
return "findsong";
}
@@ -31,7 +32,7 @@ public class FindSong extends GeniusEndpointTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("title", STRING, "The title, artitst, and song_id of the song to find.", true)
.build();
diff --git a/GeniusAPI/src/main/java/me/zacharias/neuro/dock/genius/endpoints/GetLyrics.java b/GeniusAPI/src/main/java/me/zacharias/neuro/dock/genius/endpoints/GetLyrics.java
index 5a4818e..ffacd21 100644
--- a/GeniusAPI/src/main/java/me/zacharias/neuro/dock/genius/endpoints/GetLyrics.java
+++ b/GeniusAPI/src/main/java/me/zacharias/neuro/dock/genius/endpoints/GetLyrics.java
@@ -7,6 +7,7 @@ import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
import me.zacharias.neuro.dock.genius.GeniusEndpoint;
import me.zacharias.neuro.dock.genius.GeniusEndpointTool;
import me.zacharias.neuro.dock.genius.GeniusTools;
+import org.jetbrains.annotations.NotNull;
import org.json.JSONObject;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
@@ -15,8 +16,6 @@ import org.jsoup.nodes.Node;
import org.jsoup.nodes.TextNode;
import org.jsoup.select.Elements;
-import java.util.stream.Collectors;
-
import static me.zacharias.chat.core.Core.writeLog;
/**
@@ -40,7 +39,7 @@ public class GetLyrics extends GeniusEndpointTool {
}
@Override
- public String name() {
+ public @NotNull String name() {
return "get_lyrics";
}
@@ -50,7 +49,7 @@ public class GetLyrics extends GeniusEndpointTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("song_id", OllamaPerameter.OllamaPerameterBuilder.Type.INT, "The ID of the song to get lyrics for.", true)
.build();
diff --git a/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetAnimeDetails.java b/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetAnimeDetails.java
index 08c4383..fb03cbe 100644
--- a/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetAnimeDetails.java
+++ b/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetAnimeDetails.java
@@ -7,10 +7,10 @@ import me.zacharias.chat.ollama.OllamaFunctionArgument;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
+import org.jetbrains.annotations.NotNull;
import org.json.JSONArray;
import org.json.JSONObject;
-import java.util.ArrayList;
import java.util.HashMap;
@MALEndpoint
@@ -20,7 +20,7 @@ public class GetAnimeDetails extends MALEndpointTool {
}
@Override
- public String name() {
+ public @NotNull String name() {
return "get_anime_details";
}
@@ -30,7 +30,7 @@ public class GetAnimeDetails extends MALEndpointTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("id", OllamaPerameter.OllamaPerameterBuilder.Type.INT, "The id of the anime", true)
.addProperty("fields", OllamaPerameter.OllamaPerameterBuilder.Type.ARRAY, "The fields to return, defaults to [\"id\", \"title\", \"synopsis\", \"genres\"]")
diff --git a/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetAnimeList.java b/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetAnimeList.java
index 128c635..3033521 100644
--- a/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetAnimeList.java
+++ b/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetAnimeList.java
@@ -7,11 +7,11 @@ import me.zacharias.chat.ollama.OllamaFunctionArgument;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
+import org.jetbrains.annotations.NotNull;
import org.json.JSONArray;
import org.json.JSONObject;
import java.util.HashMap;
-import java.util.Map;
@MALEndpoint
public class GetAnimeList extends MALEndpointTool {
@@ -20,7 +20,7 @@ public class GetAnimeList extends MALEndpointTool {
}
@Override
- public String name() {
+ public @NotNull String name() {
return "get_anime_list";
}
@@ -30,7 +30,7 @@ public class GetAnimeList extends MALEndpointTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("query", OllamaPerameter.OllamaPerameterBuilder.Type.STRING, "The query to search for",true)
.addProperty("offset", OllamaPerameter.OllamaPerameterBuilder.Type.INT, "The offset to start from")
diff --git a/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetManagDetails.java b/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetManagDetails.java
index 2992789..c500d60 100644
--- a/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetManagDetails.java
+++ b/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetManagDetails.java
@@ -7,6 +7,7 @@ import me.zacharias.chat.ollama.OllamaFunctionArgument;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
+import org.jetbrains.annotations.NotNull;
import org.json.JSONArray;
import org.json.JSONObject;
@@ -19,7 +20,7 @@ public class GetManagDetails extends MALEndpointTool {
}
@Override
- public String name() {
+ public @NotNull String name() {
return "get_manag_details";
}
@@ -29,7 +30,7 @@ public class GetManagDetails extends MALEndpointTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("id", OllamaPerameter.OllamaPerameterBuilder.Type.INT, "The id of the manga", true)
.addProperty("fields", OllamaPerameter.OllamaPerameterBuilder.Type.ARRAY, "The fields to return, defaults to [\"id\", \"title\", \"synopsis\", \"genres\"]")
diff --git a/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetMangaList.java b/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetMangaList.java
index eba881d..7994995 100644
--- a/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetMangaList.java
+++ b/MALAPITool/src/main/java/me/zacharias/chat/mal/api/endpoints/GetMangaList.java
@@ -7,6 +7,7 @@ import me.zacharias.chat.ollama.OllamaFunctionArgument;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
import me.zacharias.chat.ollama.exceptions.OllamaToolErrorException;
+import org.jetbrains.annotations.NotNull;
import org.json.JSONArray;
import org.json.JSONObject;
@@ -19,7 +20,7 @@ public class GetMangaList extends MALEndpointTool {
}
@Override
- public String name() {
+ public @NotNull String name() {
return "get_manga_list";
}
@@ -29,7 +30,7 @@ public class GetMangaList extends MALEndpointTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("query", OllamaPerameter.OllamaPerameterBuilder.Type.STRING, "The query to search for",true)
.addProperty("offset", OllamaPerameter.OllamaPerameterBuilder.Type.INT, "The offset to start from")
diff --git a/WikipediaTool/src/main/java/me/zacharias/neuro/dock/wikipedia/GetWikiPageText.java b/WikipediaTool/src/main/java/me/zacharias/neuro/dock/wikipedia/GetWikiPageText.java
index 33f34d8..9f9075a 100644
--- a/WikipediaTool/src/main/java/me/zacharias/neuro/dock/wikipedia/GetWikiPageText.java
+++ b/WikipediaTool/src/main/java/me/zacharias/neuro/dock/wikipedia/GetWikiPageText.java
@@ -5,12 +5,13 @@ import me.zacharias.chat.ollama.OllamaFunctionArgument;
import me.zacharias.chat.ollama.OllamaFunctionTool;
import me.zacharias.chat.ollama.OllamaPerameter;
import me.zacharias.chat.ollama.OllamaToolRespnce;
+import org.jetbrains.annotations.NotNull;
public class GetWikiPageText extends OllamaFunctionTool {
Wiki wiki = WikipediaTool.wiki;
@Override
- public String name() {
+ public @NotNull String name() {
return "get_wiki_page_text";
}
@@ -20,7 +21,7 @@ public class GetWikiPageText extends OllamaFunctionTool {
}
@Override
- public OllamaPerameter parameters() {
+ public @NotNull OllamaPerameter parameters() {
return OllamaPerameter.builder()
.addProperty("title", OllamaPerameter.OllamaPerameterBuilder.Type.STRING, "The title of the Wikipedia page to retrieve text from.", true)
.build();