Postg Mem

🚀 PostgMem

PostgMem 是一個基於模型上下文協議（MCP）的服務器實現，藉助 PostgreSQL 和 pgvector 為 AI 應用提供向量內存存儲服務。它允許 AI 代理存儲、檢索和通過相似度搜索訪問記憶，為 AI 應用的記憶管理提供了高效解決方案。

🚀 快速開始

PostgMem 是一個基於 .NET 的服務，允許 AI 代理存儲、檢索和通過相似度搜索訪問記憶。它利用 PostgreSQL 和 pgvector 擴展來提供高效的相似性搜索功能。在開始使用前，需確保滿足先決條件，並完成相應的設置步驟。

✨ 主要特性

• 使用向量嵌入存儲結構化記憶。
• 按照 ID 檢索單個記憶。
• 通過向量相似性進行語義搜索。
• 使用標籤過濾搜索結果。
• 與模型上下文協議（MCP）集成，便於與 AI 代理使用。

📚 詳細文檔

技術

• .NET 9.0
• PostgreSQL 帶有 pgvector 擴展
• 模型上下文協議（MCP）
• ASP.NET Core
• Npgsql 實現 PostgreSQL 連接

先決條件

• .NET 9.0 SDK
• 已安裝 pgvector 擴展的 PostgreSQL 數據庫
• 文本嵌入 API（默認配置使用 Ollama）

設置

數據庫配置

1. 創建一個用於應用程序的 PostgreSQL 數據庫。
2. 在數據庫中安裝 pgvector 擴展：

   CREATE EXTENSION vector;
   

3. 創建 memories 表：

   CREATE TABLE memories (
       id UUID PRIMARY KEY,
       type TEXT NOT NULL,
       content JSONB NOT NULL,
       source TEXT NOT NULL,
       embedding VECTOR(384) NOT NULL,
       tags TEXT[] NOT NULL,
       confidence DOUBLE PRECISION NOT NULL,
       created_at TIMESTAMP WITH TIME ZONE NOT NULL,
       updated_at TIMESTAMP WITH TIME ZONE NOT NULL
   );
   

環境配置

在項目的 .env 文件中配置應用程序設置：

ConnectionStrings__Storage="Host=localhost;Database=mcp_memory;Username=postgres;Password=postgres"
Embeddings__ApiUrl=http://localhost:11434/
Embeddings__Model=all-minilm:33m-l12-v2-fp16

• ConnectionStrings__Storage：PostgreSQL 連接字符串。
• Embeddings__ApiUrl：嵌入式 API 的 URL（默認使用 Ollama）。
• Embeddings__Model：使用的嵌入模型。

運行應用程序

1. 切換到 PostgMem 目錄。
2. 運行應用程序：

   dotnet run
   

3. MCP 服務器將在 http://localhost:5000 默認可用。

MCP 工具

存儲

將新記憶存儲在數據庫中。 參數：

• type（字符串）：記憶的類型（例如，'對話'、'文檔' 等）。
• content（字符串）：記憶內容作為 JSON 對象。
• source（字符串）：記憶來源（例如，'用戶'、'系統' 等）。
• tags（字符串數組）：可選的記憶分類標籤。
• confidence（雙精度浮點數）：記憶的置信度分數（0.0 到 1.0）。

搜索

通過提供的文本搜索相似記憶。 參數：

• query（字符串）：要搜索的文本。
• limit（整數）：返回結果的最大數量（默認：10）。
• minSimilarity（雙精度浮點數）：最小相似度閾值（0.0 到 1.0）（默認：0.7）。
• filterTags（字符串數組）：可選的記憶過濾標籤。

獲取

通過 ID 檢索特定記憶。 參數：

• id（Guid）：要檢索的記憶 ID。

刪除

通過 ID 刪除記憶。 參數：

• id（Guid）：要刪除的記憶 ID。

🔧 技術細節

• Memory.cs：定義記憶數據模型。
• Storage.cs：處理數據庫操作，包括存儲和檢索記憶。
• EmbeddingService.cs：生成文本向量嵌入。
• MemoryTools.cs：實現 MCP 工具以與記憶存儲交互。

📄 許可證

[在此處填寫您的許可信息]

貢獻

[在此處添加貢獻指南，如果適用]

聯繫方式

[在此處提供聯繫方式或支持信息]