gitea/infio-copilot
1
0
Fork 0
mirror of https://github.com/EthanMarti/infio-copilot.git synced 2026-01-16 16:31:56 +00:00
Code Issues 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
return {
...base,
apply_status: message.applyStatus,
content: userMessage.content ? JSON.stringify(userMessage.content) : null,
promptContent: userMessage.promptContent
? typeof userMessage.promptContent === 'string'
@ -123,6 +124,7 @@ export class ConversationManager {
const assistantMessage: ChatAssistantMessage = message
return {
...base,
apply_status: message.applyStatus,
content: assistantMessage.content,
reasoningContent: assistantMessage.reasoningContent,
metadata: assistantMessage.metadata ? JSON.stringify(assistantMessage.metadata) : null,
@ -136,6 +138,7 @@ export class ConversationManager {
return {
id: message.id,
role: 'user',
applyStatus: message.apply_status,
content: message.content ? JSON.parse(message.content) : null,
promptContent: message.prompt_content
? message.prompt_content.startsWith('{')
@ -154,6 +157,7 @@ export class ConversationManager {
} else {
return {
id: message.id,
applyStatus: message.apply_status,
role: 'assistant',
content: message.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> {
const result = await this.db.query<SelectMessage>(
`INSERT INTO messages (
id, conversation_id, role, content, reasoning_content,
id, conversation_id, apply_status, role, content, reasoning_content,
prompt_content, metadata, mentionables,
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 *`,
[
message.id,
message.conversationId,
message.apply_status,
message.role,
message.content,
message.reasoningContent,
@ -53,7 +54,7 @@ export class ConversationRepository {
message.similaritySearchResults,
message.createdAt || new Date()
]
)
)
return result.rows[0]
}