updated gitbook

Author: azep-ninja

packages/plugin-gitbook/src/actions/query.ts

@@ -1,20 +1,15 @@
 import { Plugin } from "@ai16z/eliza";
-import { queryAction } from "./actions/query";
 import { gitbookProvider } from "./providers/gitbook";
-import { gitbookTemplate } from "./templates";
-
-// Export the template for use by runtime
-export { gitbookTemplate };
 
 export const gitbookPlugin: Plugin = {
     name: "GitBook Documentation",
     description: "Plugin for querying GitBook documentation",
-    actions: [queryAction],
+    actions: [],
     providers: [gitbookProvider],
     evaluators: []
 };
 
 export default gitbookPlugin;
 
-// Export all types and components
+// Export types for external use
 export * from './types';

packages/plugin-gitbook/src/index.ts

@@ -1,5 +1,5 @@
 import { Provider, IAgentRuntime, Memory, State } from "@ai16z/eliza";
-import { GitBookResponse } from '../types';
+import { GitBookResponse, GitBookClientConfig } from '../types';
 
 function cleanText(text: string): string {
     console.log('📝 Cleaning text input:', text);
@@ -15,18 +15,90 @@ function cleanText(text: string): string {
     return cleaned;
 }
 
+async function validateQuery(runtime: IAgentRuntime, text: string): Promise<boolean> {
+    console.log('🔍 Validating query text:', text);
+
+    // Default keywords
+    let keywords = {
+        contractQueries: ['contract', 'address'],
+        generalQueries: ['how', 'what', 'where', 'explain', 'show', 'list', 'tell'],
+        mustInclude: [],
+        shouldInclude: []
+    };
+
+    let documentTriggers: string[] = [];
+
+    try {
+        const gitbookConfig = runtime.character.clientConfig?.gitbook as GitBookClientConfig;
+        console.log('📋 GitBook Config:', gitbookConfig);
+
+        if (gitbookConfig) {
+            if (gitbookConfig.keywords) {
+                keywords = {
+                    ...keywords,
+                    ...gitbookConfig.keywords
+                };
+            }
+            if (gitbookConfig.documentTriggers) {
+                documentTriggers = gitbookConfig.documentTriggers;
+            }
+        }
+
+        const containsAnyWord = (text: string, words: string[] = []) => {
+            return words.length === 0 || words.some(word => {
+                if (word.includes(' ')) {
+                    return text.includes(word.toLowerCase());
+                }
+                const regex = new RegExp(`\\b${word}\\b`, 'i');
+                return regex.test(text);
+            });
+        };
+
+        const hasDocTrigger = containsAnyWord(text, documentTriggers);
+        const hasContractQuery = containsAnyWord(text, keywords.contractQueries);
+        const hasGeneralQuery = containsAnyWord(text, keywords.generalQueries);
+        const hasMustInclude = containsAnyWord(text, keywords.mustInclude);
+
+        const validationResults = {
+            hasDocTrigger,
+            hasContractQuery,
+            hasGeneralQuery,
+            hasMustInclude,
+            text
+        };
+
+        console.log('🔍 Validation Results:', validationResults);
+
+        const isValid = (hasDocTrigger || hasContractQuery || hasGeneralQuery) && hasMustInclude;
+        console.log('✅ Validation Result:', isValid);
+        return isValid;
+
+    } catch (error) {
+        console.error("❌ Error in validation:", error);
+        return false;
+    }
+}
+
 export const gitbookProvider: Provider = {
-    get: async (runtime: IAgentRuntime, message: Memory, _state?: State): Promise<string | null> => {
+    get: async (runtime: IAgentRuntime, message: Memory, _state?: State): Promise<string> => {
         console.log('🔄 GitBook Provider executing');
 
         try {
             const spaceId = runtime.getSetting("GITBOOK_SPACE_ID");
             if (!spaceId) {
                 console.error("❌ GitBook Space ID not configured");
-                return null;
+                return "";
             }
             console.log('✓ SpaceID configured:', spaceId);
 
+            const text = message.content.text.toLowerCase().trim();
+            const isValidQuery = await validateQuery(runtime, text);
+
+            if (!isValidQuery) {
+                console.log('⚠️ Query validation failed');
+                return "";
+            }
+
             const cleanedQuery = cleanText(message.content.text);
 
             console.log('📡 Making GitBook API request...');
@@ -46,17 +118,17 @@ export const gitbookProvider: Provider = {
 
             if (!response.ok) {
                 console.error('❌ GitBook API error:', response.status);
-                return null;
+                return "";
             }
 
             console.log('✓ GitBook API response received');
             const result: GitBookResponse = await response.json();
             console.log('📄 GitBook Response:', result?.answer?.text || 'No answer text');
 
-            return result.answer?.text || null;
+            return result.answer?.text || "";
         } catch (error) {
             console.error("❌ Error in GitBook provider:", error);
-            return null;
+            return "";
         }
     }
 };

packages/plugin-gitbook/src/providers/gitbook.ts

@@ -1,17 +1,18 @@
-import { Content } from "@ai16z/eliza";
-
+// Keywords configuration interface
 export interface GitBookKeywords {
     contractQueries?: string[];
     generalQueries?: string[];
     mustInclude?: string[];
     shouldInclude?: string[];
 }
 
+// Client configuration in character.json
 export interface GitBookClientConfig {
     keywords?: GitBookKeywords;
     documentTriggers?: string[];
 }
 
+// GitBook API response type
 export interface GitBookResponse {
     answer?: {
         text: string;