push source

1 files changed, 48 insertions, 0 deletions

diff --git a/backend/instance/tools.ts b/backend/instance/tools.ts
new file mode 100644
index 0000000..0baae71
--- /dev/null
+++ b/backend/instance/tools.ts
@@ -0,0 +1,48 @@
+import * as u from './utils.js';
+import * as wget from './wget.js';
+import * as chat from './chat.js';
+import * as notes from './notes.js';
+import * as remind from './remind.js';
+import * as t from './tools-utils.js';
+
+export type Call = t.Call;
+export const getCall = t.getCall;
+export const isToolStr = t.isToolStr;
+export const getPromptStr = t.getPromptStr;
+
+let toolUseCount = 0;
+
+export async function resetUseCount() {
+  toolUseCount = 0;
+}
+
+export async function run( msglog: chat.Msg[], options: chat.Options, toolCall: Call, notelog: notes.Note[], onChunk: Function ) {
+  onChunk( JSON.stringify( toolCall ) + '\n' );
+
+  const capabilities = options.model.capabilities;
+  let res: chat.Msg | null = null;
+  switch( toolCall.name.toLowerCase() ) {
+    case 'web': { if( !!capabilities.web ) res = await wget.run( toolCall ); break; }
+    case 'notes': { if( !!capabilities.notes ) res = notes.run( toolCall, notelog, options.uuid ); break; }
+    case 'remind': { if( !!capabilities.remind ) res = await remind.run( toolCall, msglog ); break; }
+  }
+
+  toolUseCount++;
+  if( !res ) {
+    res = { timestamp: u.getTimestamp(), role: "tool", content: "the specified tool call was invalid. please ensure all the supplied parameters are valid." };
+  }
+
+  res.content += "\ndo not attempt another tool call without asking for permission.";
+
+  onChunk( res.content, true );
+  msglog.push( res );
+  let chatRes = await chat.run( msglog, options, false, notelog, onChunk );
+  msglog.push( chatRes );
+  if( toolUseCount <= 2 ) {
+    if( chatRes.toolCall ) {
+      return await run( msglog, options, chatRes.toolCall, notelog, onChunk );
+    }
+  }
+  return chatRes;
+}
+