mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-03-31 03:08:35 +08:00
8a9bbf3d6d
### What problem does this PR solve? Feat： Add Memory function by go ### Type of change - [x] New Feature (non-breaking change which adds functionality) --------- Co-authored-by: Yingfeng <yingfeng.zhang@gmail.com>
54 lines
2.7 KiB
Go
54 lines
2.7 KiB
Go
//
|
|
// Copyright 2026 The InfiniFlow Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
|
|
package model
|
|
|
|
// Memory memory model
|
|
type Memory struct {
|
|
ID string `gorm:"column:id;primaryKey;size:32" json:"id"`
|
|
Name string `gorm:"column:name;size:128;not null" json:"name"`
|
|
Avatar *string `gorm:"column:avatar;type:longtext" json:"avatar,omitempty"`
|
|
TenantID string `gorm:"column:tenant_id;size:32;not null;index" json:"tenant_id"`
|
|
MemoryType int64 `gorm:"column:memory_type;default:1;index" json:"memory_type"`
|
|
StorageType string `gorm:"column:storage_type;size:32;not null;default:table;index" json:"storage_type"`
|
|
EmbdID string `gorm:"column:embd_id;size:128;not null" json:"embd_id"`
|
|
TenantEmbdID *int64 `gorm:"column:tenant_embd_id;index" json:"tenant_embd_id,omitempty"`
|
|
LLMID string `gorm:"column:llm_id;size:128;not null" json:"llm_id"`
|
|
TenantLLMID *int64 `gorm:"column:tenant_llm_id;index" json:"tenant_llm_id,omitempty"`
|
|
Permissions string `gorm:"column:permissions;size:16;not null;default:me;index" json:"permissions"`
|
|
Description *string `gorm:"column:description;type:longtext" json:"description,omitempty"`
|
|
MemorySize int64 `gorm:"column:memory_size;default:5242880;not null" json:"memory_size"`
|
|
ForgettingPolicy string `gorm:"column:forgetting_policy;size:32;not null;default:FIFO" json:"forgetting_policy"`
|
|
Temperature float64 `gorm:"column:temperature;default:0.5;not null" json:"temperature"`
|
|
SystemPrompt *string `gorm:"column:system_prompt;type:longtext" json:"system_prompt,omitempty"`
|
|
UserPrompt *string `gorm:"column:user_prompt;type:longtext" json:"user_prompt,omitempty"`
|
|
BaseModel
|
|
}
|
|
|
|
// TableName specify table name
|
|
func (Memory) TableName() string {
|
|
return "memory"
|
|
}
|
|
|
|
// MemoryListItem represents a memory record with owner name from JOIN query.
|
|
// Uses struct embedding to extend Memory struct with owner_name from user table JOIN.
|
|
// Note: MemoryType is kept as int64 from Memory embedding; conversion to []string
|
|
// happens in the Service layer via CreateMemoryResponse.
|
|
type MemoryListItem struct {
|
|
Memory
|
|
OwnerName *string `json:"owner_name,omitempty"`
|
|
}
|