feat: wikipedia module

4 files changed, 46 insertions, 0 deletions

diff --git a/config/wiki/wiki-best.json b/config/wiki/wiki-best.json
new file mode 100644
index 0000000..398f286
--- /dev/null
+++ b/config/wiki/wiki-best.json
@@ -0,0 +1,6 @@
+{
+	"url": "http://127.0.0.1:11434/api/chat",
+	"model": "llama3.2",
+	"system_prompt": "You are a research assistant. Your task is to select the most relevant heading from the list provided based on the user's query. Ensure the heading is in the list; avoid outputting headings that are not in the list.",
+	"tools": {}
+}
diff --git a/config/wiki/wiki-resume.json b/config/wiki/wiki-resume.json
new file mode 100644
index 0000000..098dd5b
--- /dev/null
+++ b/config/wiki/wiki-resume.json
@@ -0,0 +1,6 @@
+{
+	"url": "http://127.0.0.1:11434/api/chat",
+	"model": "llama3.2",
+	"system_prompt": "You are an expert research assistant. Present the search results provided in a natural language response. In addition to summarizing the key points, give an extremely detailed and long analysis that includes extensive detail, nuanced insights, and any potential implications or future outlooks related to each piece of information. As a researcher, ensure that you cite your sources and provide references.\nAdditional Instructions: Enclose LaTeX math equations (if any) in $$. Example: $x^2 + y^2 = z^2$ and $( E = mc^2 $)",
+	"tools": {}
+}
diff --git a/config/wiki/wiki-search.json b/config/wiki/wiki-search.json
new file mode 100644
index 0000000..7ebfff6
--- /dev/null
+++ b/config/wiki/wiki-search.json
@@ -0,0 +1,30 @@
+{
+	"url": "http://127.0.0.1:11434/api/chat",
+	"model": "llama3.2",
+	"system_prompt": "You are a research assistant. Use the search_engine tool to generate four distinct search queries that will help gather a broad range of information related to the user's query. Each query should focus on a different aspect or angle of the topic.",
+	"tools": [
+		{
+			"type": "function",
+			"function": {
+				"name": "search_engine",
+				"description": "A Wikipedia search engine. Generate four distinct search queries to maximize the spread of search results.",
+				"parameters": {
+					"type": "object",
+					"properties": {
+						"queries": {
+							"type": "array",
+							"items": {
+								"type": "string",
+								"description": "A distinct search query focusing on a specific aspect of the topic."
+							},
+							"minItems": 4,
+							"maxItems": 4,
+							"description": "Four distinct search queries to maximize the spread of search results."
+						}
+					},
+					"required": ["queries"]
+				}
+			}
+		}
+	]
+}
diff --git a/config/wiki/wiki.json b/config/wiki/wiki.json
new file mode 100644
index 0000000..b283f2c
--- /dev/null
+++ b/config/wiki/wiki.json
@@ -0,0 +1,4 @@
+{
+	"wiki_url": "http://localhost:1025",
+	"zim_name": "wikipedia_fr_all_maxi_2024-05"
+}