fix: drives

.gral/functions/research.js

@@ -0,0 +1,62 @@
+import fs from 'node:fs';
+import path from 'node:path';
+
+const system = `Your goal is to write a research report based on the given prompt.
+Use the \`submit-report\` function when you have gathered all information.
+You can only read from \`docs\` and from the source code directories \`backend/src\` and \`frontend/src\`. Exploration in any other location is forbidden.
+Only read up to 10 files. If more context is needed, submit the report regardless with all the information available. This is the breakpoint to work iteratively wit hthe user
+Note: task is terminated after 20 turns, reasoning traces are not persisted after agent turns.
+`;
+
+function resolvePrompt (input) {
+  if (input.prompt) return input.prompt;
+
+  const candidates = ['prompt.md', 'prompt.txt', 'prompt'];
+  for (const name of candidates) {
+    const file = path.resolve(Gral.path, '.gral', name);
+    if (fs.existsSync(file)) {
+      return fs.readFileSync(file, 'utf8');
+    }
+  }
+
+  throw new Error("Prompt required.");
+}
+
+function writeReport(content) {
+  fs.writeFileSync(path.resolve(Gral.path, '.gral/report.md'), content);
+}
+
+export default async function research (input) {
+  const prompt = resolvePrompt(input);
+
+  let done = false;
+
+  let res = await Gral.run({
+    instruction: system,
+    input: prompt,
+    agentId: 7,
+    tools: [{
+      name: 'submit-report',
+      description: 'Submit a markdown report as your task deliverable.',
+      parameters: {
+        type: 'object',
+        properties: {
+          content: { type: 'string' },
+        },
+        required: ['content'],
+      },
+      handler: (args) => {
+        done = true;
+        writeReport(args.content);
+      }
+    }],
+    read: ['docs', 'frontend/src', 'backend/src'],
+    write: [],
+    defaultTools: ['list-dir', 'read-file', 'find-files', 'search-files'],
+
+    maxIter: 99,
+    verify: () => done,
+  });
+
+  console.log(res);
+};