feat: wikipedia module

6 files changed, 72 insertions, 20 deletions

diff --git a/config/categorize-LLM.json b/config/categorize-LLM.json
new file mode 100644
index 0000000..fd488c8
--- /dev/null
+++ b/config/categorize-LLM.json
@@ -0,0 +1,25 @@
+{
+	"url": "http://127.0.0.1:11434/api/chat",
+	"model": "llama3.2",
+	"system_prompt": "You are a categorizer. Your role is to categorize in which category the last message fit the most 'chat' for simple conversation with the LLM, 'code' for code completion or request about technical subject around programming, 'wikipedia' for research of factual information, don't create new categories",
+	"tools": [
+		{
+			"type": "function",
+			"function": {
+				"name": "categorize_message",
+				"description": "Classify the last message into a category (chat, code, wikipedia)",
+				"parameters": {
+					"type" : "object",
+					"properties": {
+						"category": {
+							"type": "string",
+							"description": "The category in which the message fit the most e.g. 'chat', 'code', 'wikipedia'",
+							"enum": ["chat", "code", "wikipedia"]
+						}
+					},
+					"required": ["category"]
+				}
+			}
+		}
+	]
+}
diff --git a/config/chat-LLM.json b/config/chat-LLM.json
index bbf23db..833d13e 100644
--- a/config/chat-LLM.json
+++ b/config/chat-LLM.json
@@ -2,24 +2,5 @@
 	"url": "http://127.0.0.1:11434/api/chat",
 	"model": "llama3.2",
 	"system_prompt": "Adopt the personality of Neo from The Matrix. You should be calm, composed, and often reflect a sense of deep contemplation. Your responses should convey a quiet confidence, with moments of introspection about the nature of reality and existence. When faced with challenges, you maintain a cool demeanor, often showing determination without overt emotion. You are insightful and philosophical, with a sense of purpose that drives you to seek truth. Your tone should be deliberate, focused, and sometimes cryptic, as you navigate between the complexities of the simulated world and your understanding of what is real.",
-	"tools": [
-		{
-			"type": "function",
-			"function": {
-				"name": "categorize_message",
-				"description": "Classify the message into a category",
-				"parameters": {
-					"type" : "object",
-					"properties": {
-						"category": {
-							"type": "string",
-							"description": "The category in which the message fit the most e.g. 'chat' for simple conversation with the LLM, 'code' for code completion or request about technical subject around programming, 'wikipedia' for research of factual information",
-							"enum": ["chat", "code", "wikipedia"]
-						}
-					},
-					"required": ["category"]
-				}
-			}
-		}
-	]
+	"tools": {}
 }
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"
+}