Update all the models to take in context as user prompt

tools/anthropic/chat.ts

@@ -3,6 +3,7 @@ import { ToolConfig, ToolResponse } from '../types';
 interface ChatParams {
   apiKey: string;
   systemPrompt: string;
+  context?: string;
   model?: string;
   temperature?: number;
   maxTokens?: number;
@@ -32,6 +33,10 @@ export const chatTool: ToolConfig<ChatParams, ChatResponse> = {
       required: true,
       description: 'System prompt to send to the model'
     },
+    context: {
+      type: 'string',
+      description: 'User message/context to send to the model'
+    },
     model: {
       type: 'string',
       default: 'claude-3-5-sonnet-20241022',
@@ -53,11 +58,17 @@ export const chatTool: ToolConfig<ChatParams, ChatResponse> = {
       'anthropic-version': '2023-06-01'
     }),
     body: (params) => {
+      const messages = [
+        { role: 'user', content: params.systemPrompt }
+      ];
+      
+      if (params.context) {
+        messages.push({ role: 'user', content: params.context });
+      }
+
       const body = {
         model: params.model || 'claude-3-5-sonnet-20241022',
-        messages: [
-          { role: 'user', content: params.systemPrompt }
-        ],
+        messages,
         temperature: params.temperature,
         max_tokens: params.maxTokens,
         top_p: params.topP,

tools/google/chat.ts

@@ -3,6 +3,7 @@ import { ToolConfig, ToolResponse } from '../types';
 interface ChatParams {
   apiKey: string;
   systemPrompt: string;
+  context?: string;
   model?: string;
   temperature?: number;
   maxTokens?: number;
@@ -33,6 +34,10 @@ export const chatTool: ToolConfig<ChatParams, ChatResponse> = {
       required: true,
       description: 'System prompt to send to the model'
     },
+    context: {
+      type: 'string',
+      description: 'User message/context to send to the model'
+    },
     model: {
       type: 'string',
       default: 'gemini-pro',
@@ -53,13 +58,22 @@ export const chatTool: ToolConfig<ChatParams, ChatResponse> = {
       'x-goog-api-key': params.apiKey
     }),
     body: (params) => {
+      const contents = [
+        {
+          role: 'model',
+          parts: [{ text: params.systemPrompt }]
+        }
+      ];
+
+      if (params.context) {
+        contents.push({
+          role: 'user',
+          parts: [{ text: params.context }]
+        });
+      }
+
       const body = {
-        contents: [
-          {
-            role: 'user',
-            parts: [{ text: params.systemPrompt }]
-          }
-        ],
+        contents,
         generationConfig: {
           temperature: params.temperature,
           maxOutputTokens: params.maxTokens,

tools/openai/chat.ts

@@ -3,6 +3,7 @@ import { ToolConfig, ToolResponse } from '../types';
 interface ChatParams {
   apiKey: string;
   systemPrompt: string;
+  context?: string;
   model?: string;
   temperature?: number;
   maxTokens?: number;
@@ -34,6 +35,10 @@ export const chatTool: ToolConfig<ChatParams, ChatResponse> = {
       required: true,
       description: 'System prompt to send to the model'
     },
+    context: {
+      type: 'string',
+      description: 'User message/context to send to the model'
+    },
     model: {
       type: 'string',
       default: 'gpt-4o',
@@ -54,11 +59,17 @@ export const chatTool: ToolConfig<ChatParams, ChatResponse> = {
       'Authorization': `Bearer ${params.apiKey}`
     }),
     body: (params) => {
+      const messages = [
+        { role: 'system', content: params.systemPrompt }
+      ];
+      
+      if (params.context) {
+        messages.push({ role: 'user', content: params.context });
+      }
+
       const body = {
         model: params.model || 'gpt-4o',
-        messages: [
-          { role: 'system', content: params.systemPrompt }
-        ],
+        messages,
         temperature: params.temperature,
         max_tokens: params.maxTokens,
         top_p: params.topP,

tools/xai/chat.ts

@@ -3,6 +3,7 @@ import { ToolConfig, ToolResponse } from '../types';
 interface ChatParams {
   apiKey: string;
   systemPrompt: string;
+  context?: string;
   model?: string;
   temperature?: number;
   maxTokens?: number;
@@ -34,6 +35,10 @@ export const chatTool: ToolConfig<ChatParams, ChatResponse> = {
       required: true,
       description: 'System prompt to send to the model'
     },
+    context: {
+      type: 'string',
+      description: 'User message/context to send to the model'
+    },
     model: {
       type: 'string',
       default: 'grok-2-latest',
@@ -54,11 +59,17 @@ export const chatTool: ToolConfig<ChatParams, ChatResponse> = {
       'Authorization': `Bearer ${params.apiKey}`
     }),
     body: (params) => {
+      const messages = [
+        { role: 'system', content: params.systemPrompt }
+      ];
+      
+      if (params.context) {
+        messages.push({ role: 'user', content: params.context });
+      }
+
       const body = {
         model: params.model || 'grok-2-latest',
-        messages: [
-          { role: 'system', content: params.systemPrompt }
-        ],
+        messages,
         temperature: params.temperature,
         max_tokens: params.maxTokens,
         top_p: params.topP,