Merge pull request #8 from IQAIcom/fix-add-wiki-search-tool-to-get-answers-and-search-with-natural-language

Author: Royal-lobster

.changeset/grumpy-sites-fold.md

@@ -0,0 +1,5 @@
+---
+"@iqai/mcp-iqwiki": patch
+---
+
+Adds wiki search tool ot get answers and search with natural language

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+	"typescript.tsdk": "node_modules/typescript/lib"
+}

biome.json

@@ -1,7 +1,7 @@
 {
 	"$schema": "https://biomejs.dev/schemas/1.9.4/schema.json",
 	"files": {
-		"ignore": ["node_modules/**", "dist/**"]
+		"ignore": ["node_modules/**", "dist/**", "src/graphql-env.d.ts"]
 	},
 	"linter": {
 		"enabled": true,

src/graphql-env.d.ts

@@ -614,6 +614,14 @@ export type introspection_types = {
 					ofType: { kind: "SCALAR"; name: "String"; ofType: null };
 				};
 			};
+			block: {
+				name: "block";
+				type: {
+					kind: "NON_NULL";
+					name: never;
+					ofType: { kind: "SCALAR"; name: "Int"; ofType: null };
+				};
+			};
 			created: {
 				name: "created";
 				type: {
@@ -864,7 +872,6 @@ export type introspection_types = {
 	MarketRankData: {
 		kind: "UNION";
 		name: "MarketRankData";
-		// biome-ignore lint/complexity/noBannedTypes: <explanation>
 		fields: {};
 		possibleTypes: "NftRankListData" | "TokenRankListData";
 	};
@@ -1478,7 +1485,8 @@ export type introspection_types = {
 			| "CREATED"
 			| "UPDATED"
 			| "DAY"
-			| "DATE";
+			| "DATE"
+			| "TOKENS";
 	};
 	Project: {
 		kind: "OBJECT";
@@ -1942,6 +1950,14 @@ export type introspection_types = {
 					};
 				};
 			};
+			search: {
+				name: "search";
+				type: {
+					kind: "NON_NULL";
+					name: never;
+					ofType: { kind: "OBJECT"; name: "SearchResult"; ofType: null };
+				};
+			};
 			searchExplorers: {
 				name: "searchExplorers";
 				type: {
@@ -2339,6 +2355,28 @@ export type introspection_types = {
 			};
 		};
 	};
+	SearchResult: {
+		kind: "OBJECT";
+		name: "SearchResult";
+		fields: {
+			answer: {
+				name: "answer";
+				type: { kind: "SCALAR"; name: "String"; ofType: null };
+			};
+			suggestions: {
+				name: "suggestions";
+				type: {
+					kind: "LIST";
+					name: never;
+					ofType: {
+						kind: "NON_NULL";
+						name: never;
+						ofType: { kind: "OBJECT"; name: "WikiSuggestion"; ofType: null };
+					};
+				};
+			};
+		};
+	};
 	SignaturePayloadInput: {
 		kind: "INPUT_OBJECT";
 		name: "SignaturePayloadInput";
@@ -2417,7 +2455,6 @@ export type introspection_types = {
 	SlugResult: {
 		kind: "UNION";
 		name: "SlugResult";
-		// biome-ignore lint/complexity/noBannedTypes: <explanation>
 		fields: {};
 		possibleTypes: "Slug" | "Valid";
 	};
@@ -3193,7 +3230,6 @@ export type introspection_types = {
 	UserWikis: {
 		kind: "UNION";
 		name: "UserWikis";
-		// biome-ignore lint/complexity/noBannedTypes: <explanation>
 		fields: {};
 		possibleTypes: "UserActivity" | "WikiCount";
 	};
@@ -3505,6 +3541,28 @@ export type introspection_types = {
 			};
 		};
 	};
+	WikiSuggestion: {
+		kind: "OBJECT";
+		name: "WikiSuggestion";
+		fields: {
+			id: {
+				name: "id";
+				type: {
+					kind: "NON_NULL";
+					name: never;
+					ofType: { kind: "SCALAR"; name: "ID"; ofType: null };
+				};
+			};
+			title: {
+				name: "title";
+				type: {
+					kind: "NON_NULL";
+					name: never;
+					ofType: { kind: "SCALAR"; name: "String"; ofType: null };
+				};
+			};
+		};
+	};
 	WikiUrl: {
 		kind: "OBJECT";
 		name: "WikiUrl";

src/index.ts

@@ -4,6 +4,7 @@ import { getUserCreatedWikisTool } from "./tools/get-user-created-wikis.js";
 import { getUserEditedWikisTool } from "./tools/get-user-edited-wikis.js";
 import { getWikiActivitiesTool } from "./tools/get-user-wiki-activities.js";
 import { getWikiTool } from "./tools/get-wiki.js";
+import { searchWikiTool } from "./tools/search-wiki.js";
 
 async function main() {
 	console.log("Initializing IQ Wiki MCP Server...");
@@ -17,6 +18,7 @@ async function main() {
 	server.addTool(getUserCreatedWikisTool);
 	server.addTool(getUserEditedWikisTool);
 	server.addTool(getWikiActivitiesTool);
+	server.addTool(searchWikiTool);
 
 	try {
 		await server.start({

src/lib/queries.ts

@@ -148,3 +148,15 @@ export const WIKI_QUERY = graphql(`
     }
   }
 `);
+
+export const SEARCH_WIKIS_QUERY = graphql(`
+  query searchWikis($query: String!) {
+    search(query: $query) {
+      answer
+      suggestions {
+        id
+        title
+      }
+    }
+  }
+`);

src/services/search-wiki.ts

@@ -0,0 +1,27 @@
+import dedent from "dedent";
+import { client } from "../lib/graphql.js";
+import { SEARCH_WIKIS_QUERY } from "../lib/queries.js";
+
+export class SearchWikiService {
+	async execute(query: string) {
+		const response = await client.request(SEARCH_WIKIS_QUERY, { query });
+		return response.search;
+	}
+
+	format(search: Awaited<ReturnType<typeof this.execute>>) {
+		const formattedSearch = dedent`
+			📜 Search Results
+			- Answer: ${search.answer}
+			- Wiki Suggestions: ${(search.suggestions ?? [])
+				.map(
+					(suggestion: {
+						id: string;
+						title: string;
+					}) => `${suggestion.title} (${suggestion.id})`,
+				)
+				.join(", ")}
+		`;
+
+		return formattedSearch;
+	}
+}

src/tools/search-wiki.ts

@@ -0,0 +1,28 @@
+import { z } from "zod";
+import { SearchWikiService } from "../services/search-wiki.js";
+
+const searchWikiParams = z.object({
+	query: z.string().min(1).describe("The query to search for"),
+});
+
+type SearchWikiParams = z.infer<typeof searchWikiParams>;
+
+export const searchWikiTool = {
+	name: "SEARCH_WIKI",
+	description: "Search for a wiki from IQ.wiki by query",
+	parameters: searchWikiParams,
+	execute: async (params: SearchWikiParams) => {
+		try {
+			const service = new SearchWikiService();
+			const search = await service.execute(params.query);
+
+			return service.format(search);
+		} catch (error) {
+			if (error instanceof Error) {
+				console.log(`Error in SEARCH_WIKI tool: ${error.message}`);
+				return `Error searching for wiki: ${error.message}`;
+			}
+			return "An unknown error occurred while searching for wiki data";
+		}
+	},
+} as const;