Jayden/AFFiNE-Mirror
0
0
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2026-07-12 15:46:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(core): call copilot in tools (#13024)

Browse Source 
fix AI-298

#### PR Dependency Tree


* **PR #13024** 👈

This tree was auto-generated by
[Charcoal](https://github.com/danerwilliams/charcoal)

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

* **New Features**
* Document and code artifact generation tools now use a single prompt
field for user input, enabling more flexible content creation powered by
AI.

* **Bug Fixes**
* Improved error handling for missing prompt templates or providers
during document and code artifact generation.

* **Refactor**
* Simplified input schemas for document and code artifact tools,
consolidating multiple input fields into a single prompt.
* Output schemas updated to remove metadata and other unused fields for
a cleaner result.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->
This commit is contained in:
Peng Xiao
2025-07-04 15:23:09 +08:00
committed by GitHub
parent 24f1181069
commit d0beab9638
4 changed files with 81 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files
packages/backend/server/src/plugins/copilot/tools/doc-compose.ts
+26 -39
View File
@@ -2,64 +2,51 @@ import { Logger } from '@nestjs/common';
import { tool } from 'ai';
import { z } from 'zod';
import type { PromptService } from '../prompt';
import type { CopilotProviderFactory } from '../providers';
import { toolError } from './error';
const logger = new Logger('DocComposeTool');
export const createDocComposeTool = () => {
export const createDocComposeTool = (
promptService: PromptService,
factory: CopilotProviderFactory
) => {
return tool({
description:
'Write a new document with markdown content. This tool creates structured markdown content for documents including titles, sections, and formatting.',
parameters: z.object({
title: z.string().describe('The title of the document'),
content: z
userPrompt: z
.string()
.describe(
'The main content to write in markdown format. Include proper markdown formatting like headers (# ## ###), lists (- * 1.), links [text](url), code blocks ```code```, and other markdown elements as needed.'
'The user description of the document, will be used to generate the document'
),
sections: z
.array(
z.object({
heading: z.string().describe('Section heading'),
content: z.string().describe('Section content in markdown'),
})
)
.optional()
.describe('Optional structured sections for the document'),
metadata: z
.object({
tags: z
.array(z.string())
.optional()
.describe('Optional tags for the document'),
description: z
.string()
.optional()
.describe('Optional brief description of the document'),
})
.optional()
.describe('Optional metadata for the document'),
}),
execute: async ({ title, content, sections, metadata }) => {
execute: async ({ title, userPrompt }) => {
try {
let markdownContent = '';
markdownContent += `${content}\n\n`;
if (sections && sections.length > 0) {
for (const section of sections) {
markdownContent += `## ${section.heading}\n\n`;
markdownContent += `${section.content}\n\n`;
}
const prompt = await promptService.get('Write an article about this');
if (!prompt) {
throw new Error('Prompt not found');
}
const provider = await factory.getProviderByModel(prompt.model);
if (!provider) {
throw new Error('Provider not found');
}
const content = await provider.text(
{
modelId: prompt.model,
},
[...prompt.finish({}), { role: 'user', content: userPrompt }]
);
return {
title,
markdown: markdownContent.trim(),
markdown: content,
wordCount: content.split(/\s+/).length,
metadata: metadata || {},
tags: metadata?.tags || [],
description: metadata?.description || '',
};
} catch (err: any) {
logger.error(`Failed to write document: ${title}`, err);