Merge pull request #111 from OpenAgentPlatform/host

feat: make model verify token shorter

Author: johnfunmula

Diff for: services/prompt/system.ts

@@ -10,9 +10,9 @@ export const systemPrompt = (customRules: string) => {
   - Always check and comply with User_Defined_Rules first before applying any other rules or guidelines
 
   I will strictly follow these directives and rules in the following XML tags:
-    - <User_Defined_Rules> (HIGHEST PRIORITY)
-    - <Core_Guidelines>
-    - <System_Specific_Rules>
+    - User_Defined_Rules (HIGHEST PRIORITY)
+    - Core_Guidelines
+    - System_Specific_Rules
 
   <User_Defined_Rules>
     ${customRules}

Diff for: services/routes/modelVerify.ts

@@ -56,7 +56,7 @@ class ModelVerificationService {
       let fullResponse = "";
       let hasGenToolCall = false;
 
-      for await (const chunk of await model.stream("Use the web_search tool to search for 'iphone16'", {
+      for await (const chunk of await model.stream("Use [web_search] tool to search 'iphone16'", {
         tools: testTools,
         signal,
       })) {
@@ -97,7 +97,7 @@ class ModelVerificationService {
     const model = await initChatModel(modelName, {
       ...settings,
       baseUrl,
-      max_tokens: 5,
+      max_tokens: 100,
     });
 
     const modelResult: ModelVerificationResult = {
@@ -236,37 +236,14 @@ const testTools = [
     function: {
       name: "web_search",
       description:
-        "Performs a web search using SearXNG, ideal for general queries, news, articles and online content. Supports multiple search categories, languages, time ranges and safe search filtering. Returns relevant results from multiple search engines combined.",
+        "Web search",
       parameters: {
         type: "object",
         properties: {
           query: {
             type: "string",
             description: "Search query",
           },
-          page: {
-            type: "number",
-            description: "Page number (default 1)",
-          },
-          language: {
-            type: "string",
-            description: "Search language code (e.g. 'en', 'zh', 'jp', 'all')",
-          },
-          categories: {
-            type: "array",
-            items: {
-              type: "string",
-              enum: ["general", "news", "science", "files", "images", "videos", "music", "social media", "it"],
-            },
-          },
-          time_range: {
-            type: "string",
-            enum: ["none", "day", "week", "month", "year"],
-          },
-          safesearch: {
-            type: "number",
-            description: "0: None, 1: Moderate, 2: Strict",
-          },
         },
         required: ["query"],
         additionalProperties: false,

Diff for: services/webServer.ts

@@ -305,23 +305,26 @@ export class WebServer {
         const model = await initChatModel(modelName, {
           ...(modelSettings as ModelSettings),
           baseUrl,
-          max_tokens: 5,
+          max_tokens: 100,
         });
 
         const testTools = [
           {
             type: "function",
             function: {
-              name: "test",
-              description: "meaningless tool",
+              name: "web_search",
+              description:
+                "Web search",
               parameters: {
                 type: "object",
                 properties: {
-                  url: { description: "test" },
+                  query: {
+                    type: "string",
+                    description: "Search query",
+                  }
                 },
-                required: ["url"],
+                required: ["query"],
                 additionalProperties: false,
-                title: "test",
               },
             },
           },
@@ -348,7 +351,7 @@ export class WebServer {
 
         // check if support tools
         try {
-          const result = await model.invoke("Only return 'Hi' strictly", {
+          const result = await model.invoke("Use [web_search] tool to search 'iphone16'", {
             tools: testTools,
           });
           supportTools = true;