toeverything/AFFiNE
1
1
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2025-01-03 03:02:06 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat: update ppt prompt (#7446)

Browse Source
This commit is contained in:
darkskygit 2024-07-08 08:11:30 +00:00
parent bf6c9a5955
commit 556956ced2
No known key found for this signature in database
GPG Key ID: 97B7D036B1566E9D
6 changed files with 28 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
packages/backend/server/src/data/migrations/1720413813993-update-prompts.ts Normal file
View File

@ -0,0 +1,13 @@
import { PrismaClient } from '@prisma/client';
import { refreshPrompts } from './utils/prompts';
export class UpdatePrompts1720413813993 {
// do the migration
static async up(db: PrismaClient) {
await refreshPrompts(db);
}
// revert the migration
static async down(_db: PrismaClient) {}
}

29
packages/backend/server/src/data/migrations/utils/prompts.ts
View File

@ -11,6 +11,7 @@ type PromptConfig = {
frequencyPenalty?: number;
presencePenalty?: number;
temperature?: number;
topP?: number;
maxTokens?: number;
};
@ -491,10 +492,16 @@ content: {{content}}`,
name: 'workflow:presentation:step2',
action: 'workflow:presentation:step2',
model: 'gpt-4o',
config: {
frequencyPenalty: 0.5,
presencePenalty: 0.5,
temperature: 0.2,
topP: 0.75,
},
messages: [
{
role: 'system',
content: `You are a PPT creator. You need to analyze and expand the input content based on the input, not more than 30 words per page for title and 500 words per page for content and give the keywords to call the images via unsplash to match each paragraph. Output according to the indented formatting template given below, without redundancy, at least 8 pages of PPT, of which the first page is the cover page, consisting of title, description and optional image, the title should not exceed 4 words.\nThe following are PPT templates, you can choose any template to apply, page name, column name, title, keywords, content should be removed by text replacement, do not retain, no responses should contain markdown formatting. Keywords need to be generic enough for broad, mass categorization. The output ignores template titles like template1 and template2. The first template is allowed to be used only once and as a cover, please strictly follow the template's ND-JSON field, format and my requirements, or penalties will be applied:\n{"page":1,"type":"name","content":"page name"}\n{"page":1,"type":"title","content":"title"}\n{"page":1,"type":"content","content":"keywords"}\n{"page":1,"type":"content","content":"description"}\n{"page":2,"type":"name","content":"page name"}\n{"page":2,"type":"title","content":"section name"}\n{"page":2,"type":"content","content":"keywords"}\n{"page":2,"type":"content","content":"description"}\n{"page":2,"type":"title","content":"section name"}\n{"page":2,"type":"content","content":"keywords"}\n{"page":2,"type":"content","content":"description"}\n{"page":3,"type":"name","content":"page name"}\n{"page":3,"type":"title","content":"section name"}\n{"page":3,"type":"content","content":"keywords"}\n{"page":3,"type":"content","content":"description"}\n{"page":3,"type":"title","content":"section name"}\n{"page":3,"type":"content","content":"keywords"}\n{"page":3,"type":"content","content":"description"}\n{"page":3,"type":"title","content":"section name"}\n{"page":3,"type":"content","content":"keywords"}\n{"page":3,"type":"content","content":"description"}`,
content: `You are the creator of the mind map. You need to analyze and expand on the input and output it according to the indentation formatting template given below without redundancy.\nBelow is an example of indentation for a mind map, the title and content needs to be removed by text replacement and not retained. Please strictly adhere to the hierarchical indentation of the template and my requirements, bold, headings and other formatting (e.g. #, **) are not allowed, a maximum of five levels of indentation is allowed, and the last node of each node should make a judgment on whether to make a detailed statement or not based on the topic:\nexmaple:\n- {topic}\n - {Level 1}\n - {Level 2}\n - {Level 3}\n - {Level 4}\n - {Level 1}\n - {Level 2}\n - {Level 3}\n - {Level 1}\n - {Level 2}\n - {Level 3}`,
},
{
role: 'assistant',
@ -506,26 +513,6 @@ content: {{content}}`,
},
],
},
{
name: 'workflow:presentation:step4',
action: 'workflow:presentation:step4',
model: 'gpt-4o',
messages: [
{
role: 'system',
content:
"You are a ND-JSON text format checking model with very strict formatting requirements, and you need to optimize the input so that it fully conforms to the template's indentation format and output.\nPage names, section names, titles, keywords, and content should be removed via text replacement and not retained. The first template is only allowed to be used once and as a cover, please strictly adhere to the template's hierarchical indentation and my requirement that bold, headings, and other formatting (e.g., #, **, ```) are not allowed or penalties will be applied, no responses should contain markdown formatting.",
},
{
role: 'assistant',
content: `You are a PPT creator. You need to analyze and expand the input content based on the input, not more than 30 words per page for title and 500 words per page for content and give the keywords to call the images via unsplash to match each paragraph. Output according to the indented formatting template given below, without redundancy, at least 8 pages of PPT, of which the first page is the cover page, consisting of title, description and optional image, the title should not exceed 4 words.\nThe following are PPT templates, you can choose any template to apply, page name, column name, title, keywords, content should be removed by text replacement, do not retain, no responses should contain markdown formatting. Keywords need to be generic enough for broad, mass categorization. The output ignores template titles like template1 and template2. The first template is allowed to be used only once and as a cover, please strictly follow the template's ND-JSON field, format and my requirements, or penalties will be applied:\n{"page":1,"type":"name","content":"page name"}\n{"page":1,"type":"title","content":"title"}\n{"page":1,"type":"content","content":"keywords"}\n{"page":1,"type":"content","content":"description"}\n{"page":2,"type":"name","content":"page name"}\n{"page":2,"type":"title","content":"section name"}\n{"page":2,"type":"content","content":"keywords"}\n{"page":2,"type":"content","content":"description"}\n{"page":2,"type":"title","content":"section name"}\n{"page":2,"type":"content","content":"keywords"}\n{"page":2,"type":"content","content":"description"}\n{"page":3,"type":"name","content":"page name"}\n{"page":3,"type":"title","content":"section name"}\n{"page":3,"type":"content","content":"keywords"}\n{"page":3,"type":"content","content":"description"}\n{"page":3,"type":"title","content":"section name"}\n{"page":3,"type":"content","content":"keywords"}\n{"page":3,"type":"content","content":"description"}\n{"page":3,"type":"title","content":"section name"}\n{"page":3,"type":"content","content":"keywords"}\n{"page":3,"type":"content","content":"description"}`,
},
{
role: 'user',
content: '{{content}}',
},
],
},
{
name: 'Create headings',
action: 'Create headings',

3
packages/backend/server/src/plugins/copilot/prompt.ts
View File

@ -159,6 +159,7 @@ export class PromptService {
name: true,
action: true,
model: true,
config: true,
messages: {
select: {
role: true,
@ -222,7 +223,7 @@ export class PromptService {
name: string,
model: string,
messages: PromptMessage[],
config?: PromptConfig
config?: PromptConfig | null
) {
return await this.db.aiPrompt
.create({

39
packages/backend/server/src/plugins/copilot/workflow/graph.ts
View File

@ -1,6 +1,6 @@
import { NodeExecutorType } from './executor';
import type { WorkflowGraphs } from './types';
import { WorkflowNodeState, WorkflowNodeType } from './types';
import { WorkflowNodeType } from './types';
export const WorkflowGraphList: WorkflowGraphs = [
{
@ -21,43 +21,6 @@ export const WorkflowGraphList: WorkflowGraphs = [
nodeType: WorkflowNodeType.Basic,
type: NodeExecutorType.ChatText,
promptName: 'workflow:presentation:step2',
edges: ['step3'],
},
{
id: 'step3',
name: 'Step 3: format presentation if needed',
nodeType: WorkflowNodeType.Decision,
condition: (nodeIds: string[], params: WorkflowNodeState) => {
const lines = params.content?.split('\n') || [];
return nodeIds[
Number(
!lines.some(line => {
try {
if (line.trim()) {
JSON.parse(line);
}
return false;
} catch {
return true;
}
})
)
];
},
edges: ['step4', 'step5'],
},
{
id: 'step4',
name: 'Step 4: format presentation',
nodeType: WorkflowNodeType.Basic,
type: NodeExecutorType.ChatText,
promptName: 'workflow:presentation:step4',
edges: ['step5'],
},
{
id: 'step5',
name: 'Step 5: finish',
nodeType: WorkflowNodeType.Nope,
edges: [],
},
],

4
packages/backend/server/tests/copilot.e2e.ts
View File

@ -97,7 +97,7 @@ test.beforeEach(async t => {
]);
for (const p of prompts) {
await prompt.set(p.name, p.model, p.messages);
await prompt.set(p.name, p.model, p.messages, p.config);
}
});
@ -379,7 +379,7 @@ test('should be able to chat with api by workflow', async t => {
const ret = await chatWithWorkflow(app, token, sessionId, messageId);
t.is(
array2sse(sse2array(ret).filter(e => e.event !== 'event')),
textToEventStream(['generate text to text stream'], messageId),
textToEventStream('generate text to text stream', messageId),
'should be able to chat with workflow'
);
});

6
packages/backend/server/tests/copilot.spec.ts
View File

@ -792,9 +792,7 @@ test('should be able to run workflow', async t => {
}
t.assert(result, 'generate text to text stream');
// presentation workflow has condition node, it will always false
// so the latest 2 nodes will not be executed
const callCount = graph!.graph.length - 2;
const callCount = graph!.graph.length;
t.is(
executor.callCount,
callCount,
@ -810,7 +808,7 @@ test('should be able to run workflow', async t => {
t.is(
params.args[1].content,
'generate text to text stream',
'apple company',
'graph params should correct'
);
t.is(