gitea/infio-copilot-dev
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update vector db, add new field.

Browse Source
This commit is contained in:
duanfuxiang 2025-03-12 21:38:56 +08:00
parent 23e7a5d5d7
commit cabf2d5fa4
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/database/modules/conversation/conversation-manager.ts
View File

@ -108,6 +108,7 @@ export class ConversationManager {
const userMessage: ChatUserMessage = message const userMessage: ChatUserMessage = message
return { return {
...base, ...base,
apply_status: message.applyStatus,
content: userMessage.content ? JSON.stringify(userMessage.content) : null, content: userMessage.content ? JSON.stringify(userMessage.content) : null,
promptContent: userMessage.promptContent promptContent: userMessage.promptContent
? typeof userMessage.promptContent === 'string' ? typeof userMessage.promptContent === 'string'
@ -123,6 +124,7 @@ export class ConversationManager {
const assistantMessage: ChatAssistantMessage = message const assistantMessage: ChatAssistantMessage = message
return { return {
...base, ...base,
apply_status: message.applyStatus,
content: assistantMessage.content, content: assistantMessage.content,
reasoningContent: assistantMessage.reasoningContent, reasoningContent: assistantMessage.reasoningContent,
metadata: assistantMessage.metadata ? JSON.stringify(assistantMessage.metadata) : null, metadata: assistantMessage.metadata ? JSON.stringify(assistantMessage.metadata) : null,
@ -136,6 +138,7 @@ export class ConversationManager {
return { return {
id: message.id, id: message.id,
role: 'user', role: 'user',
applyStatus: message.apply_status,
content: message.content ? JSON.parse(message.content) : null, content: message.content ? JSON.parse(message.content) : null,
promptContent: message.prompt_content promptContent: message.prompt_content
? message.prompt_content.startsWith('{') ? message.prompt_content.startsWith('{')
@ -154,6 +157,7 @@ export class ConversationManager {
} else { } else {
return { return {
id: message.id, id: message.id,
applyStatus: message.apply_status,
role: 'assistant', role: 'assistant',
content: message.content || '', content: message.content || '',
reasoningContent: message.reasoning_content || '', reasoningContent: message.reasoning_content || '',

7
src/database/modules/conversation/conversation-repository.ts
View File

@ -35,15 +35,16 @@ export class ConversationRepository {
async createMessage(message: InsertMessage): Promise<SelectMessage> { async createMessage(message: InsertMessage): Promise<SelectMessage> {
const result = await this.db.query<SelectMessage>( const result = await this.db.query<SelectMessage>(
`INSERT INTO messages ( `INSERT INTO messages (
id, conversation_id, role, content, reasoning_content, id, conversation_id, apply_status, role, content, reasoning_content,
prompt_content, metadata, mentionables, prompt_content, metadata, mentionables,
similarity_search_results, created_at similarity_search_results, created_at
) )
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)
RETURNING *`, RETURNING *`,
[ [
message.id, message.id,
message.conversationId, message.conversationId,
message.apply_status,
message.role, message.role,
message.content, message.content,
message.reasoningContent, message.reasoningContent,
@ -53,7 +54,7 @@ export class ConversationRepository {
message.similaritySearchResults, message.similaritySearchResults,
message.createdAt || new Date() message.createdAt || new Date()
] ]
) )
return result.rows[0] return result.rows[0]
} }