File size: 3,976 Bytes
57aa51e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 | import { Injectable } from '@nestjs/common';
import { BaseMessage, HumanMessage } from '@langchain/core/messages';
import { END, START, StateGraph } from '@langchain/langgraph';
import { ChatOpenAI } from '@langchain/openai';
import { ChatPromptTemplate } from '@langchain/core/prompts';
import { agentCategories } from '@gitroom/nestjs-libraries/agent/agent.categories';
import { z } from 'zod';
import { agentTopics } from '@gitroom/nestjs-libraries/agent/agent.topics';
import { PostsService } from '@gitroom/nestjs-libraries/database/prisma/posts/posts.service';
const model = new ChatOpenAI({
apiKey: process.env.OPENAI_API_KEY || 'sk-proj-',
model: 'gpt-4o-2024-08-06',
temperature: 0,
});
interface WorkflowChannelsState {
messages: BaseMessage[];
topic?: string;
category: string;
hook?: string;
content?: string;
}
const category = z.object({
category: z.string().describe('The category for the post'),
});
const topic = z.object({
topic: z.string().describe('The topic of the post'),
});
const hook = z.object({
hook: z.string().describe('The hook of the post'),
});
@Injectable()
export class AgentGraphInsertService {
constructor(private _postsService: PostsService) {}
static state = () =>
new StateGraph<WorkflowChannelsState>({
channels: {
messages: {
reducer: (currentState, updateValue) =>
currentState.concat(updateValue),
default: () => [],
},
topic: null,
category: null,
hook: null,
content: null,
},
});
async findCategory(state: WorkflowChannelsState) {
const { messages } = state;
const structuredOutput = model.withStructuredOutput(category);
return ChatPromptTemplate.fromTemplate(
`
You are an assistant that get a social media post and categorize it into to one from the following categories:
{categories}
Here is the post:
{post}
`
)
.pipe(structuredOutput)
.invoke({
post: messages[0].content,
categories: agentCategories.join(', '),
});
}
findTopic(state: WorkflowChannelsState) {
const { messages } = state;
const structuredOutput = model.withStructuredOutput(topic);
return ChatPromptTemplate.fromTemplate(
`
You are an assistant that get a social media post and categorize it into one of the following topics:
{topics}
Here is the post:
{post}
`
)
.pipe(structuredOutput)
.invoke({
post: messages[0].content,
topics: agentTopics.join(', '),
});
}
findHook(state: WorkflowChannelsState) {
const { messages } = state;
const structuredOutput = model.withStructuredOutput(hook);
return ChatPromptTemplate.fromTemplate(
`
You are an assistant that get a social media post and extract the hook, the hook is usually the first or second of both sentence of the post, but can be in a different place, make sure you don't change the wording of the post use the exact text:
{post}
`
)
.pipe(structuredOutput)
.invoke({
post: messages[0].content,
});
}
async savePost(state: WorkflowChannelsState) {
await this._postsService.createPopularPosts({
category: state.category,
topic: state.topic!,
hook: state.hook!,
content: state.messages[0].content! as string,
});
return {};
}
newPost(post: string) {
const state = AgentGraphInsertService.state();
const workflow = state
.addNode('find-category', this.findCategory)
.addNode('find-topic', this.findTopic)
.addNode('find-hook', this.findHook)
.addNode('save-post', this.savePost.bind(this))
.addEdge(START, 'find-category')
.addEdge('find-category', 'find-topic')
.addEdge('find-topic', 'find-hook')
.addEdge('find-hook', 'save-post')
.addEdge('save-post', END);
const app = workflow.compile();
return app.invoke({
messages: [new HumanMessage(post)],
});
}
}
|